fix: MCP calling node error #3752 (#3762)

This commit is contained in:
shaohuzhang1 2025-07-29 10:48:30 +08:00 committed by GitHub
parent 9cc051ded4
commit d33ae01d5a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,7 +4,6 @@ import json
from typing import List
from langchain_mcp_adapters.client import MultiServerMCPClient
from langchain_mcp_adapters.sessions import create_session
from application.flow.i_step_node import NodeResult
from application.flow.step_node.mcp_node.i_mcp_node import IMcpNode
@ -25,7 +24,7 @@ class BaseMcpNode(IMcpNode):
async def call_tool(t, a):
client = MultiServerMCPClient(servers)
async with create_session(client.connections[mcp_server]) as s:
async with client.session(mcp_server) as s:
return await s.call_tool(t, a)
res = asyncio.run(call_tool(mcp_tool, params))