Skip to content

Commit b6e27c2

Browse files
committed
feat: include skill tool IDs in node processing logic
1 parent c0dae45 commit b6e27c2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

apps/application/flow/tools.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -560,7 +560,8 @@ async def anext_async(agen):
560560
target_source_node_mapping = {
561561
'TOOL': {'tool-lib-node': lambda n: [n.get('properties').get('node_data').get('tool_lib_id')],
562562
'ai-chat-node': lambda n: [*(n.get('properties').get('node_data').get('mcp_tool_ids') or []),
563-
*(n.get('properties').get('node_data').get('tool_ids') or [])],
563+
*(n.get('properties').get('node_data').get('tool_ids') or []),
564+
*(n.get('properties').get('node_data').get('skill_tool_ids') or [])],
564565
'mcp-node': lambda n: [n.get('properties').get('node_data').get('mcp_tool_id')]
565566
},
566567
'MODEL': {'ai-chat-node': lambda n: [n.get('properties').get('node_data').get('model_id')],
@@ -672,8 +673,9 @@ def get_tool_id_list(workflow):
672673
elif node.get('type') == 'ai-chat-node':
673674
node_data = node.get('properties', {}).get('node_data', {})
674675
mcp_tool_ids = node_data.get('mcp_tool_ids') or []
676+
skill_tool_ids = node_data.get('skill_tool_ids') or []
675677
tool_ids = node_data.get('tool_ids') or []
676-
for _id in mcp_tool_ids + tool_ids:
678+
for _id in mcp_tool_ids + tool_ids + skill_tool_ids:
677679
_result.append(_id)
678680
elif node.get('type') == 'mcp-node':
679681
mcp_tool_id = node.get('properties', {}).get('node_data', {}).get('mcp_tool_id')

0 commit comments

Comments
 (0)