fix: limite de gemini 2.5 pro

_utils/gerar_documento.py

@@ -96,9 +96,11 @@ async def gerar_documento(
         axiom_instance.send_axiom(
             "COMEÇANDO REQUISIÇÃO PARA GERAR O QUERY DINAMICAMENTE DO VECTOR STORE"
         )
-        query_gerado_dinamicamente_para_o_vector_store = await llm.google_gemini(
-            "gemini-2.5-pro-exp-03-25"
-        ).ainvoke([HumanMessage(content=prompt_para_gerar_query_dinamico)])
+        query_gerado_dinamicamente_para_o_vector_store = (
+            await llm.google_gemini_ainvoke(
+                prompt_para_gerar_query_dinamico, "gemini-2.0-flash"
+            )
+        )
 
         axiom_instance.send_axiom(
             f"query_gerado_dinamicamente_para_o_vector_store: {query_gerado_dinamicamente_para_o_vector_store.content}",
@@ -111,9 +113,6 @@ async def gerar_documento(
             )
         )
 
-        if isinstance(query_gerado_dinamicamente_para_o_vector_store.content, list):
-            query_gerado_dinamicamente_para_o_vector_store.content = "\n".join(query_gerado_dinamicamente_para_o_vector_store.content)  # type: ignore
-
         llm_ultimas_requests = serializer.llm_ultimas_requests
         axiom_instance.send_axiom("COMEÇANDO A FAZER ÚLTIMA REQUISIÇÃO")
         structured_summaries = await summarizer.gerar_documento_final(

_utils/gerar_documento_utils/GerarDocumento.py

@@ -313,20 +313,29 @@ class GerarDocumento:
 
         while tentativas < 5 and not documento_gerado:
             tentativas += 1
-            resposta = llm.invoke(prompt)
-            if hasattr(resposta, "content") and resposta.content.strip():  # type: ignore
-                documento_gerado = resposta.content.strip()  # type: ignore
-            else:
-                print(f"Tentativa {tentativas}: resposta vazia ou inexistente.")
-                time.sleep(5)
+            try:
+                resposta = llm.invoke(prompt)
+                if hasattr(resposta, "content") and resposta.content.strip():  # type: ignore
+                    documento_gerado = resposta.content.strip()  # type: ignore
+                else:
+                    print(f"Tentativa {tentativas}: resposta vazia ou inexistente.")
+            except Exception as e:
+                llm = self.select_model_for_last_requests("gemini-2.0-flash")
+                print(f"Tentativa {tentativas}: erro ao invocar o modelo: {e}")
+            time.sleep(5)
 
         if not documento_gerado:
-            self.axiom_instance.send_axiom(
-                "TENTANDO GERAR DOCUMENTO FINAL COM GPT 4o-mini COMO ÚLTIMA TENTATIVA"
-            )
-            documento_gerado = self.gerar_documento_utils.ultima_tentativa_requisicao(
-                prompt
-            )
+            try:
+                self.axiom_instance.send_axiom(
+                    "TENTANDO GERAR DOCUMENTO FINAL COM GPT 4o-mini COMO ÚLTIMA TENTATIVA"
+                )
+                documento_gerado = (
+                    self.gerar_documento_utils.ultima_tentativa_requisicao(prompt)
+                )
+            except Exception as e:
+                raise Exception(
+                    "Falha ao gerar o documento final na última tentativa."
+                ) from e
 
         return documento_gerado
 

_utils/langchain_utils/LLM_class.py

@@ -45,26 +45,30 @@ class LLM:
         self,
         prompt: str,
         model: Google_llms = "gemini-2.0-flash",
+        max_retries: int = 3,
     ):
-        try:
-            response = await self.google_gemini(model).ainvoke(
-                [HumanMessage(content=prompt)]
-            )
-            return response
-        except:
+        for attempt in range(max_retries):
             try:
-                response = await self.google_gemini("gemini-2.0-flash").ainvoke(
+                response = await self.google_gemini(model).ainvoke(
                     [HumanMessage(content=prompt)]
                 )
+
+                if isinstance(response.content, list):
+                    response.content = "\n".join(response.content)  # type: ignore
+
                 return response
-            except:
-                try:
-                    response = await self.google_gemini("gemini-2.0-flash").ainvoke(
-                        [HumanMessage(content=prompt)]
-                    )
-                    return response
+            except Exception as e:
+                model = "gemini-2.0-flash"
+                print(f"Attempt {attempt + 1} failed with error: {e}")
 
-                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."
-                    )
+        # Final attempt fallback logic (optional)
+        try:
+            print("Final attempt with fallback model...")
+            response = await self.open_ai("chat-gpt-4o-mini").ainvoke(
+                [HumanMessage(content=prompt)]
+            )
+            return response
+        except Exception as e:
+            raise Exception(
+                "Failed to generate the final document after 5 retries and the fallback attempt with chat-gpt-4o-mini."
+            ) from e