This commit is contained in:
Timothy Jaeryang Baek 2024-11-11 21:52:39 -08:00
parent 1277579f71
commit 3ba7b6f0e5

View File

@ -288,7 +288,7 @@ async def get_all_models_raw() -> list:
aiohttp_get(f"{url}/models", app.state.config.OPENAI_API_KEYS[idx]) aiohttp_get(f"{url}/models", app.state.config.OPENAI_API_KEYS[idx])
) )
else: else:
api_config = app.state.config.OPENAI_API_CONFIGS[url] api_config = app.state.config.OPENAI_API_CONFIGS.get(url, {})
enabled = api_config.get("enabled", True) enabled = api_config.get("enabled", True)
model_ids = api_config.get("model_ids", []) model_ids = api_config.get("model_ids", [])
@ -322,7 +322,7 @@ async def get_all_models_raw() -> list:
for idx, response in enumerate(responses): for idx, response in enumerate(responses):
if response: if response:
url = app.state.config.OPENAI_API_BASE_URLS[idx] url = app.state.config.OPENAI_API_BASE_URLS[idx]
api_config = app.state.config.OPENAI_API_CONFIGS[url] api_config = app.state.config.OPENAI_API_CONFIGS.get(url, {})
prefix_id = api_config.get("prefix_id", None) prefix_id = api_config.get("prefix_id", None)