🐛 修复重复发送消息给llm的问题

This commit is contained in:
FuQuan233 2025-08-20 12:38:39 +08:00
parent 5014d3014b
commit ea635fd147
2 changed files with 3 additions and 5 deletions

View file

@ -1,5 +1,5 @@
from contextlib import AsyncExitStack
import asyncio
from contextlib import AsyncExitStack
from mcp import ClientSession, StdioServerParameters
from mcp.client.sse import sse_client
@ -66,9 +66,7 @@ class MCPClient:
logger.info(f"正在服务器[{server_name}]上调用工具[{real_tool_name}]")
session = self.sessions[server_name]
try:
response = await asyncio.wait_for(
session.call_tool(real_tool_name, tool_args), timeout=10
)
response = await asyncio.wait_for(session.call_tool(real_tool_name, tool_args), timeout=10)
except asyncio.TimeoutError:
logger.error(f"调用工具[{real_tool_name}]超时")
return f"调用工具[{real_tool_name}]超时"