Upload 2 files

main.py

@@ -770,7 +770,7 @@ 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)
+            gt = GeminiLatest(api_key=selected_api_key, mongo_url=item.mongo_url, user_id=item.user_id)
             answer, gemini_chat = gt._get_response_gemini(item.query, item.gemini_chat)
             return SuccessResponse(
                 status="True",

models.py

@@ -72,6 +72,8 @@ class ChatgptCustom(BaseModel):
 class GeminiPro(BaseModel):
     query: str
     bard_api_key: Optional[str] = None
+    mongo_url: Optional[str] = None
+    user_id: Optional[int] = None
     gemini_api_key: Optional[str] = None
     gemini_chat: Optional[list] = None
     is_login: Optional[bool] = False