from mcp.server.fastmcp import FastMCP
from mlflow_mcp_server.tools.experiment_tools import (
get_experiment,
get_experiment_by_name,
search_experiments,
)
from mlflow_mcp_server.tools.run_tools import (
get_run,
get_experiment_runs,
)
from mlflow_mcp_server.tools.registered_models import (
get_registered_models,
get_model_versions,
)
mcp = FastMCP(
name="MLFlow MCP Server",
dependencies=["mlflow", "pydantic"],
)
mcp.add_tool(get_experiment)
mcp.add_tool(get_experiment_by_name)
mcp.add_tool(search_experiments)
mcp.add_tool(get_run)
mcp.add_tool(get_experiment_runs)
mcp.add_tool(get_registered_models)
mcp.add_tool(get_model_versions)
