luanpoppe commited on
Commit
64ed488
·
1 Parent(s): 4b3b841

feat: adicionando opção de fazer requisições para o gemini 2.5 pro

Browse files
_utils/gerar_documento_utils/GerarDocumento.py CHANGED
@@ -211,7 +211,7 @@ class GerarDocumento:
211
  def select_model_for_last_requests(
212
  self,
213
  llm_ultimas_requests: Literal[
214
- "gpt-4o-mini", "deepseek-chat", "gemini-2.0-flash"
215
  ],
216
  ):
217
  llm_instance = LLM()
@@ -225,6 +225,8 @@ class GerarDocumento:
225
  llm = llm_instance.deepseek()
226
  elif llm_ultimas_requests == "gemini-2.0-flash":
227
  llm = llm_instance.google_gemini("gemini-2.0-flash")
 
 
228
  return llm
229
 
230
  async def gerar_documento_final(
 
211
  def select_model_for_last_requests(
212
  self,
213
  llm_ultimas_requests: Literal[
214
+ "gpt-4o-mini", "deepseek-chat", "gemini-2.0-flash", "gemini-2.5-pro"
215
  ],
216
  ):
217
  llm_instance = LLM()
 
225
  llm = llm_instance.deepseek()
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(