Skip to content

Commit d33ae01

Browse files
authored
fix: MCP calling node error #3752 (#3762)
1 parent 9cc051d commit d33ae01

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

apps/application/flow/step_node/mcp_node/impl/base_mcp_node.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
from typing import List
55

66
from langchain_mcp_adapters.client import MultiServerMCPClient
7-
from langchain_mcp_adapters.sessions import create_session
87

98
from application.flow.i_step_node import NodeResult
109
from application.flow.step_node.mcp_node.i_mcp_node import IMcpNode
@@ -25,7 +24,7 @@ def execute(self, mcp_servers, mcp_server, mcp_tool, tool_params, **kwargs) -> N
2524

2625
async def call_tool(t, a):
2726
client = MultiServerMCPClient(servers)
28-
async with create_session(client.connections[mcp_server]) as s:
27+
async with client.session(mcp_server) as s:
2928
return await s.call_tool(t, a)
3029

3130
res = asyncio.run(call_tool(mcp_tool, params))

0 commit comments

Comments
 (0)