Hora do jogo: execute o agente completo
Seu agente de técnico de basquete está quase pronto para entrar em quadra. Você escreveu uma função de callback para mostrar o número da etapa e o uso de tokens.
Agora é hora de montar tudo: você vai configurar um CodeAgent que usa esse callback.
Você já tem acesso a:
- Uma
basketball_toolque permite ao agente pesquisar relatórios de observação - Uma variável
modelcom seu modelo de linguagem - Uma função de callback:
action_callback
Seu objetivo é finalizar a conexão do agente.
Este exercício faz parte do curso
Agentes de IA com Hugging Face smolagents
Instruções do exercício
- Importe a classe
ActionSteppara que ela possa ser usada ao registrar o callback de ação. - Registre a função
action_callbackpara a classeActionSteppara que o agente mostre suas ações enquanto executa.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Import ActionStep to register a callback for it
from smolagents import ____
coach_agent = CodeAgent(
tools=[basketball_tool],
model=model,
verbosity_level=0,
# Register a callback that runs when an ActionStep is triggered
step_callbacks={ActionStep: ____}
)
# Run a question through the agent
result = coach_agent.run("What defensive strategy should we use to stop their point guard who averages 25 points per game?")
print(result)