luanpoppe commited on
Commit
e12451f
·
1 Parent(s): fa506ce

feat: atualizando valor do gemini 2.5 pro

Browse files
_utils/gerar_documento_utils/GerarDocumento.py CHANGED
@@ -171,7 +171,7 @@ class GerarDocumentoUtils:
171
  )
172
  elif llm_ultimas_requests == "gemini-2.5-pro":
173
  llm = llm_instance.google_gemini(
174
- "gemini-2.5-pro-preview-05-06", temperature=self.temperature
175
  )
176
  elif llm_ultimas_requests == "gemini-2.5-flash":
177
  llm = llm_instance.google_gemini(
 
171
  )
172
  elif llm_ultimas_requests == "gemini-2.5-pro":
173
  llm = llm_instance.google_gemini(
174
+ "gemini-2.5-pro-preview-06-05", temperature=self.temperature
175
  )
176
  elif llm_ultimas_requests == "gemini-2.5-flash":
177
  llm = llm_instance.google_gemini(
_utils/gerar_documento_utils/llm_calls.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  from typing import Literal, cast
3
 
4
  from pydantic import SecretStr
5
- from _utils.langchain_utils.LLM_class import LLM
6
  from setup.environment import default_model
7
  from langchain_core.messages import HumanMessage
8
  from langchain_openai import ChatOpenAI
@@ -59,9 +59,7 @@ def gpt_answer(
59
 
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
  temperature=0.4,
66
  ) -> str:
67
  gemini = llm.google_gemini(model, temperature)
 
2
  from typing import Literal, cast
3
 
4
  from pydantic import SecretStr
5
+ from _utils.langchain_utils.LLM_class import LLM, Google_llms
6
  from setup.environment import default_model
7
  from langchain_core.messages import HumanMessage
8
  from langchain_openai import ChatOpenAI
 
59
 
60
  async def agemini_answer(
61
  prompt,
62
+ model: Google_llms = "gemini-2.0-flash",
 
 
63
  temperature=0.4,
64
  ) -> str:
65
  gemini = llm.google_gemini(model, temperature)
_utils/langchain_utils/LLM_class.py CHANGED
@@ -11,7 +11,7 @@ google_api_key = cast(str, os.environ.get("GOOGLE_API_KEY_PEIXE"))
11
  open_ai_token = cast(str, os.environ.get("OPENAI_API_KEY"))
12
 
13
  Google_llms = Literal[
14
- "gemini-2.5-pro-preview-05-06",
15
  "gemini-2.0-flash",
16
  "gemini-2.0-flash-lite",
17
  "gemini-2.5-flash-preview-04-17",
 
11
  open_ai_token = cast(str, os.environ.get("OPENAI_API_KEY"))
12
 
13
  Google_llms = Literal[
14
+ "gemini-2.5-pro-preview-06-05",
15
  "gemini-2.0-flash",
16
  "gemini-2.0-flash-lite",
17
  "gemini-2.5-flash-preview-04-17",