Upload 2 files

main.py

@@ -761,8 +761,13 @@ def gemini_pro(item: GeminiPro):
     if item.is_multi_chat:
         selected_api_key = ASSISTANT_GOOGLE_API_KEYS or item.gemini_api_key
         try:
-            gt = GeminiLatest(api_key=selected_api_key, mongo_url=item.mongo_url, user_id=item.user_id)
-            answer, gemini_chat = gt._GeminiLatest__get_response_gemini(item.query)
+            geni = GeminiLatest(
+                api_key=selected_api_key,
+                mongo_url=item.mongo_url,
+                version=item.version,
+                user_id=item.user_id
+            )
+            answer, gemini_chat = geni._GeminiLatest__get_response_gemini(item.query)
             return SuccessResponse(
                 status="True",
                 randydev={

models.py

@@ -74,6 +74,7 @@ class GeminiPro(BaseModel):
     bard_api_key: Optional[str] = None
     mongo_url: Optional[str] = None
     user_id: Optional[int] = None
+    version: Optional[str] = "v1"
     gemini_api_key: Optional[str] = None
     is_login: Optional[bool] = False
     is_multi_chat: Optional[bool] = False