fccoelho commited on
Commit
f2cfb58
·
1 Parent(s): 95baff4

refactor: atualizar modelo Gemini e ajustar limite de texto para extração de referências

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -60,7 +60,7 @@ def extract_references_with_llm(text):
60
 
61
  # Criar o agente Pydantic AI
62
  agent = Agent(
63
- 'gemini-2.0-flash-exp', # Modelo Gemini 2.0 Flash
64
  result_type=ReferencesResponse,
65
  system_prompt="""
66
  Você é um especialista em análise de artigos científicos.
@@ -80,7 +80,7 @@ def extract_references_with_llm(text):
80
  )
81
 
82
  # Limitar o texto para evitar exceder limites da API
83
- limited_text = text[:15000] # Gemini tem limite maior que GPT
84
 
85
  # Executar o agente
86
  result = agent.run_sync(f"Extraia as referências bibliográficas do seguinte texto de artigo científico:\n\n{limited_text}")
@@ -164,9 +164,9 @@ def main():
164
  load_dotenv() # Carrega variáveis de ambiente do arquivo .env
165
 
166
  # Verificar se a chave da API está configurada
167
- if not os.getenv("GOOGLE_API_KEY"):
168
  print("⚠️ AVISO: Chave da API Google não encontrada!")
169
- print("Crie um arquivo .env com: GOOGLE_API_KEY=sua_chave_aqui")
170
  print("Obtenha sua chave em: https://aistudio.google.com/app/apikey")
171
 
172
  interface = create_interface()
 
60
 
61
  # Criar o agente Pydantic AI
62
  agent = Agent(
63
+ 'gemini-2.5-pro', # Modelo Gemini 2.0 Flash
64
  result_type=ReferencesResponse,
65
  system_prompt="""
66
  Você é um especialista em análise de artigos científicos.
 
80
  )
81
 
82
  # Limitar o texto para evitar exceder limites da API
83
+ limited_text = text[:150000] # Gemini tem limite maior que GPT
84
 
85
  # Executar o agente
86
  result = agent.run_sync(f"Extraia as referências bibliográficas do seguinte texto de artigo científico:\n\n{limited_text}")
 
164
  load_dotenv() # Carrega variáveis de ambiente do arquivo .env
165
 
166
  # Verificar se a chave da API está configurada
167
+ if not os.getenv("GEMINI_API_KEY"):
168
  print("⚠️ AVISO: Chave da API Google não encontrada!")
169
+ print("Crie um arquivo .env com: GEMINI_API_KEY=sua_chave_aqui")
170
  print("Obtenha sua chave em: https://aistudio.google.com/app/apikey")
171
 
172
  interface = create_interface()