Skip to content

Commit fbc37ed

Browse files
committed
fix: include workspace_id in batch save for document processing
1 parent f1d9bfa commit fbc37ed

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

apps/knowledge/serializers/document.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -842,17 +842,19 @@ def save_qa(self, instance: Dict, with_valid=True):
842842
self.is_valid(raise_exception=True)
843843
file_list = instance.get('file_list')
844844
document_list = flat_map([self.parse_qa_file(file) for file in file_list])
845-
return DocumentSerializers.Batch(data={'knowledge_id': self.data.get('knowledge_id')}).batch_save(
846-
document_list)
845+
return DocumentSerializers.Batch(data={
846+
'knowledge_id': self.data.get('knowledge_id'), 'workspace_id': self.data.get('workspace_id')
847+
}).batch_save(document_list)
847848

848849
def save_table(self, instance: Dict, with_valid=True):
849850
if with_valid:
850851
DocumentInstanceTableSerializer(data=instance).is_valid(raise_exception=True)
851852
self.is_valid(raise_exception=True)
852853
file_list = instance.get('file_list')
853854
document_list = flat_map([self.parse_table_file(file) for file in file_list])
854-
return DocumentSerializers.Batch(data={'knowledge_id': self.data.get('knowledge_id')}).batch_save(
855-
document_list)
855+
return DocumentSerializers.Batch(data={
856+
'knowledge_id': self.data.get('knowledge_id'), 'workspace_id': self.data.get('workspace_id')
857+
}).batch_save(document_list)
856858

857859
def parse_qa_file(self, file):
858860
get_buffer = FileBufferHandle().get_buffer

0 commit comments

Comments
 (0)