feat: add scope field to knowledge creation serializers and update related views

This commit is contained in:
CaptainB 2025-06-06 15:13:17 +08:00
parent a2cc83ec31
commit 643200cc06
2 changed files with 4 additions and 3 deletions

View File

@ -405,6 +405,7 @@ class KnowledgeSerializer(serializers.Serializer):
class Create(serializers.Serializer):
user_id = serializers.UUIDField(required=True, label=_('user id'))
workspace_id = serializers.CharField(required=True, label=_('workspace id'))
scope = serializers.ChoiceField(required=False, label=_('scope'), default=KnowledgeScope.WORKSPACE, choices=KnowledgeScope.choices)
@staticmethod
def post_embedding_knowledge(document_list, knowledge_id):
@ -433,7 +434,7 @@ class KnowledgeSerializer(serializers.Serializer):
desc=instance.get('desc'),
type=instance.get('type', KnowledgeType.BASE),
user_id=self.data.get('user_id'),
scope=KnowledgeScope.WORKSPACE,
scope=self.data.get('scope', KnowledgeScope.WORKSPACE),
folder_id=instance.get('folder_id', 'root'),
embedding_model_id=instance.get('embedding'),
meta=instance.get('meta', {}),
@ -492,7 +493,7 @@ class KnowledgeSerializer(serializers.Serializer):
desc=instance.get('desc'),
user_id=self.data.get('user_id'),
type=instance.get('type', KnowledgeType.WEB),
scope=KnowledgeScope.WORKSPACE,
scope=self.data.get('scope', KnowledgeScope.WORKSPACE),
folder_id=instance.get('folder_id', 'root'),
embedding_model_id=instance.get('embedding'),
meta={

View File

@ -334,7 +334,7 @@ class KnowledgeWebView(APIView):
'first_list': r.FILES.getlist('file'),
'meta': {'source_url': r.data.get('source_url'),
'selector': r.data.get('selector'),
'embedding_mode_id': r.data.get('embedding_mode_id')}}
'embedding_model_id': r.data.get('embedding_model_id')}}
)
def post(self, request: Request, workspace_id: str):
return result.success(KnowledgeSerializer.Create(