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

feat: ollama 平台api域名之需要填写到端口

Browse Source
This commit is contained in:
zhangshaohu 2024-03-22 23:44:53 +08:00
parent 8ed2754eba
commit 31a441c1c0
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/setting/models_provider/impl/ollama_model_provider/ollama_model_provider.py
View File

@ -148,7 +148,9 @@ class OllamaModelProvider(IModelProvider):
return ollama_llm_model_credential return ollama_llm_model_credential
def get_model(self, model_type, model_name, model_credential: Dict[str, object], **model_kwargs) -> BaseChatModel: def get_model(self, model_type, model_name, model_credential: Dict[str, object], **model_kwargs) -> BaseChatModel:
return OllamaChatModel(model=model_name, openai_api_base=model_credential.get('api_base'), api_base = model_credential.get('api_base')
base_url = get_base_url(api_base)
return OllamaChatModel(model=model_name, openai_api_base=(base_url + '/v1'),
openai_api_key=model_credential.get('api_key')) openai_api_key=model_credential.get('api_key'))
def get_dialogue_number(self): def get_dialogue_number(self):