luanpoppe commited on
Commit
99fb68e
·
1 Parent(s): 9ee4c9e

feat: adicionando opção de gemini 2.5 flash

Browse files
_utils/gerar_documento_utils/GerarDocumento.py CHANGED
@@ -227,6 +227,8 @@ class GerarDocumento:
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(
 
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
+ elif llm_ultimas_requests == "gemini-2.5-flash":
231
+ llm = llm_instance.google_gemini("gemini-2.5-flash-preview-04-17")
232
  return llm
233
 
234
  async def gerar_documento_final(
_utils/langchain_utils/LLM_class.py CHANGED
@@ -10,7 +10,10 @@ 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-preview-05-06", "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",
14
+ "gemini-2.0-flash",
15
+ "gemini-2.0-flash-lite",
16
+ "gemini-2.5-flash-preview-04-17",
17
  ]
18
 
19