fix get model list

This commit is contained in:
Ishaan Jaff 2024-09-12 17:12:04 -07:00
parent 3490862795
commit de419a9497
2 changed files with 4 additions and 7 deletions

View file

@ -6,11 +6,11 @@ model_list:
vertex_project: "adroit-crow-413218"
vertex_location: "us-central1"
vertex_credentials: "/Users/ishaanjaffer/Downloads/adroit-crow-413218-a956eef1a2a8.json"
- model_name: fake-openai-endpoint
- model_name: fake-azure-endpoint
litellm_params:
model: openai/fake
model: openai/429
api_key: fake-key
api_base: https://exampleopenaiendpoint-production.up.railway.app/
api_base: https://exampleopenaiendpoint-production.up.railway.app
general_settings:
master_key: sk-1234

View file

@ -4679,10 +4679,7 @@ class Router:
returned_models += self.model_list
return returned_models
for model in self.model_list:
returned_models.extend(self._get_all_deployments(model_name=model_name))
returned_models.extend(self._get_all_deployments(model_name=model_name))
return returned_models
return None