Skip to content

Commit fff46e5

Browse files
committed
feat: add workspace_id field to Create serializer in paragraph.py
1 parent f1e19a1 commit fff46e5

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

apps/knowledge/serializers/paragraph.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,7 @@ def delete(self, with_valid=False):
234234
delete_embedding_by_paragraph(paragraph_id)
235235

236236
class Create(serializers.Serializer):
237+
workspace_id = serializers.CharField(required=True, label='Workspace ID')
237238
knowledge_id = serializers.UUIDField(required=True, label=_('knowledge id'))
238239
document_id = serializers.UUIDField(required=True, label=_('document id'))
239240

@@ -269,7 +270,12 @@ def save(self, instance: Dict, with_valid=True, with_embedding=True):
269270
model_id = get_embedding_model_id_by_knowledge_id(knowledge_id)
270271
embedding_by_paragraph(str(paragraph.id), model_id)
271272
return ParagraphSerializers.Operate(
272-
data={'paragraph_id': str(paragraph.id), 'knowledge_id': knowledge_id, 'document_id': document_id}
273+
data={
274+
'paragraph_id': str(paragraph.id),
275+
'knowledge_id': knowledge_id,
276+
'document_id': document_id,
277+
'workspace_id': self.data.get('workspace_id')
278+
}
273279
).one(with_valid=True)
274280

275281
@staticmethod

0 commit comments

Comments
 (0)