ComeçarComece de graça

Acompanhe os passos do agente, monitore a ação

Você está criando um assistente smolagents para um técnico de basquete que precisa de ajuda para analisar relatórios em PDF com estatísticas de jogadores, análises de olheiros e estratégias de jogo.

O técnico depende do agente para responder perguntas como: "Que estratégias defensivas devemos usar contra a segunda unidade deles?"

Mas o técnico não quer apenas respostas — ele quer visibilidade sobre o que o agente está fazendo nos bastidores.

Neste exercício, você vai escrever um callback de ação que é executado toda vez que o agente dá um passo, como ao chamar uma ferramenta ou usar o modelo. Esse callback vai:

  • Mostrar o número de passos,
  • E, se o agente terminou, exibir quantos tokens foram usados.

Isso vai ajudar o técnico (e você!) a monitorar como o agente está progredindo e quanto trabalho ele está fazendo para chegar a uma conclusão.

Este exercício faz parte do curso

Agentes de IA com Hugging Face smolagents

Ver curso

Instruções do exercício

  • Complete a assinatura da função adicionando o parâmetro agent_step.
  • Verifique se o passo atual produziu uma resposta final usando o atributo .is_final_answer de agent_step.
  • Se for a resposta final, obtenha o número total de tokens de total_tokens e imprima-o.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

# Define an action callback that accepts the agent step and the agent
def action_callback(____, agent):
    step_num = agent_step.step_number
    print(f"Step {step_num}: Analyzing basketball data!")
    
    # Check if the agent step includes token usage
    if agent_step.____:
        total_tokens = agent_step.token_usage.total_tokens
        # Print how many tokens were used
        print(f"Analysis complete! Total tokens used: {____}")
Editar e executar o código