Upload 2 files

main.py

@@ -64,12 +64,12 @@ from fastapi.responses import FileResponse
 from pymongo import MongoClient
 
 from RyuzakiLib.hackertools.chatgpt import RendyDevChat
+from RyuzakiLib.hackertools.gemini import GeminiLatest
 from RyuzakiLib.hackertools.openai_api import OpenAiToken
 from RyuzakiLib.mental import BadWordsList
 from RyuzakiLib.system import OpenReadSystem
 
 from bardapi import Bard
-import google.generativeai as genai
 from models import *
 from gpytranslate import SyncTranslator
 
@@ -116,8 +116,6 @@ collection = db["users"]
 
 trans = SyncTranslator()
 
-genai.configure(api_key=ASSISTANT_GOOGLE_API_KEYS)
-
 app = FastAPI(docs_url=None, redoc_url="/")
 
 def get_all_api_keys():
@@ -772,15 +770,13 @@ def gemini_pro(item: GeminiPro):
     """
     if item.is_multi_chat:
         try:
-            item.multi_chat_messages.append({"role": "user", "parts": [item.query]})
-            model = genai.GenerativeModel("gemini-pro")
-            response_genai = model.generate_content(item.multi_chat_messages)
-            item.multi_chat_messages.append({"role": "model", "parts": [response_genai.text]})
+            gt = GeminiLatest(api_key=ASSISTANT_GOOGLE_API_KEYS)
+            response = gt._get_response_gemini(query)
             return SuccessResponse(
                 status="True",
                 randydev={
-                    "message": response_genai.text,
-                    "chat_history": item.multi_chat_messages
+                    "message": response[0],
+                    "chat_history": response[1]
                 }
             )
         except:

models.py

@@ -71,7 +71,6 @@ class ChatgptCustom(BaseModel):
 
 class GeminiPro(BaseModel):
     query: str
-    multi_chat_messages: Optional[list] = []
     bard_api_key: Optional[str] = None
     is_login: Optional[bool] = False
     is_multi_chat: Optional[bool] = False