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