À vous de jouer : exécuter l’agent complet
Votre agent d’entraîneur de basketball est presque prêt à entrer sur le terrain. Vous avez écrit une fonction de callback qui affiche le numéro de l’étape et l’utilisation des jetons.
Il est maintenant temps d’assembler le tout : vous allez configurer un CodeAgent qui utilise ce callback.
Vous avez déjà accès à :
- Un
basketball_toolqui permet à l’agent de rechercher dans des rapports de scouting - Une variable
modelavec votre modèle de langue - Une fonction de callback :
action_callback
Votre objectif est de terminer le câblage de l’agent.
Cet exercice fait partie du cours
Agents IA avec Hugging Face smolagents
Instructions
- Importez la classe
ActionSteppour pouvoir enregistrer le callback d’action. - Enregistrez la fonction
action_callbackpour la classeActionStepafin que l’agent affiche ses actions pendant l’exécution.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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)