From fff46e5c285caaf681c151e8369f41033ba89a04 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Mon, 9 Jun 2025 17:55:58 +0800 Subject: [PATCH] feat: add workspace_id field to Create serializer in paragraph.py --- apps/knowledge/serializers/paragraph.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/apps/knowledge/serializers/paragraph.py b/apps/knowledge/serializers/paragraph.py index fab9bdcfd..47ad1c140 100644 --- a/apps/knowledge/serializers/paragraph.py +++ b/apps/knowledge/serializers/paragraph.py @@ -234,6 +234,7 @@ class ParagraphSerializers(serializers.Serializer): delete_embedding_by_paragraph(paragraph_id) class Create(serializers.Serializer): + workspace_id = serializers.CharField(required=True, label='Workspace ID') knowledge_id = serializers.UUIDField(required=True, label=_('knowledge 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) embedding_by_paragraph(str(paragraph.id), model_id) 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) @staticmethod