Skip to content

Commit 1b08643

Browse files
committed
fix: enhance embedding model retrieval with default parameters
1 parent ed19db0 commit 1b08643

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
from common.utils.common import get_file_content
2323
from knowledge.models import Paragraph, Knowledge
2424
from knowledge.models import SearchMode
25+
from knowledge.serializers.common import get_embedding_model_default_params
2526
from maxkb.conf import PROJECT_DIR
2627
from models_provider.models import Model
2728
from models_provider.tools import get_model, get_model_by_id
@@ -64,7 +65,8 @@ def execute(self, problem_text: str, knowledge_id_list: list[str], exclude_docum
6465
if model.model_type != "EMBEDDING":
6566
raise Exception(_("Model does not exist"))
6667
self.context['model_name'] = model.name
67-
embedding_model = ModelManage.get_model(model_id, lambda _id: get_model(model))
68+
default_params = get_embedding_model_default_params(model)
69+
embedding_model = ModelManage.get_model(model_id, lambda _id: get_model(model, **{**default_params}))
6870
embedding_value = embedding_model.embed_query(exec_problem_text)
6971
vector = VectorStore.get_embedding_vector()
7072
embedding_list = vector.query(exec_problem_text, embedding_value, knowledge_id_list, None, exclude_document_id_list,

0 commit comments

Comments
 (0)