diff --git a/src/backend/bisheng/workflow/nodes/agent/agent.py b/src/backend/bisheng/workflow/nodes/agent/agent.py
index be1cacf63..795610a77 100644
--- a/src/backend/bisheng/workflow/nodes/agent/agent.py
+++ b/src/backend/bisheng/workflow/nodes/agent/agent.py
@@ -114,6 +114,9 @@ class AgentNode(BaseNode):
                 es_client = self.init_knowledge_es(knowledge_info)
             else:
                 file_metadata = self.graph_state.get_variable_by_str(knowledge_id)
+                if not file_metadata:
+                    raise Exception(f'未找到对应的临时文件数据：{knowledge_id}')
+
                 name = f'knowledge_{knowledge_id.replace(".", "").replace("#", "")}'
                 description = f'file name: {file_metadata.get("source")}'
 
diff --git a/src/backend/bisheng/workflow/nodes/rag/rag.py b/src/backend/bisheng/workflow/nodes/rag/rag.py
index 7048513e6..bf04fc940 100644
--- a/src/backend/bisheng/workflow/nodes/rag/rag.py
+++ b/src/backend/bisheng/workflow/nodes/rag/rag.py
@@ -158,6 +158,8 @@ class RagNode(BaseNode):
             file_ids = []
             for one in self._knowledge_value:
                 file_metadata = self.graph_state.get_variable_by_str(one)
+                if not file_metadata:
+                    raise Exception(f'未找到对应的临时文件数据：{one}')
                 file_ids.append(file_metadata['file_id'])
             self._sort_chunks = len(file_ids) == 1
             node_type = 'Milvus'