MLflow MCP 服务器:自然语言接口
为 MLflow 提供自然语言查询接口,简化机器学习实验和模型管理。
快速入门
-
克隆仓库: git clone https://github.com/iRahulPandey/mlflowMCPServer.git cd mlflowMCPServer
-
创建虚拟环境并激活: python -m venv venv source venv/bin/activate (Windows 使用 venv\Scripts\activate)
-
安装依赖: pip install mcp[cli] langchain-mcp-adapters langchain-openai langgraph mlflow
-
设置 OpenAI API 密钥和 MLflow URI: export OPENAI_API_KEY=your_key_here export MLFLOW_TRACKING_URI=http://localhost:8080
-
启动 MCP 服务器: python mlflow_server.py
-
使用客户端查询: python mlflow_client.py "我的注册模型有哪些?"