Skip to content

Commit 15339be

Browse files
committed
fix: Handling MCP node resource association relationships
1 parent 509c5ee commit 15339be

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

apps/application/flow/tools.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -437,7 +437,8 @@ async def anext_async(agen):
437437
target_source_node_mapping = {
438438
'TOOL': {'tool-lib-node': lambda n: [n.get('properties').get('node_data').get('tool_lib_id')],
439439
'ai-chat-node': lambda n: [*(n.get('properties').get('node_data').get('mcp_tool_ids') or []),
440-
*(n.get('properties').get('node_data').get('tool_ids') or [])]
440+
*(n.get('properties').get('node_data').get('tool_ids') or [])],
441+
'mcp-node': lambda n: [n.get('properties').get('node_data').get('mcp_tool_id')]
441442
},
442443
'MODEL': {'ai-chat-node': lambda n: [n.get('properties').get('node_data').get('model_id')],
443444
'question-node': lambda n: [n.get('properties').get('node_data').get('model_id')],

0 commit comments

Comments
 (0)