From 27b74580a49eae38eb976fb2d1925087faa6a2c0 Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Thu, 23 Oct 2025 16:40:29 +0800
Subject: [PATCH] chore: update document.py to create and associate new source
 file ID in metadata

---
 apps/knowledge/serializers/document.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/apps/knowledge/serializers/document.py b/apps/knowledge/serializers/document.py
index f26d5209b..292dbe81b 100644
--- a/apps/knowledge/serializers/document.py
+++ b/apps/knowledge/serializers/document.py
@@ -1539,13 +1539,25 @@ class DocumentSerializers(serializers.Serializer):
 
             if not source_file:
                 # 不存在手动关联一个文档
+                new_source_file_id = uuid.uuid7()
                 new_source_file = File(
-                    id=uuid.uuid7(),
+                    id=new_source_file_id,
                     file_name=file.name,
                     source_type=FileSourceType.DOCUMENT,
                     source_id=self.data.get('document_id'),
                 )
                 new_source_file.save(file.read())
+                # 更新Document的meta字段
+                QuerySet(Document).filter(id=self.data.get('document_id')).update(
+                    meta=Func(
+                        F("meta"),
+                        Value(["source_file_id"]),
+                        Value(json.dumps(str(new_source_file_id))),
+                        Value(True),  # create_missing = true
+                        function="jsonb_set",
+                        output_field=JSONField(),
+                    )
+                )
             else:
                 # 获取原文件的sha256_hash
                 original_hash = source_file.sha256_hash