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

fix: MCP node display nesting issue (#3822)

Browse Source
This commit is contained in:
shaohuzhang1 2025-08-05 19:23:48 +08:00 committed by GitHub
parent f61a567bc3
commit ec12ad0c2e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -33,13 +33,25 @@ tool_message_template = """
<strong>Called MCP Tool: <em>%s</em></strong> <strong>Called MCP Tool: <em>%s</em></strong>
</summary> </summary>
```json
%s %s
```
</details> </details>
""" """
tool_message_json_template = """
```json
%s
```
"""
def generate_tool_message_template(name, context):
if '```' in context:
return tool_message_template % (name, context)
else:
return tool_message_template % (name, tool_message_json_template % (context))
def _write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow, answer: str, def _write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow, answer: str,
reasoning_content: str): reasoning_content: str):
@ -110,7 +122,7 @@ async def _yield_mcp_response(chat_model, message_list, mcp_servers):
response = agent.astream({"messages": message_list}, stream_mode='messages') response = agent.astream({"messages": message_list}, stream_mode='messages')
async for chunk in response: async for chunk in response:
if isinstance(chunk[0], ToolMessage): if isinstance(chunk[0], ToolMessage):
content = tool_message_template % (chunk[0].name, chunk[0].content) content = generate_tool_message_template(chunk[0].name, chunk[0].content)
chunk[0].content = content chunk[0].content = content
yield chunk[0] yield chunk[0]
if isinstance(chunk[0], AIMessageChunk): if isinstance(chunk[0], AIMessageChunk):