File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -434,7 +434,9 @@ async def anext_async(agen):
434434
435435
436436target_source_node_mapping = {
437- 'TOOL' : {'tool-lib-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('tool_lib_id' )]},
437+ 'TOOL' : {'tool-lib-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('tool_lib_id' )],
438+ 'ai-chat-node' : lambda n : [...([n .get ('properties' ).get ('node_data' ).get ('mcp_tool_ids' )] or []),
439+ ...([n .get ('properties' ).get ('node_data' ).get ('tool_ids' )] or [])]},
438440 'MODEL' : {'ai-chat-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('model_id' )],
439441 'question-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('model_id' )],
440442 'speech-to-text-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('stt_model_id' )],
@@ -444,7 +446,7 @@ async def anext_async(agen):
444446 'intent-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('model_id' )],
445447 'image-understand-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('model_id' )],
446448 'parameter-extraction-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('model_id' )],
447- 'video-understand-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('model_id' )]
449+ 'video-understand-node' : lambda n : [n .get ('properties' ).get ('node_data' ).get ('model_id' )],
448450 },
449451 'KNOWLEDGE' : {'search-knowledge-node' : lambda n : n .get ('properties' ).get ('node_data' ).get ('knowledge_id_list' )},
450452 'APPLICATION' : {
You can’t perform that action at this time.
0 commit comments