19 lines
867 B
Python
19 lines
867 B
Python
|
from mcp import ClientSession,StdioServerParameters
|
|||
|
|
from mcp.client.stdio import stdio_client
|
|||
|
|
|
|||
|
|
class SimpleClient:
|
|||
|
|
def __init__(self,command:str,args:list[str],env:dict=None):
|
|||
|
|
# 指定要启动的工具和参数
|
|||
|
|
self.params = StdioServerParameters(command=command,args=args,env=env)
|
|||
|
|
|
|||
|
|
async def run_once(self,tool_name:str,tool_args:dict):
|
|||
|
|
# 语法糖: async with 自动帮我们 打开连接 -> 运行 -> 关闭连接
|
|||
|
|
async with stdio_client(self.params) as (read,write):
|
|||
|
|
# 建立父子进程管道（stdin/stdout）
|
|||
|
|
async with ClientSession(read,write) as session:
|
|||
|
|
# 用JSON-RPC与工具对话
|
|||
|
|
await session.initialize()
|
|||
|
|
|
|||
|
|
# 直接调用工具
|
|||
|
|
result = await session.call_tool(tool_name,tool_args)
|
|||
|
|
return result.content[0].text
|