Spaces:
Sleeping
Sleeping
refactor: atualizar modelo Gemini e ajustar limite de texto para extração de referências
Browse files
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.
|
| 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[:
|
| 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("
|
| 168 |
print("⚠️ AVISO: Chave da API Google não encontrada!")
|
| 169 |
-
print("Crie um arquivo .env com:
|
| 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()
|