luanpoppe commited on
Commit
250b607
·
1 Parent(s): e2aa91f

feat: atualizando modelo do gemini 2.5 pro

Browse files
_utils/gerar_documento_utils/GerarDocumento.py CHANGED
@@ -226,7 +226,7 @@ class GerarDocumento:
226
  elif llm_ultimas_requests == "gemini-2.0-flash":
227
  llm = llm_instance.google_gemini("gemini-2.0-flash")
228
  elif llm_ultimas_requests == "gemini-2.5-pro":
229
- llm = llm_instance.google_gemini("gemini-2.5-pro-exp-03-25")
230
  return llm
231
 
232
  async def gerar_documento_final(
 
226
  elif llm_ultimas_requests == "gemini-2.0-flash":
227
  llm = llm_instance.google_gemini("gemini-2.0-flash")
228
  elif llm_ultimas_requests == "gemini-2.5-pro":
229
+ llm = llm_instance.google_gemini("gemini-2.5-pro-preview-05-06")
230
  return llm
231
 
232
  async def gerar_documento_final(
_utils/gerar_documento_utils/llm_calls.py CHANGED
@@ -60,7 +60,7 @@ def gpt_answer(
60
  async def agemini_answer(
61
  prompt,
62
  model: Literal[
63
- "gemini-2.5-pro-exp-03-25", "gemini-2.0-flash", "gemini-2.0-flash-lite"
64
  ] = "gemini-2.0-flash",
65
  ) -> str:
66
  gemini = llm.google_gemini(model)
 
60
  async def agemini_answer(
61
  prompt,
62
  model: Literal[
63
+ "gemini-2.5-pro-preview-05-06", "gemini-2.0-flash", "gemini-2.0-flash-lite"
64
  ] = "gemini-2.0-flash",
65
  ) -> str:
66
  gemini = llm.google_gemini(model)
_utils/langchain_utils/LLM_class.py CHANGED
@@ -10,7 +10,7 @@ google_api_key = cast(str, os.environ.get("GOOGLE_API_KEY_PEIXE"))
10
  open_ai_token = cast(str, os.environ.get("OPENAI_API_KEY"))
11
 
12
  Google_llms = Literal[
13
- "gemini-2.5-pro-exp-03-25", "gemini-2.0-flash", "gemini-2.0-flash-lite"
14
  ]
15
 
16
 
 
10
  open_ai_token = cast(str, os.environ.get("OPENAI_API_KEY"))
11
 
12
  Google_llms = Literal[
13
+ "gemini-2.5-pro-preview-05-06", "gemini-2.0-flash", "gemini-2.0-flash-lite"
14
  ]
15
 
16