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