diff --git a/apps/application/flow/tools.py b/apps/application/flow/tools.py
index d14e8319ca2..91cb98bf980 100644
--- a/apps/application/flow/tools.py
+++ b/apps/application/flow/tools.py
@@ -815,7 +815,9 @@ async def anext_async(agen):
               'parameter-extraction-node': lambda n: [n.get('properties').get('node_data').get('model_id')],
               'video-understand-node': lambda n: [n.get('properties').get('node_data').get('model_id')],
               },
-    'KNOWLEDGE': {'search-knowledge-node': lambda n: n.get('properties').get('node_data').get('knowledge_id_list')},
+    'KNOWLEDGE': {'search-knowledge-node': lambda n: n.get('properties').get('node_data').get('knowledge_id_list'),
+                  'search-document-node': lambda n: n.get('properties').get('node_data').get('knowledge_id_list')
+                  },
     'APPLICATION': {
         'application-node': lambda n: [n.get('properties').get('node_data').get('application_id')],
         'ai-chat-node': lambda n: [*(n.get('properties').get('node_data').get('application_ids') or [])],