Fix the issue where the original model name cannot be requested.

.github/workflows/main.yml

@@ -5,15 +5,16 @@ on:
     branches:
       - main
     paths:
-      - Dockerfile
-      - .dockerignore
-      - requirements.txt
       - main.py
+      - utils.py
       - models.py
       - request.py
+      - Dockerfile
       - response.py
-      - .github/workflows/main.yml
+      - .dockerignore
+      - requirements.txt
       - docker-compose.yml
+      - .github/workflows/main.yml
 
 jobs:
   build-and-push:

main.py

@@ -92,12 +92,12 @@ class ModelRequestHandler:
                 model = model.split("/")[1]
                 for provider in config['providers']:
                     if provider['provider'] == provider_name:
-                        models_list = provider['model'].keys()
-                if (model and model_name == model) or (model == "*" and model_name in models_list):
+                        models_list = list(provider['model'].keys()) + list(provider['model'].values())
+                if (model and model_name in models_list) or (model == "*" and model_name in models_list):
                     provider_rules.append(provider_name)
         provider_list = []
         for provider in config['providers']:
-            if model_name in provider['model'].keys() and ((provider_rules and provider['provider'] in provider_rules) or provider_rules == []):
+            if model_name in list(provider['model'].keys()) + list(provider['model'].values()) and ((provider_rules and provider['provider'] in provider_rules) or provider_rules == []):
                 provider_list.append(provider)
         return provider_list
 

utils.py

@@ -14,7 +14,8 @@ def load_config():
                     if type(model) == str:
                         model_dict[model] = model
                     if type(model) == dict:
-                        model_dict.update({value: key for key, value in model.items()})
+                        model_dict.update({new: old for old, new in model.items()})
+                        model_dict.update({old: old for old, new in model.items()})
                 provider['model'] = model_dict
                 conf['providers'][index] = provider
             api_keys_db = conf['api_keys']