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

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