zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: include workspace_id in batch save for document processing

Browse Source
This commit is contained in:
CaptainB 2025-06-17 20:20:08 +08:00
parent f1d9bfa648
commit fbc37edf3c
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apps/knowledge/serializers/document.py
View File

@ -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