fix: tmp file not found raise exception

This commit is contained in:
GuoQing Zhang
2024-12-12 14:49:00 +08:00
parent 8fa77f5026
commit 3e0b7b4737
2 changed files with 5 additions and 0 deletions
@@ -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")}'
@@ -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'