fix: google gemini

_utils/langchain_utils/LLM_class.py

@@ -3,11 +3,16 @@ from pydantic import SecretStr
 from setup.environment import default_model
 from setup.easy_imports import ChatOpenAI, ChatGoogleGenerativeAI
 import os
+from langchain_core.messages import HumanMessage
 
 deepseek_api_key = cast(str, os.environ.get("DEEPSEEKK_API_KEY"))
 google_api_key = cast(str, os.environ.get("GOOGLE_API_KEY_PEIXE"))
 open_ai_token = cast(str, os.environ.get("OPENAI_API_KEY"))
 
+Google_llms = Literal[
+    "gemini-2.5-pro-exp-03-25", "gemini-2.0-flash", "gemini-2.0-flash-lite"
+]
+
 
 class LLM:
     def __init__(self):
@@ -25,9 +30,7 @@ class LLM:
 
     def google_gemini(
         self,
-        model: Literal[
-            "gemini-2.5-pro-exp-03-25", "gemini-2.0-flash", "gemini-2.0-flash-lite"
-        ] = "gemini-2.0-flash",
+        model: Google_llms = "gemini-2.0-flash",
     ):
         return ChatGoogleGenerativeAI(
             api_key=SecretStr(google_api_key),
@@ -37,3 +40,31 @@ class LLM:
             timeout=None,
             max_retries=2,
         )
+
+    async def google_gemini_ainvoke(
+        self,
+        prompt: str,
+        model: Google_llms = "gemini-2.0-flash",
+    ):
+        try:
+            response = await self.google_gemini(model).ainvoke(
+                [HumanMessage(content=prompt)]
+            )
+            return response
+        except:
+            try:
+                response = await self.google_gemini("gemini-2.0-flash").ainvoke(
+                    [HumanMessage(content=prompt)]
+                )
+                return response
+            except:
+                try:
+                    response = await self.google_gemini("gemini-2.0-flash").ainvoke(
+                        [HumanMessage(content=prompt)]
+                    )
+                    return response
+
+                except:
+                    Exception(
+                        "Falha ao tentar gerar o documento final por 5 tentativas e também ao tentar na última tentativa com o chat-gpt 4o mini."
+                    )