Sigue los pasos del agente y monitoriza la acción
Estás creando un asistente de smolagents para un entrenador de baloncesto que necesita ayuda para analizar informes en PDF con estadísticas de jugadores, observaciones de scouting y estrategias de juego.
El entrenador confía en el agente para responder preguntas como: "¿Qué estrategias defensivas deberíamos aplicar contra su segunda unidad?"
Pero el entrenador no solo quiere respuestas: también quiere ver qué está haciendo el agente entre bambalinas.
En este ejercicio, escribirás un callback de acción que se ejecuta cada vez que el agente da un paso, como cuando llama a una herramienta o usa el modelo. Este callback:
- Mostrará el número de pasos,
- Y, si el agente ha terminado, mostrará cuántos tokens se han usado.
Esto ayudará al entrenador (¡y a ti!) a monitorizar cómo progresa el agente y cuánto trabajo realiza para llegar a una conclusión.
Este ejercicio forma parte del curso
Agentes de IA con Hugging Face smolagents
Instrucciones del ejercicio
- Completa la firma de la función añadiendo el parámetro
agent_step. - Comprueba si el paso actual produjo una respuesta final usando el atributo
.is_final_answerdeagent_step. - Si es la respuesta final, obtén el número total de tokens de
total_tokense imprímelo.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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: {____}")