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

feat: add workspace_id field to Create serializer in paragraph.py

Browse Source
This commit is contained in:
CaptainB 2025-06-09 17:55:58 +08:00
parent f1e19a1ace
commit fff46e5c28
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/knowledge/serializers/paragraph.py
View File

@ -234,6 +234,7 @@ class ParagraphSerializers(serializers.Serializer):
delete_embedding_by_paragraph(paragraph_id) delete_embedding_by_paragraph(paragraph_id)
class Create(serializers.Serializer): class Create(serializers.Serializer):
workspace_id = serializers.CharField(required=True, label='Workspace ID')
knowledge_id = serializers.UUIDField(required=True, label=_('knowledge id')) knowledge_id = serializers.UUIDField(required=True, label=_('knowledge id'))
document_id = serializers.UUIDField(required=True, label=_('document id')) document_id = serializers.UUIDField(required=True, label=_('document id'))
@ -269,7 +270,12 @@ class ParagraphSerializers(serializers.Serializer):
model_id = get_embedding_model_id_by_knowledge_id(knowledge_id) model_id = get_embedding_model_id_by_knowledge_id(knowledge_id)
embedding_by_paragraph(str(paragraph.id), model_id) embedding_by_paragraph(str(paragraph.id), model_id)
return ParagraphSerializers.Operate( return ParagraphSerializers.Operate(
data={'paragraph_id': str(paragraph.id), 'knowledge_id': knowledge_id, 'document_id': document_id} data={
'paragraph_id': str(paragraph.id),
'knowledge_id': knowledge_id,
'document_id': document_id,
'workspace_id': self.data.get('workspace_id')
}
).one(with_valid=True) ).one(with_valid=True)
@staticmethod @staticmethod