diff --git a/apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py b/apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py index ee6c2ff1d..7264b544a 100644 --- a/apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py +++ b/apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py @@ -22,6 +22,7 @@ from common.db.search import native_search from common.utils.common import get_file_content from knowledge.models import Paragraph, Knowledge from knowledge.models import SearchMode +from knowledge.serializers.common import get_embedding_model_default_params from maxkb.conf import PROJECT_DIR from models_provider.models import Model from models_provider.tools import get_model, get_model_by_id @@ -64,7 +65,8 @@ class BaseSearchDatasetStep(ISearchDatasetStep): if model.model_type != "EMBEDDING": raise Exception(_("Model does not exist")) self.context['model_name'] = model.name - embedding_model = ModelManage.get_model(model_id, lambda _id: get_model(model)) + default_params = get_embedding_model_default_params(model) + embedding_model = ModelManage.get_model(model_id, lambda _id: get_model(model, **{**default_params})) embedding_value = embedding_model.embed_query(exec_problem_text) vector = VectorStore.get_embedding_vector() embedding_list = vector.query(exec_problem_text, embedding_value, knowledge_id_list, None, exclude_document_id_list,