Skip to content

Commit 643200c

Browse files
committed
feat: add scope field to knowledge creation serializers and update related views
1 parent a2cc83e commit 643200c

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

apps/knowledge/serializers/knowledge.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,7 @@ def merge_problem(paragraph_list: List[Dict], problem_mapping_list: List[Dict]):
405405
class Create(serializers.Serializer):
406406
user_id = serializers.UUIDField(required=True, label=_('user id'))
407407
workspace_id = serializers.CharField(required=True, label=_('workspace id'))
408+
scope = serializers.ChoiceField(required=False, label=_('scope'), default=KnowledgeScope.WORKSPACE, choices=KnowledgeScope.choices)
408409

409410
@staticmethod
410411
def post_embedding_knowledge(document_list, knowledge_id):
@@ -433,7 +434,7 @@ def save_base(self, instance, with_valid=True):
433434
desc=instance.get('desc'),
434435
type=instance.get('type', KnowledgeType.BASE),
435436
user_id=self.data.get('user_id'),
436-
scope=KnowledgeScope.WORKSPACE,
437+
scope=self.data.get('scope', KnowledgeScope.WORKSPACE),
437438
folder_id=instance.get('folder_id', 'root'),
438439
embedding_model_id=instance.get('embedding'),
439440
meta=instance.get('meta', {}),
@@ -492,7 +493,7 @@ def save_web(self, instance: Dict, with_valid=True):
492493
desc=instance.get('desc'),
493494
user_id=self.data.get('user_id'),
494495
type=instance.get('type', KnowledgeType.WEB),
495-
scope=KnowledgeScope.WORKSPACE,
496+
scope=self.data.get('scope', KnowledgeScope.WORKSPACE),
496497
folder_id=instance.get('folder_id', 'root'),
497498
embedding_model_id=instance.get('embedding'),
498499
meta={

apps/knowledge/views/knowledge.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,7 @@ class KnowledgeWebView(APIView):
334334
'first_list': r.FILES.getlist('file'),
335335
'meta': {'source_url': r.data.get('source_url'),
336336
'selector': r.data.get('selector'),
337-
'embedding_mode_id': r.data.get('embedding_mode_id')}}
337+
'embedding_model_id': r.data.get('embedding_model_id')}}
338338
)
339339
def post(self, request: Request, workspace_id: str):
340340
return result.success(KnowledgeSerializer.Create(

0 commit comments

Comments
 (0)