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