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

fix: 修复Ollama模型中没有已下载的模型时，在 MaxKB 添加 Ollama模型报错。#437 (#453)

Browse Source
This commit is contained in:
shaohuzhang1 2024-05-15 11:03:46 +08:00 committed by GitHub
parent 84a770e689
commit 052487a22d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/setting/models_provider/impl/ollama_model_provider/ollama_model_provider.py
View File

@ -35,7 +35,7 @@ class OllamaLLMModelCredential(BaseForm, BaseModelCredential):
model_list = OllamaModelProvider.get_base_model_list(model_credential.get('api_base')) model_list = OllamaModelProvider.get_base_model_list(model_credential.get('api_base'))
except Exception as e: except Exception as e:
raise AppApiException(ValidCode.valid_error.value, "API 域名无效") raise AppApiException(ValidCode.valid_error.value, "API 域名无效")
exist = [model for model in model_list.get('models') if exist = [model for model in (model_list.get('models') if model_list.get('models') is not None else []) if
model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name] model.get('model') == model_name or model.get('model').replace(":latest", "") == model_name]
if len(exist) == 0: if len(exist) == 0:
raise AppApiException(ValidCode.model_not_fount, "模型不存在,请先下载模型") raise AppApiException(ValidCode.model_not_fount, "模型不存在,请先下载模型")