fccoelho commited on
Commit
32d9fa7
·
1 Parent(s): 252b190

refactor: ajustar limites de texto para modelos de IA e simplificar prompt

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 apenas referências válidas e completas.
89
  """
90
  )
91
 
92
  # Ajustar limite de texto baseado no modelo
93
  if model_name.startswith('gemini'):
94
- limited_text = text[:150000] # Gemini tem limite maior
95
  else:
96
- limited_text = text[:50000] # 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}")
 
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}")