From 8e93e1ea2eef1fc7eef638d8770fc4a3b92bae2b Mon Sep 17 00:00:00 2001 From: CaptainB Date: Mon, 16 Jun 2025 16:57:10 +0800 Subject: [PATCH] fix: handle None source_file_id in link_file method --- apps/knowledge/serializers/document.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/apps/knowledge/serializers/document.py b/apps/knowledge/serializers/document.py index 1574a2149..4b1ce4755 100644 --- a/apps/knowledge/serializers/document.py +++ b/apps/knowledge/serializers/document.py @@ -969,6 +969,8 @@ class DocumentSerializers(serializers.Serializer): @staticmethod def link_file(source_file_id, document_id): + if source_file_id is None: + return source_file = QuerySet(File).filter(id=source_file_id).first() if source_file: # 获取原始文件内容 @@ -1016,7 +1018,7 @@ class DocumentSerializers(serializers.Serializer): ) # 保存文档和文件的关系 document_instance = document_paragraph_dict_model.get('document') - self.link_file(document['source_file_id'], document_instance.id) + self.link_file(document.get('source_file_id'), document_instance.id) document_model_list.append(document_instance) for paragraph in document_paragraph_dict_model.get('paragraph_model_list'): paragraph_model_list.append(paragraph)