fix api key empty bug
This commit is contained in:
parent
3e51d604e0
commit
374855f99a
@ -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}")
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user