From a7cc8ce0739c988dbb60f357cb29ef7672a3e19f Mon Sep 17 00:00:00 2001
From: Okk Hex <78189163+okkhex@users.noreply.github.com>
Date: Wed, 2 Apr 2025 00:39:23 +0530
Subject: [PATCH] fix gemini dynamic model (#470)

---
 plugins/aiwrapper.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/plugins/aiwrapper.py b/plugins/aiwrapper.py
index c5fa34c..2da7273 100644
--- a/plugins/aiwrapper.py
+++ b/plugins/aiwrapper.py
@@ -36,7 +36,7 @@ import asyncio
 ENDPOINTS = {
     "gpt": "https://api.openai.com/v1/chat/completions",
     "antr": "https://api.anthropic.com/v1/messages",
-    "gemini": "https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent",
+    "gemini": "https://generativelanguage.googleapis.com/v1/models/",
     "deepseek": "https://api.deepseek.com/chat/completions"
 }
 
@@ -163,6 +163,7 @@ async def get_ai_response(provider, prompt, api_key, stream=False):
                                 continue
 
         elif provider == "gemini":
+            endpoint = f"{ENDPOINTS[provider]}{model}:generateContent"
             params = {"key": api_key}
             data = {
                 "contents": [{
@@ -170,7 +171,7 @@ async def get_ai_response(provider, prompt, api_key, stream=False):
                 }]
             }
             response = await async_searcher(
-                ENDPOINTS[provider],
+                endpoint,
                 params=params,
                 headers=headers,
                 post=True,