Databricks MCP认证服务器

专为Databricks设计的自定义模型上下文协议(MCP)服务器,支持OBO(On-Behalf-Of)认证,用于测试连接、回显消息和获取用户信息。

  • by ksavchyn
  • Python
  • 本地部署
  • 1 星标
  • 更新于
在 GitHub 查看 ksavchyn/mcp-testing

使用说明

用于Rancher的自定义MCP服务器

具有OBO(代表)身份验证的自定义模型上下文协议(MCP)服务器,设计用于部署为copula应用程序。

特性

服务器实现了三个MCP工具:

  • test_连接:测试服务器连接和OBO身份验证状态
  • 回声:回显消息(有助于测试基本功能)
  • get_user_info:通过OBO身份验证检索当前用户信息

认证

此服务器通过以下方式使用copula OBO(代表)身份验证 ModelServingUserCredentials,允许它代表发出请求的经过身份验证的用户行事。

部署到ViewModel

先决条件

  1. 安装copula命令行界面
  2. 验证您的工作区:
    databricks auth login --host https://<your-workspace-hostname>
    

部署步骤

  1. 创建copula应用程序:

    databricks apps create mcp-cust-kat
    
  2. 上传和部署:

    DATABRICKS_USERNAME=$(databricks current-user me | jq -r .userName)
    databricks sync . "/Users/$DATABRICKS_USERNAME/mcp-cust-kat"
    databricks apps deploy mcp-cust-kat --source-code-path "/Workspace/Users/$DATABRICKS_USERNAME/mcp-cust-kat"
    
  3. 访问您的应用程序:

    • 应用程序的URL将在copula UI中可用
    • MCP端点: https://<app-url>/mcp
    • 健康检查: https://<app-url>/health

API终点

  • 得到/ -带有服务器信息的根端点
  • GET/健康 -具有OBO身份验证状态的健康检查端点
  • POST/mcp -主MCP JSON-RPC端点

测试服务器

部署后,您可以使用可用的MCP工具测试连接:

  1. test_连接:验证服务器是否正在运行并显示OBO身份验证状态
  2. 回声:简单的回声测试,验证请求/响应功能
  3. get_user_info:返回经过身份验证的用户的详细信息

本地开发

在本地运行以进行测试:

uv run python server.py

服务器将在端口8000上启动,同时支持HTTP和MCP协议。

项目结构

  • server.py -实现MCP协议的主要FastAPI应用程序
  • app.yaml -copula应用程序配置
  • pyproject.toml -Python项目配置和依赖关系
  • requirements.txt -应用程序的直接依赖关系