merge issue, missed arguments

langgenius · Sep 10, 2024 · 3764e28 · 3764e28
1 parent dac8c7f
commit 3764e28
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/api/services/dataset_service.py b/api/services/dataset_service.py
@@ -1057,12 +1057,9 @@ def save_document_without_dataset_id(tenant_id: str, document_data: dict, accoun
         dataset_collection_binding_id = None
         retrieval_model = None
         if document_data["indexing_technique"] == "high_quality":
-            model_manager = ModelManager()
-            embedding_model = model_manager.get_default_model_instance(
-                tenant_id=current_user.current_tenant_id, model_type=ModelType.TEXT_EMBEDDING
-            )
             dataset_collection_binding = DatasetCollectionBindingService.get_dataset_collection_binding(
-                embedding_model.provider, embedding_model.model
+                document_data['embedding_model_provider'],
+                document_data['embedding_model']
             )
             dataset_collection_binding_id = dataset_collection_binding.id
             if document_data.get("retrieval_model"):