diff --git a/src/models/chat_model.py b/src/models/chat_model.py
index fe5c9c82..600cfe06 100644
--- a/src/models/chat_model.py
+++ b/src/models/chat_model.py
@@ -37,7 +37,7 @@ class OpenAIBase():
             stream=False,
         )
         return response.choices[0].message
-    
+
     def get_models(self):
         try:
             return self.client.models.list()
@@ -58,7 +58,7 @@ class OpenModel(OpenAIBase):
 class CustomModel(OpenAIBase):
     def __init__(self, model_info):
         model_name = model_info["name"]
-        api_key = model_info.get("api_key", "custom_model")
+        api_key = model_info.get("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}")