zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: remove unused MCP response function from base_chat_node.py

Browse Source
This commit is contained in:
CaptainB 2025-03-27 10:05:37 +08:00
parent eed4d6857e
commit 0e29ce28cf
1 changed files with 0 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -119,17 +119,6 @@ def mcp_response_generator(chat_model, message_list, mcp_servers):
async def anext_async(agen): async def anext_async(agen):
return await agen.__anext__() 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): def write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow):
""" """