fix: nomes dos modelos do gemini

_utils/gerar_documento_utils/GerarDocumento.py

@@ -171,11 +171,11 @@ class GerarDocumentoUtils:
             )
         elif llm_ultimas_requests == "gemini-2.5-pro":
             llm = llm_instance.google_gemini(
-                "gemini-2.5-pro-preview-06-05", temperature=self.temperature
+                "gemini-2.5-pro", temperature=self.temperature
             )
         elif llm_ultimas_requests == "gemini-2.5-flash":
             llm = llm_instance.google_gemini(
-                "gemini-2.5-flash-preview-04-17", temperature=self.temperature
+                "gemini-2.5-flash", temperature=self.temperature
             )
         return llm
 

_utils/langchain_utils/LLM_class.py

@@ -11,10 +11,11 @@ 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-preview-06-05",
+    "gemini-2.5-pro",
     "gemini-2.0-flash",
     "gemini-2.0-flash-lite",
-    "gemini-2.5-flash-preview-04-17",
+    "gemini-2.5-flash",
+    "gemini-2.5-flash",
 ]
 
 
@@ -79,7 +80,7 @@ class LLM:
         self,
         prompt: str,
         list_of_pdfs: List[str],
-        model: Google_llms = "gemini-2.5-flash-preview-04-17",
+        model: Google_llms = "gemini-2.5-flash",
         max_retries: int = 3,
     ) -> str | None:
         message_parts = [