Skip to content

Commit f2f05f5

Browse files
committed
fix: handle original document processing and source file creation in knowledge.py
1 parent 06c132d commit f2f05f5

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

apps/knowledge/serializers/knowledge.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -891,6 +891,18 @@ def import_knowledge(self, file, is_import_tool=False, with_valid=True):
891891
directly_return_similarity=float(similarity) if similarity else 0.9,
892892
meta=doc_meta,
893893
)
894+
# 处理原文档
895+
if "source_file_id" in doc_meta:
896+
old_file = QuerySet(File).filter(id=doc_meta["source_file_id"]).first()
897+
if old_file:
898+
source_file = File(
899+
id=uuid.uuid7(),
900+
file_name=old_file.file_name,
901+
source_type=FileSourceType.DOCUMENT,
902+
source_id=document_id,
903+
meta={},
904+
)
905+
source_file.save(old_file.get_bytes())
894906

895907
document_model_list.append(document)
896908
if tags_str:

0 commit comments

Comments
 (0)