Spaces:
Sleeping
Sleeping
refactor: ajustar limites de texto para modelos de IA e simplificar prompt
Browse files
app.py
CHANGED
|
@@ -85,15 +85,15 @@ def extract_references_with_llm(text, model_name):
|
|
| 85 |
- pages: páginas (se disponível)
|
| 86 |
- doi: DOI (se disponível)
|
| 87 |
|
| 88 |
-
Seja preciso e extraia
|
| 89 |
"""
|
| 90 |
)
|
| 91 |
|
| 92 |
# Ajustar limite de texto baseado no modelo
|
| 93 |
if model_name.startswith('gemini'):
|
| 94 |
-
limited_text = text[:
|
| 95 |
else:
|
| 96 |
-
limited_text = text[:
|
| 97 |
|
| 98 |
# Executar o agente
|
| 99 |
result = agent.run_sync(f"Extraia as referências bibliográficas do seguinte texto de artigo científico:\n\n{limited_text}")
|
|
|
|
| 85 |
- pages: páginas (se disponível)
|
| 86 |
- doi: DOI (se disponível)
|
| 87 |
|
| 88 |
+
Seja preciso e extraia referências completas.
|
| 89 |
"""
|
| 90 |
)
|
| 91 |
|
| 92 |
# Ajustar limite de texto baseado no modelo
|
| 93 |
if model_name.startswith('gemini'):
|
| 94 |
+
limited_text = text[:1500000] # Gemini tem limite maior
|
| 95 |
else:
|
| 96 |
+
limited_text = text[:500000] # OpenAI tem limite menor
|
| 97 |
|
| 98 |
# Executar o agente
|
| 99 |
result = agent.run_sync(f"Extraia as referências bibliográficas do seguinte texto de artigo científico:\n\n{limited_text}")
|