Spaces:
Running
Running
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
|
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:
|
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
|
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",
|