Add get models endpoint.

main.py

@@ -8,7 +8,7 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI, HTTPException, Depends
 
 from models import RequestModel
-from utils import config, api_keys_db, api_list, error_handling_wrapper, get_all_models, verify_api_key
+from utils import config, api_keys_db, api_list, error_handling_wrapper, get_all_models, verify_api_key, post_all_models
 from request import get_payload
 from response import fetch_response, fetch_response_stream
 
@@ -160,7 +160,15 @@ async def options_handler():
 
 @app.post("/v1/models")
 async def list_models(token: str = Depends(verify_api_key)):
-    models = get_all_models(token)
+    models = post_all_models(token)
+    return {
+        "object": "list",
+        "data": models
+    }
+
+@app.get("/v1/models")
+async def list_models():
+    models = get_all_models()
     return {
         "object": "list",
         "data": models

utils.py

@@ -61,7 +61,7 @@ async def error_handling_wrapper(generator, status_code=200):
         # 处理生成器为空的情况
         return []
 
-def get_all_models(token):
+def post_all_models(token):
     all_models = []
     unique_models = set()
 
@@ -84,7 +84,8 @@ def get_all_models(token):
                                     "id": model_item,
                                     "object": "model",
                                     "created": 1720524448858,
-                                    "owned_by": provider_item['provider']
+                                    "owned_by": "uni-api"
+                                    # "owned_by": provider_item['provider']
                                 }
                                 all_models.append(model_info)
                 else:
@@ -98,7 +99,7 @@ def get_all_models(token):
                                     "id": model_item,
                                     "object": "model",
                                     "created": 1720524448858,
-                                    "owned_by": provider_item['provider']
+                                    "owned_by": "uni-api"
                                 }
                                 all_models.append(model_info)
                 continue
@@ -112,18 +113,24 @@ def get_all_models(token):
                     "owned_by": model
                 }
                 all_models.append(model_info)
-    else:
-        for provider in config["providers"]:
-            for model in provider['model'].keys():
-                if model not in unique_models:
-                    unique_models.add(model)
-                    model_info = {
-                        "id": model,
-                        "object": "model",
-                        "created": 1720524448858,
-                        "owned_by": provider['provider']
-                    }
-                    all_models.append(model_info)
+
+    return all_models
+
+def get_all_models():
+    all_models = []
+    unique_models = set()
+
+    for provider in config["providers"]:
+        for model in provider['model'].keys():
+            if model not in unique_models:
+                unique_models.add(model)
+                model_info = {
+                    "id": model,
+                    "object": "model",
+                    "created": 1720524448858,
+                    "owned_by": "uni-api"
+                }
+                all_models.append(model_info)
 
     return all_models