MLflow MCP 服务器:自然语言接口

MLflow MCP 服务器:自然语言接口

为 MLflow 提供自然语言查询接口,简化机器学习实验和模型管理。

快速入门

  1. 克隆仓库: git clone https://github.com/iRahulPandey/mlflowMCPServer.git cd mlflowMCPServer

  2. 创建虚拟环境并激活: python -m venv venv source venv/bin/activate (Windows 使用 venv\Scripts\activate)

  3. 安装依赖: pip install mcp[cli] langchain-mcp-adapters langchain-openai langgraph mlflow

  4. 设置 OpenAI API 密钥和 MLflow URI: export OPENAI_API_KEY=your_key_here export MLFLOW_TRACKING_URI=http://localhost:8080

  5. 启动 MCP 服务器: python mlflow_server.py

  6. 使用客户端查询: python mlflow_client.py "我的注册模型有哪些?"