Spaces:
Running
Running
luanpoppe
commited on
Commit
·
64ed488
1
Parent(s):
4b3b841
feat: adicionando opção de fazer requisições para o gemini 2.5 pro
Browse files
_utils/gerar_documento_utils/GerarDocumento.py
CHANGED
@@ -211,7 +211,7 @@ class GerarDocumento:
|
|
211 |
def select_model_for_last_requests(
|
212 |
self,
|
213 |
llm_ultimas_requests: Literal[
|
214 |
-
"gpt-4o-mini", "deepseek-chat", "gemini-2.0-flash"
|
215 |
],
|
216 |
):
|
217 |
llm_instance = LLM()
|
@@ -225,6 +225,8 @@ class GerarDocumento:
|
|
225 |
llm = llm_instance.deepseek()
|
226 |
elif llm_ultimas_requests == "gemini-2.0-flash":
|
227 |
llm = llm_instance.google_gemini("gemini-2.0-flash")
|
|
|
|
|
228 |
return llm
|
229 |
|
230 |
async def gerar_documento_final(
|
|
|
211 |
def select_model_for_last_requests(
|
212 |
self,
|
213 |
llm_ultimas_requests: Literal[
|
214 |
+
"gpt-4o-mini", "deepseek-chat", "gemini-2.0-flash", "gemini-2.5-pro"
|
215 |
],
|
216 |
):
|
217 |
llm_instance = LLM()
|
|
|
225 |
llm = llm_instance.deepseek()
|
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-exp-03-25")
|
230 |
return llm
|
231 |
|
232 |
async def gerar_documento_final(
|