使用 Trino MCP 访问《 bullshit 数据集》

概述

Trino 的《bullshit 数据集》通过 MCP（机器学习平台）提供了一个 REST API 接口，允许用户通过 SQL 查询直接访问数据。本文将介绍如何利用该接口进行数据查询，并展示一个简单的 Python 示例。

安装 Docker 环境

为了运行 Trino MCP 服务，请确保您的系统已安装 Docker 并且 Docker 已启动。然后，您可以使用以下命令拉取必要的镜像并启动服务：

docker-compose up -d trino-mcp

发送 SQL 查询请求

您可以通过发送 POST 请求到 Trino MCP 的 API 端点来执行 SQL 查询。以下是一个 Python 示例代码：

import requests
import json

# 设置 API 端点地址（默认 Docker 容器的 9097 端口）
api_url = "http://localhost:9097/api/query"

# 定义您的 SQL 查询
query_data = {
    "query": "SELECT * FROM memory.bullshit.real_bullshit_data LIMIT 5",
    "catalog": "memory",
    "schema": "bullshit"
}

# 发送 POST 请求
response = requests.post(api_url, json=query_data)

# 处理响应结果
if response.status_code == 200:
    results = response.json()
    if results["success"]:
        print(f"查询成功，返回 {results['results']['row_count']} 行数据")
        for row in results['results']['rows']:
            print(row)
    else:
        print(f"查询失败：{results.get('message', '未知错误')}")
else:
    print(f"HTTP 请求失败，状态码：{response.status_code}")

支持的 API 端点

Trino MCP 提供以下两个主要 API 端点：

1. Docker 容器 API（端口 9097）

import requests
import json

# 设置 API 端点地址（默认 Docker 容器的 9097 端口）
api_url = "http://localhost:9097/api/query"

# 定义您的 SQL 查询
query_data = {
    "query": "SELECT * FROM memory.bullshit.real_bullshit_data LIMIT 5",
    "catalog": "memory",
    "schema": "bullshit"
}

# 发送 POST 请求
response = requests.post(api_url, json=query_data)

# 处理响应结果
if response.status_code == 200:
    results = response.json()
    if results["success"]:
        print(f"查询成功，返回 {results['results']['row_count']} 行数据")
        for row in results['results']['rows']:
            print(row)
    else:
        print(f"查询失败：{results.get('message', '未知错误')}")
else:
    print(f"HTTP 请求失败，状态码：{response.status_code}")

2. 独立 Python API（端口 8008）

# 同样的代码，但使用不同的端口
api_url = "http://localhost:8008/query"

这两个 API 提供以下功能：

• GET /api：获取 API 文档和使用示例。
• POST /api/query：执行针对 Trino 的 SQL 查询。

这些 API 通过直接调用 REST 接口去除了包装脚本的需求，使得集成到 Claude、GPT 等 AI 系统中变得更加简单。

常见问题解答

1. API 返回 503 服务不可用错误

如果 Docker 容器的 API 返回 503 错误，请按照以下步骤排查：

解决方法

• 重建容器：

  docker-compose stop trino-mcp
  docker-compose rm -f trino-mcp
  docker-compose up -d trino-mcp
  

• 检查容器日志：

  docker logs trino_mcp_trino-mcp_1
  

• 验证 Trino 运行状态：

  curl -s http://localhost:9095/v1/info | jq
  

2. 端口冲突问题

如果独立 API 使用的默认端口（8008）出现占用，请按照以下步骤解决：

解决方法

• 修改代码中的端口号：

  打开 llm_trino_api.py 文件，找到最后一行并修改为所需的端口。

  uvicorn.run(app, host="127.0.0.1", port=8009)
  

• 运行修改后的服务：

  python llm_trino_api.py
  

总结

通过 Tr