host
972203c38f
- cloud/mcp_server.py：stdio MCP 服务，把云端任务 REST 翻成 tools/list / tools/call（health、list_projects、list_tasks、 get_task、create_task、cancel_task、get_task_cost、 get_task_trace），底层只走已有 REST，不另开任务逻辑； 鉴权只认调用方自己的 CODEX_CLOUD_API_KEY（缺了启动失败， 绝不打印），服务端权限 403 原样透传；create_task 只传已知 字段，未知字段丢弃；REST 报错时 isError=true。 单测 test_mcp_server 9 项（假 transport，不联网、不碰真 key）。 - 上游对齐 #80 的可合部分：config.example.toml 加 [features] daemon_auto_start = false（防 managed daemon 绕过 版本 pin）+ docs/upstream-164-alignment.md 对齐表；换稳定版 二进制是另一次主机侧部署动作，本批不做、不谎称已升级。
232 lines
9.9 KiB
Python
232 lines
9.9 KiB
Python
"""云端任务管理的 MCP 入口：把 REST API 翻成 MCP tools。
|
||
|
||
方向说明（别跟任务内 MCP 搞混）：项目内 agent 用的 MCP 是
|
||
`[mcp_servers.*]`（给 agent 接外部工具）；本文件是反方向——
|
||
让外部 MCP 客户端（编辑器、agent）用 tools/list / tools/call
|
||
管云端任务，底层一律走已有的 REST API，不另开一套任务逻辑。
|
||
|
||
协议：stdio 上每行一个 JSON-RPC 2.0 请求，支持 initialize、
|
||
ping、tools/list、tools/call；notifications/* 不回包（按 MCP
|
||
惯例）。tools/call 的结果一律是 content=[{type: text,
|
||
text: <REST 响应 JSON>}]，REST 报错时 isError=true，文本里
|
||
含 HTTP 状态与服务端 error，不伪造成功。
|
||
|
||
鉴权：只认调用方自己的 X-API-Key，从环境变量
|
||
CODEX_CLOUD_API_KEY 读（缺了就启动失败，绝不读文件、
|
||
绝不打印 key）；BASE 从 CODEX_CLOUD_BASE_URL 读，缺省
|
||
https://api.diante.me。MCP 服务本身不存任何凭证、不建
|
||
用户、不绕过服务端权限——服务端按 key 归属做 viewer/
|
||
contributor 等权限校验，403 原样透传。
|
||
|
||
工具（只包最常用的七个，不包管理端）：
|
||
health、list_projects、list_tasks、get_task、create_task、
|
||
cancel_task、get_task_cost、get_task_trace。
|
||
create_task 只传 project_id + prompt + 可选 executor/model/
|
||
effort/approval/priority，其余字段走服务端缺省；cancel_task
|
||
只做取消，不做真 kill 之外的任何事（真 kill 由 worker 的
|
||
kill_requests 机制处理，不在这里另起一套）。
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import sys
|
||
import urllib.error
|
||
import urllib.parse
|
||
import urllib.request
|
||
|
||
__all__ = ["MCPServer", "main"]
|
||
|
||
|
||
def _tool(name, description, properties, required):
|
||
return {
|
||
"name": name,
|
||
"description": description,
|
||
"inputSchema": {
|
||
"type": "object",
|
||
"properties": properties,
|
||
"required": required,
|
||
},
|
||
}
|
||
|
||
|
||
TOOLS = [
|
||
_tool("health", "云端控制面健康检查（不含任务数据）。",
|
||
{}, []),
|
||
_tool("list_projects", "列出当前 key 可见的项目。",
|
||
{}, []),
|
||
_tool("list_tasks", "列出某项目的任务（需 project_id）。",
|
||
{"project_id": {"type": "string"}}, ["project_id"]),
|
||
_tool("get_task", "取单个任务详情（状态、结果、成本字段等）。",
|
||
{"task_id": {"type": "string"}}, ["task_id"]),
|
||
_tool("create_task", "在某项目建一个任务（需 project_id 与 prompt）。",
|
||
{"project_id": {"type": "string"},
|
||
"prompt": {"type": "string"},
|
||
"executor": {"type": "string"},
|
||
"model": {"type": "string"},
|
||
"effort": {"type": "string"},
|
||
"approval": {"type": "string"},
|
||
"priority": {"type": "integer"}},
|
||
["project_id", "prompt"]),
|
||
_tool("cancel_task", "取消一个任务（标 interrupted，容器真杀由 worker 处理）。",
|
||
{"task_id": {"type": "string"}}, ["task_id"]),
|
||
_tool("get_task_cost", "取单个任务的成本明细。",
|
||
{"task_id": {"type": "string"}}, ["task_id"]),
|
||
_tool("get_task_trace", "取单个任务的链路条目（摘要，完整结果在任务详情）。",
|
||
{"task_id": {"type": "string"}}, ["task_id"]),
|
||
]
|
||
|
||
|
||
class MCPServer:
|
||
"""无状态的 REST→MCP 翻译层；transport 可注入，单测用假 transport。"""
|
||
|
||
def __init__(self, base_url, api_key, transport=None):
|
||
if not api_key:
|
||
raise ValueError("CODEX_CLOUD_API_KEY 未设置")
|
||
self.base_url = base_url.rstrip("/")
|
||
self.api_key = api_key
|
||
self.transport = transport or self._http
|
||
|
||
def _http(self, method, path, body=None):
|
||
url = self.base_url + path
|
||
data = json.dumps(body).encode("utf-8") if body is not None else None
|
||
req = urllib.request.Request(url, data=data, method=method)
|
||
req.add_header("X-API-Key", self.api_key)
|
||
if data is not None:
|
||
req.add_header("Content-Type", "application/json")
|
||
try:
|
||
with urllib.request.urlopen(req, timeout=60) as resp:
|
||
raw = resp.read().decode("utf-8") or "{}"
|
||
return resp.status, json.loads(raw)
|
||
except urllib.error.HTTPError as exc:
|
||
raw = exc.read().decode("utf-8") or "{}"
|
||
try:
|
||
payload = json.loads(raw)
|
||
except ValueError:
|
||
payload = {"raw": raw[:500]}
|
||
return exc.code, payload
|
||
except Exception as exc: # noqa: BLE001 - 网络层一律转错误结果
|
||
return 0, {"error": str(exc)[:500]}
|
||
|
||
def call_tool(self, name, args):
|
||
"""返回 (is_error, payload_dict)。未知工具、缺参一律 is_error。"""
|
||
args = args or {}
|
||
if name == "health":
|
||
return self._wrap(*self.transport("GET", "/health"))
|
||
if name == "list_projects":
|
||
return self._wrap(*self.transport("GET", "/projects"))
|
||
if name == "list_tasks":
|
||
if not args.get("project_id"):
|
||
return True, {"error": "project_id required"}
|
||
q = urllib.parse.urlencode({"project_id": args["project_id"]})
|
||
return self._wrap(*self.transport("GET", "/tasks?" + q))
|
||
if name == "get_task":
|
||
if not args.get("task_id"):
|
||
return True, {"error": "task_id required"}
|
||
return self._wrap(*self.transport(
|
||
"GET", "/tasks/" + args["task_id"]))
|
||
if name == "create_task":
|
||
if not args.get("project_id") or not args.get("prompt"):
|
||
return True, {"error": "project_id and prompt required"}
|
||
body = {"project_id": args["project_id"],
|
||
"prompt": args["prompt"]}
|
||
for key in ("executor", "model", "effort", "approval",
|
||
"priority"):
|
||
if args.get(key) is not None:
|
||
body[key] = args[key]
|
||
return self._wrap(*self.transport("POST", "/tasks", body))
|
||
if name == "cancel_task":
|
||
if not args.get("task_id"):
|
||
return True, {"error": "task_id required"}
|
||
return self._wrap(*self.transport(
|
||
"POST", "/tasks/" + args["task_id"] + "/cancel", {}))
|
||
if name == "get_task_cost":
|
||
if not args.get("task_id"):
|
||
return True, {"error": "task_id required"}
|
||
return self._wrap(*self.transport(
|
||
"GET", "/tasks/" + args["task_id"] + "/cost"))
|
||
if name == "get_task_trace":
|
||
if not args.get("task_id"):
|
||
return True, {"error": "task_id required"}
|
||
return self._wrap(*self.transport(
|
||
"GET", "/tasks/" + args["task_id"] + "/trace"))
|
||
return True, {"error": "unknown tool: %s" % name}
|
||
|
||
@staticmethod
|
||
def _wrap(status, payload):
|
||
if status == 0 or status >= 400:
|
||
return True, {"http_status": status, "error": payload}
|
||
return False, payload
|
||
|
||
def handle(self, request):
|
||
"""处理一个 JSON-RPC 请求，返回响应 dict 或 None（notification）。"""
|
||
if not isinstance(request, dict):
|
||
return {"jsonrpc": "2.0", "id": None,
|
||
"error": {"code": -32600, "message": "invalid request"}}
|
||
method = request.get("method", "")
|
||
req_id = request.get("id")
|
||
if method.startswith("notifications/"):
|
||
return None
|
||
if method == "initialize":
|
||
return {"jsonrpc": "2.0", "id": req_id, "result": {
|
||
"protocolVersion": request.get("params", {}).get(
|
||
"protocolVersion", "2024-11-05"),
|
||
"capabilities": {"tools": {}},
|
||
"serverInfo": {"name": "codex-cloud-mcp",
|
||
"version": "0.1.0"}}}
|
||
if method == "ping":
|
||
return {"jsonrpc": "2.0", "id": req_id, "result": {}}
|
||
if method == "tools/list":
|
||
return {"jsonrpc": "2.0", "id": req_id,
|
||
"result": {"tools": TOOLS}}
|
||
if method == "tools/call":
|
||
params = request.get("params", {}) or {}
|
||
is_error, payload = self.call_tool(
|
||
params.get("name"), params.get("arguments") or {})
|
||
return {"jsonrpc": "2.0", "id": req_id, "result": {
|
||
"content": [{"type": "text",
|
||
"text": json.dumps(payload,
|
||
ensure_ascii=False)}],
|
||
"isError": is_error}}
|
||
return {"jsonrpc": "2.0", "id": req_id,
|
||
"error": {"code": -32601,
|
||
"message": "method not found: %s" % method}}
|
||
|
||
def serve(self, instream, outstream):
|
||
"""stdio 主循环：每行一个请求，EOF 结束；坏行回 -32700 不炸循环。"""
|
||
for line in instream:
|
||
line = line.strip()
|
||
if not line:
|
||
continue
|
||
try:
|
||
request = json.loads(line)
|
||
except ValueError:
|
||
outstream.write(json.dumps(
|
||
{"jsonrpc": "2.0", "id": None,
|
||
"error": {"code": -32700,
|
||
"message": "parse error"}}) + "\n")
|
||
outstream.flush()
|
||
continue
|
||
response = self.handle(request)
|
||
if response is not None:
|
||
outstream.write(json.dumps(response,
|
||
ensure_ascii=False) + "\n")
|
||
outstream.flush()
|
||
|
||
|
||
def main(argv=None):
|
||
argv = list(sys.argv[1:] if argv is None else argv)
|
||
base_url = os.environ.get("CODEX_CLOUD_BASE_URL",
|
||
"https://api.diante.me")
|
||
api_key = os.environ.get("CODEX_CLOUD_API_KEY", "")
|
||
try:
|
||
server = MCPServer(base_url, api_key)
|
||
except ValueError as exc:
|
||
sys.stderr.write("mcp_server: %s\n" % exc)
|
||
return 2
|
||
server.serve(sys.stdin, sys.stdout)
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|