mirror of
https://github.com/dataelement/bisheng.git
synced 2026-09-24 23:19:52 +08:00
fix: tmp file not found raise exception
This commit is contained in:
@@ -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'
|
||||
|
||||
Reference in New Issue
Block a user