57 lines
2.6 KiB
Python
57 lines
2.6 KiB
Python
|
import asyncio
|
|||
|
|
from langchain_mcp_adapters.tools import load_mcp_tools
|
|||
|
|
from mcp import ClientSession, StdioServerParameters
|
|||
|
|
from mcp.client.stdio import stdio_client
|
|||
|
|
|
|||
|
|
# 配置mcp服务器脚本路径
|
|||
|
|
server_script = r".\stdio_server.py"
|
|||
|
|
|
|||
|
|
# 配置mcp服务器启动参数
|
|||
|
|
server_params = StdioServerParameters(
|
|||
|
|
command="python" if server_script.endswith(".py") else "node",
|
|||
|
|
args=[server_script],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 定义mcp客户端
|
|||
|
|
mcp_client = None
|
|||
|
|
|
|||
|
|
# 主要的异步函数run_agent
|
|||
|
|
async def run():
|
|||
|
|
global mcp_client
|
|||
|
|
# 启动 MCP server，并通过标准输入输出建立异步连接。
|
|||
|
|
async with stdio_client(server_params) as (read, write):
|
|||
|
|
# 使用读写通道创建 MCP 会话。
|
|||
|
|
async with ClientSession(read, write) as session:
|
|||
|
|
await session.initialize()
|
|||
|
|
# 动态创建一个临时类 MCPClientHolder，把 session 放进去。这样就可以在函数外部通过 mcp_client.session 调用 MCP 工具
|
|||
|
|
mcp_client = type("MCPClientHolder", (), {"session": session})()
|
|||
|
|
|
|||
|
|
# 从 session 自动获取 MCP server 提供的工具列表。
|
|||
|
|
tools = await load_mcp_tools(session)
|
|||
|
|
print(f"tools-->{tools}")
|
|||
|
|
|
|||
|
|
# 调用 MCP server 的 get_weather 工具
|
|||
|
|
response = await session.call_tool("get_weather", arguments={})
|
|||
|
|
print(f"response-->{response}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 启动运行
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
asyncio.run(run())
|
|||
|
|
# tools -->[StructuredTool(name='query_high_frequency_question',
|
|||
|
|
# description='从知识库中检索常见问题解答（FAQ）,返回包含问题和答案的结构化JSON数据。',
|
|||
|
|
# args_schema={'properties': {}, 'title': 'query_high_frequency_questionArguments',
|
|||
|
|
# 'type': 'object'}, response_format='content_and_artifact',
|
|||
|
|
# coroutine= < function convert_mcp_tool_to_langchain_tool. < locals >.call_tool
|
|||
|
|
# at
|
|||
|
|
# 0x000001DB7A188FE0 >), StructuredTool(name='get_weather', description='查询天气',
|
|||
|
|
# args_schema={'properties': {}, 'title': 'get_weatherArguments',
|
|||
|
|
# 'type': 'object'}, response_format='content_and_artifact',
|
|||
|
|
# coroutine= < function
|
|||
|
|
# convert_mcp_tool_to_langchain_tool. < locals >.call_tool
|
|||
|
|
# at
|
|||
|
|
# 0x000001DB7A1E3060 >)]
|
|||
|
|
# response -->meta = None
|
|||
|
|
# content = [TextContent(type='text', text='北京的天气是多云', annotations=None, meta=None)]
|
|||
|
|
# structuredContent = {'result': '北京的天气是多云'}
|
|||
|
|
# isError = False
|