vertex

Paused

App Files Files Community

bibibi12345 commited on May 6, 2025

Commit

13b9958

verified ·

1 Parent(s): e6b1705

Update app/main.py

Browse files

Files changed (1) hide show

app/main.py +8 -17

app/main.py CHANGED Viewed

@@ -1369,15 +1369,6 @@ def create_final_chunk(model: str, response_id: str, candidate_count: int = 1) -
 async def list_models(api_key: str = Depends(get_api_key)):
     # Based on current information for Vertex AI models
     models = [
-        {
-            "id": "gemini-2.5-pro-preview-05-06",
-            "object": "model",
-            "created": int(time.time()),
-            "owned_by": "google",
-            "permission": [],
-            "root": "gemini-2.5-pro-pre-05-06",
-            "parent": None,
-        },
         {
             "id": "gemini-2.5-pro-exp-03-25",
             "object": "model",
@@ -1433,39 +1424,39 @@ async def list_models(api_key: str = Depends(get_api_key)):
             "parent": None,
         },
         {
-            "id": "gemini-2.5-pro-preview-03-25",
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
-            "root": "gemini-2.5-pro-preview-03-25",
             "parent": None,
         },
         {
-            "id": "gemini-2.5-pro-preview-03-25-search",
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
-            "root": "gemini-2.5-pro-preview-03-25",
             "parent": None,
         },
         {
-            "id": "gemini-2.5-pro-preview-03-25-encrypt",
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
-            "root": "gemini-2.5-pro-preview-03-25",
             "parent": None,
         },
         {
-            "id": "gemini-2.5-pro-preview-03-25-auto", # New auto model
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
-            "root": "gemini-2.5-pro-preview-03-25",
             "parent": None,
         },
         {

 async def list_models(api_key: str = Depends(get_api_key)):
     # Based on current information for Vertex AI models
     models = [
         {
             "id": "gemini-2.5-pro-exp-03-25",
             "object": "model",
             "parent": None,
         },
         {
+            "id": "gemini-2.5-pro-preview-05-06",
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
+            "root": "gemini-2.5-pro-preview-05-06",
             "parent": None,
         },
         {
+            "id": "gemini-2.5-pro-preview-05-06-search",
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
+            "root": "gemini-2.5-pro-preview-05-06",
             "parent": None,
         },
         {
+            "id": "gemini-2.5-pro-preview-05-06-encrypt",
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
+            "root": "gemini-2.5-pro-preview-05-06",
             "parent": None,
         },
         {
+            "id": "gemini-2.5-pro-preview-05-06-auto", # New auto model
             "object": "model",
             "created": int(time.time()),
             "owned_by": "google",
             "permission": [],
+            "root": "gemini-2.5-pro-preview-05-06",
             "parent": None,
         },
         {