15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-27 02:52:48 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

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

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