From fbc37edf3cd0960a6f6d5f5df77450617b5c0968 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Tue, 17 Jun 2025 20:20:08 +0800 Subject: [PATCH] fix: include workspace_id in batch save for document processing --- apps/knowledge/serializers/document.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/apps/knowledge/serializers/document.py b/apps/knowledge/serializers/document.py index 9a2cae419..b79e5a308 100644 --- a/apps/knowledge/serializers/document.py +++ b/apps/knowledge/serializers/document.py @@ -842,8 +842,9 @@ class DocumentSerializers(serializers.Serializer): self.is_valid(raise_exception=True) file_list = instance.get('file_list') document_list = flat_map([self.parse_qa_file(file) for file in file_list]) - return DocumentSerializers.Batch(data={'knowledge_id': self.data.get('knowledge_id')}).batch_save( - document_list) + return DocumentSerializers.Batch(data={ + 'knowledge_id': self.data.get('knowledge_id'), 'workspace_id': self.data.get('workspace_id') + }).batch_save(document_list) def save_table(self, instance: Dict, with_valid=True): if with_valid: @@ -851,8 +852,9 @@ class DocumentSerializers(serializers.Serializer): self.is_valid(raise_exception=True) file_list = instance.get('file_list') document_list = flat_map([self.parse_table_file(file) for file in file_list]) - return DocumentSerializers.Batch(data={'knowledge_id': self.data.get('knowledge_id')}).batch_save( - document_list) + return DocumentSerializers.Batch(data={ + 'knowledge_id': self.data.get('knowledge_id'), 'workspace_id': self.data.get('workspace_id') + }).batch_save(document_list) def parse_qa_file(self, file): get_buffer = FileBufferHandle().get_buffer