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

修复 ollama 连接的问题
2025-03-13 20:51:35 +08:00 · 2025-03-13 20:51:35 +08:00 · cba8a05703
commit cba8a05703
parent 1c065398b3 97d9f5c7c8
1 changed files with 1 additions and 1 deletions
--- a/src/models/chat_model.py
+++ b/src/models/chat_model.py
@ -58,7 +58,7 @@ class OpenModel(OpenAIBase):
 class CustomModel(OpenAIBase):
    def __init__(self, model_info):
        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"])
        logger.info(f"> Custom model: {model_name}, base_url: {base_url}")