From 0e29ce28cf862e5f5ddd99363506058f6313fcfb Mon Sep 17 00:00:00 2001 From: CaptainB Date: Thu, 27 Mar 2025 10:05:37 +0800 Subject: [PATCH] chore: remove unused MCP response function from base_chat_node.py --- .../ai_chat_step_node/impl/base_chat_node.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py index ae84c4bbb..f8927bf0a 100644 --- a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py +++ b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py @@ -119,17 +119,6 @@ def mcp_response_generator(chat_model, message_list, mcp_servers): async def anext_async(agen): return await agen.__anext__() -async def _get_mcp_response(chat_model, message_list, mcp_servers): - async with MultiServerMCPClient(json.loads(mcp_servers)) as client: - agent = create_react_agent(chat_model, client.get_tools()) - response = agent.astream({"messages": message_list}, stream_mode='messages') - result = [] - async for chunk in response: - # if isinstance(chunk[0], ToolMessage): - # print(chunk[0].content) - if isinstance(chunk[0], AIMessageChunk): - result.append(chunk[0]) - return result def write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow): """