panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: enhance embedding model retrieval with default parameters

v3.2
CaptainB 2025-10-30 17:05:23 +08:00
parent ed19db07d1
commit 1b08643b98
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py
View File

@ -22,6 +22,7 @@ from common.db.search import native_search
from common.utils.common import get_file_content from common.utils.common import get_file_content
from knowledge.models import Paragraph, Knowledge from knowledge.models import Paragraph, Knowledge
from knowledge.models import SearchMode from knowledge.models import SearchMode
from knowledge.serializers.common import get_embedding_model_default_params
from maxkb.conf import PROJECT_DIR from maxkb.conf import PROJECT_DIR
from models_provider.models import Model from models_provider.models import Model
from models_provider.tools import get_model, get_model_by_id from models_provider.tools import get_model, get_model_by_id
@ -64,7 +65,8 @@ class BaseSearchDatasetStep(ISearchDatasetStep):
if model.model_type != "EMBEDDING": if model.model_type != "EMBEDDING":
raise Exception(_("Model does not exist")) raise Exception(_("Model does not exist"))
self.context['model_name'] = model.name 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) embedding_value = embedding_model.embed_query(exec_problem_text)
vector = VectorStore.get_embedding_vector() vector = VectorStore.get_embedding_vector()
embedding_list = vector.query(exec_problem_text, embedding_value, knowledge_id_list, None, exclude_document_id_list, embedding_list = vector.query(exec_problem_text, embedding_value, knowledge_id_list, None, exclude_document_id_list,