d01c4d3d06
- 新增 MCP client 配置加载，支持 CLI/chat 通过配置文件接入 MCP - 完善 chat 交互命令，支持参数查看、事件查看、checkpoint 列表与加载 - 增加 LLM action 后诊断能力，支持真实 LLM 和本地规则兜底 - 将 chat 人工确认点接入 LangGraph interrupt/checkpointer - 更新 README、流程图、待办文档和打包说明 - 补充相关单元测试
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
from pam_deploy_graph.mcp_client import (
|
|
FunctionMcpToolClient,
|
|
load_mcp_client_config,
|
|
SessionMcpToolClient,
|
|
StdioMcpToolClient,
|
|
normalize_mcp_sdk_result,
|
|
)
|
|
from pam_deploy_graph.mcp_factory import build_mcp_runner_from_config
|
|
|
|
|
|
def test_function_mcp_client_wraps_callable():
|
|
client = FunctionMcpToolClient(lambda name, args: {"tool": name, "args": args})
|
|
assert client.call_tool("pam_get_online_ips", {"airportCode": "HET"})["tool"] == "pam_get_online_ips"
|
|
|
|
|
|
def test_normalize_mcp_sdk_result_structured_content():
|
|
result = type("Result", (), {"structuredContent": {"ok": True}})()
|
|
assert normalize_mcp_sdk_result(result) == {"ok": True}
|
|
|
|
|
|
def test_session_mcp_client_normalizes_text_json_content():
|
|
content = [type("Text", (), {"text": '{"ok": true}'})()]
|
|
result = type("Result", (), {"content": content})()
|
|
|
|
class Session:
|
|
def call_tool(self, tool_name, arguments):
|
|
return result
|
|
|
|
client = SessionMcpToolClient(Session())
|
|
assert client.call_tool("tool", {}) == {"ok": True}
|
|
|
|
|
|
def test_load_mcp_client_config(tmp_path):
|
|
path = tmp_path / "mcp.json"
|
|
path.write_text(
|
|
(
|
|
'{"server_name": "pam-node-prod", "transport": "stdio", '
|
|
'"command": "python", "args": ["-m", "server"], '
|
|
'"env": {"PAM_ENV": "test"}, "cwd": "/tmp", "timeout_seconds": 3, '
|
|
'"tool_names": {"get-online-ips": "custom_ips"}}'
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
config = load_mcp_client_config(path)
|
|
|
|
assert config.server_name == "pam-node-prod"
|
|
assert config.transport == "stdio"
|
|
assert config.command == "python"
|
|
assert config.args == ["-m", "server"]
|
|
assert config.env == {"PAM_ENV": "test"}
|
|
assert config.cwd == "/tmp"
|
|
assert config.timeout_seconds == 3
|
|
assert config.tool_names["get-online-ips"] == "custom_ips"
|
|
|
|
|
|
def test_build_mcp_runner_from_stdio_config(tmp_path):
|
|
path = tmp_path / "mcp.json"
|
|
path.write_text(
|
|
'{"transport": "stdio", "command": "python", "tool_names": {"verify-ip": "custom_verify"}}',
|
|
encoding="utf-8",
|
|
)
|
|
|
|
runner = build_mcp_runner_from_config(path)
|
|
|
|
assert isinstance(runner.client, StdioMcpToolClient)
|
|
assert runner.tool_names["verify-ip"] == "custom_verify"
|