🐛 Bug: Fix the bug when the model list is empty.

yym68686 · Sep 6, 2024 · 7477ff7 · 7477ff7
1 parent f3ea9b1
commit 7477ff7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -260,8 +260,9 @@ def get_matching_providers(self, model_name, token):
         config = app.state.config
         # api_keys_db = app.state.api_keys_db
         api_list = app.state.api_list
-
         api_index = api_list.index(token)
+        if not safe_get(config, 'api_keys', api_index, 'model'):
+            raise HTTPException(status_code=404, detail="No matching model found")
         provider_rules = []
 
         for model in config['api_keys'][api_index]['model']: