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 |
|