Update app/main.py
Browse files- app/main.py +8 -17
app/main.py
CHANGED
|
@@ -1369,15 +1369,6 @@ def create_final_chunk(model: str, response_id: str, candidate_count: int = 1) -
|
|
| 1369 |
async def list_models(api_key: str = Depends(get_api_key)):
|
| 1370 |
# Based on current information for Vertex AI models
|
| 1371 |
models = [
|
| 1372 |
-
{
|
| 1373 |
-
"id": "gemini-2.5-pro-preview-05-06",
|
| 1374 |
-
"object": "model",
|
| 1375 |
-
"created": int(time.time()),
|
| 1376 |
-
"owned_by": "google",
|
| 1377 |
-
"permission": [],
|
| 1378 |
-
"root": "gemini-2.5-pro-pre-05-06",
|
| 1379 |
-
"parent": None,
|
| 1380 |
-
},
|
| 1381 |
{
|
| 1382 |
"id": "gemini-2.5-pro-exp-03-25",
|
| 1383 |
"object": "model",
|
|
@@ -1433,39 +1424,39 @@ async def list_models(api_key: str = Depends(get_api_key)):
|
|
| 1433 |
"parent": None,
|
| 1434 |
},
|
| 1435 |
{
|
| 1436 |
-
"id": "gemini-2.5-pro-preview-
|
| 1437 |
"object": "model",
|
| 1438 |
"created": int(time.time()),
|
| 1439 |
"owned_by": "google",
|
| 1440 |
"permission": [],
|
| 1441 |
-
"root": "gemini-2.5-pro-preview-
|
| 1442 |
"parent": None,
|
| 1443 |
},
|
| 1444 |
{
|
| 1445 |
-
"id": "gemini-2.5-pro-preview-
|
| 1446 |
"object": "model",
|
| 1447 |
"created": int(time.time()),
|
| 1448 |
"owned_by": "google",
|
| 1449 |
"permission": [],
|
| 1450 |
-
"root": "gemini-2.5-pro-preview-
|
| 1451 |
"parent": None,
|
| 1452 |
},
|
| 1453 |
{
|
| 1454 |
-
"id": "gemini-2.5-pro-preview-
|
| 1455 |
"object": "model",
|
| 1456 |
"created": int(time.time()),
|
| 1457 |
"owned_by": "google",
|
| 1458 |
"permission": [],
|
| 1459 |
-
"root": "gemini-2.5-pro-preview-
|
| 1460 |
"parent": None,
|
| 1461 |
},
|
| 1462 |
{
|
| 1463 |
-
"id": "gemini-2.5-pro-preview-
|
| 1464 |
"object": "model",
|
| 1465 |
"created": int(time.time()),
|
| 1466 |
"owned_by": "google",
|
| 1467 |
"permission": [],
|
| 1468 |
-
"root": "gemini-2.5-pro-preview-
|
| 1469 |
"parent": None,
|
| 1470 |
},
|
| 1471 |
{
|
|
|
|
| 1369 |
async def list_models(api_key: str = Depends(get_api_key)):
|
| 1370 |
# Based on current information for Vertex AI models
|
| 1371 |
models = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1372 |
{
|
| 1373 |
"id": "gemini-2.5-pro-exp-03-25",
|
| 1374 |
"object": "model",
|
|
|
|
| 1424 |
"parent": None,
|
| 1425 |
},
|
| 1426 |
{
|
| 1427 |
+
"id": "gemini-2.5-pro-preview-05-06",
|
| 1428 |
"object": "model",
|
| 1429 |
"created": int(time.time()),
|
| 1430 |
"owned_by": "google",
|
| 1431 |
"permission": [],
|
| 1432 |
+
"root": "gemini-2.5-pro-preview-05-06",
|
| 1433 |
"parent": None,
|
| 1434 |
},
|
| 1435 |
{
|
| 1436 |
+
"id": "gemini-2.5-pro-preview-05-06-search",
|
| 1437 |
"object": "model",
|
| 1438 |
"created": int(time.time()),
|
| 1439 |
"owned_by": "google",
|
| 1440 |
"permission": [],
|
| 1441 |
+
"root": "gemini-2.5-pro-preview-05-06",
|
| 1442 |
"parent": None,
|
| 1443 |
},
|
| 1444 |
{
|
| 1445 |
+
"id": "gemini-2.5-pro-preview-05-06-encrypt",
|
| 1446 |
"object": "model",
|
| 1447 |
"created": int(time.time()),
|
| 1448 |
"owned_by": "google",
|
| 1449 |
"permission": [],
|
| 1450 |
+
"root": "gemini-2.5-pro-preview-05-06",
|
| 1451 |
"parent": None,
|
| 1452 |
},
|
| 1453 |
{
|
| 1454 |
+
"id": "gemini-2.5-pro-preview-05-06-auto", # New auto model
|
| 1455 |
"object": "model",
|
| 1456 |
"created": int(time.time()),
|
| 1457 |
"owned_by": "google",
|
| 1458 |
"permission": [],
|
| 1459 |
+
"root": "gemini-2.5-pro-preview-05-06",
|
| 1460 |
"parent": None,
|
| 1461 |
},
|
| 1462 |
{
|