zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复缓存模型key错误 (#878)

Browse Source
This commit is contained in:
shaohuzhang1 2024-07-26 09:52:25 +08:00 committed by GitHub
parent 7471076716
commit d6543ce3b0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/dataset/serializers/common_serializers.py
View File

@ -140,14 +140,14 @@ def get_embedding_model_by_dataset_id_list(dataset_id_list: List):
raise Exception("知识库未向量模型不一致") raise Exception("知识库未向量模型不一致")
if len(dataset_list) == 0: if len(dataset_list) == 0:
raise Exception("知识库设置错误,请重新设置知识库") raise Exception("知识库设置错误,请重新设置知识库")
return ModelManage.get_model(str(dataset_list[0].id), return ModelManage.get_model(str(dataset_list[0].embedding_mode_id),
lambda _id: get_model(dataset_list[0].embedding_mode)) lambda _id: get_model(dataset_list[0].embedding_mode))
def get_embedding_model_by_dataset_id(dataset_id: str): def get_embedding_model_by_dataset_id(dataset_id: str):
dataset = QuerySet(DataSet).select_related('embedding_mode').filter(id=dataset_id).first() dataset = QuerySet(DataSet).select_related('embedding_mode').filter(id=dataset_id).first()
return ModelManage.get_model(dataset_id, lambda _id: get_model(dataset.embedding_mode)) return ModelManage.get_model(dataset.embedding_mode_id, lambda _id: get_model(dataset.embedding_mode))
def get_embedding_model_by_dataset(dataset): def get_embedding_model_by_dataset(dataset):
return ModelManage.get_model(str(dataset.id), lambda _id: get_model(dataset.embedding_mode)) return ModelManage.get_model(str(dataset.embedding_mode_id), lambda _id: get_model(dataset.embedding_mode))