fix: prefix issue

This commit is contained in:
Timothy Jaeryang Baek 2024-11-20 06:12:20 -08:00
parent bd1dae2c66
commit d5c65e36c9

View File

@ -313,7 +313,9 @@ async def get_all_models_responses() -> list:
prefix_id = api_config.get("prefix_id", None) prefix_id = api_config.get("prefix_id", None)
if prefix_id: if prefix_id:
for model in response["data"]: for model in (
response if isinstance(response, list) else response.get("data", [])
):
model["id"] = f"{prefix_id}.{model['id']}" model["id"] = f"{prefix_id}.{model['id']}"
log.debug(f"get_all_models:responses() {responses}") log.debug(f"get_all_models:responses() {responses}")