Skip to content

Commit 89023d1

Browse files
committed
refactor: update embedding model ID references in knowledge.py and dataset_lark_serializer.py
1 parent 63d8906 commit 89023d1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

apps/knowledge/serializers/knowledge.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -480,7 +480,7 @@ def save_base(self, instance, with_valid=True):
480480
user_id=self.data.get('user_id'),
481481
scope=self.data.get('scope', KnowledgeScope.WORKSPACE),
482482
folder_id=folder_id,
483-
embedding_model_id=instance.get('embedding'),
483+
embedding_model_id=instance.get('embedding_model_id'),
484484
meta=instance.get('meta', {}),
485485
)
486486

@@ -542,11 +542,11 @@ def save_web(self, instance: Dict, with_valid=True):
542542
scope=self.data.get('scope', KnowledgeScope.WORKSPACE),
543543
folder_id=folder_id,
544544
workspace_id=self.data.get('workspace_id'),
545-
embedding_model_id=instance.get('embedding'),
545+
embedding_model_id=instance.get('embedding_model_id'),
546546
meta={
547547
'source_url': instance.get('source_url'),
548548
'selector': instance.get('selector'),
549-
'embedding_model_id': instance.get('embedding')
549+
'embedding_model_id': instance.get('embedding_model_id')
550550
},
551551
)
552552
knowledge.save()

0 commit comments

Comments
 (0)