kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #72 from xerrors/49-bug-ollama-connection-error

Browse Source 
修复 ollama 连接的问题
This commit is contained in:
Wenjie Zhang 2025-03-13 20:51:35 +08:00 committed by GitHub
commit cba8a05703
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
src/models/chat_model.py
View File

@ -58,7 +58,7 @@ class OpenModel(OpenAIBase):
class CustomModel(OpenAIBase): class CustomModel(OpenAIBase):
def __init__(self, model_info): def __init__(self, model_info):
model_name = model_info["name"] model_name = model_info["name"]
api_key = model_info["api_key"] api_key = model_info["api_key"] or "custom_model"
base_url = get_docker_safe_url(model_info["api_base"]) base_url = get_docker_safe_url(model_info["api_base"])
logger.info(f"> Custom model: {model_name}, base_url: {base_url}") logger.info(f"> Custom model: {model_name}, base_url: {base_url}")