change gemini 1.0 pro to gemini 1.5 flash

bruceunx · Jun 13, 2024 · 14d858e · 14d858e
1 parent 68703f8
commit 14d858e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gpt_server/routers/gemini.py b/gpt_server/routers/gemini.py
@@ -128,8 +128,8 @@ async def get_content(data, url):
 @router.post("/chat", status_code=200)
 @check_limit
 async def chat(request: Request, prompt: Prompt):
-    return handle_request(prompt, "models/gemini-1.0-pro", API_URL)
+    return handle_request(prompt, "models/gemini-1.5-flash", API_URL)
 
 
 async def chat_pro(request: Request, prompt: Prompt):
-    return handle_request(prompt, "models/gemini-1.5-pro-latest", API_URL_PRO)
+    return handle_request(prompt, "models/gemini-1.5-pro", API_URL_PRO)