fix api key empty bug

This commit is contained in:
Wenjie Zhang 2025-03-29 19:19:43 +08:00
parent 3e51d604e0
commit 374855f99a

View File

@ -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}")