L’addition, s’il vous plaît ! Donner aux agents l’accès aux données
Chez AgentsCafé, toutes les commandes de boissons sont enregistrées dans un fichier appelé orders.csv, qui contient les colonnes : table_id, drink_name et size.
Plutôt que de faire défiler le fichier à la main chaque fois qu’un client demande l’addition, vous allez créer un outil qui récupère toutes les commandes en cours pour une table donnée.
Remarque : tool et pandas ont déjà été importés pour vous.
Cet exercice fait partie du cours
Agents IA avec Hugging Face smolagents
Instructions
- Utilisez
table_idcomme paramètre de fonction pour que l’agent sache de quelle table récupérer les commandes. - Lisez le fichier
orders.csv, qui contient toutes les commandes de boissons. - Renvoyez la liste des commandes de la table.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Create a tool that receives the table_id as input
@tool
def lookup_orders(____: str) -> list[str]:
"""
Retrieves the current drink orders for a café table.
Args:
table_id (str): The table's identifier (e.g., "T5").
Returns:
list[str]: A list of drink orders, each formatted like "Latte (Large)".
"""
# Read the orders.csv file
df = pd.read_csv('____')
orders = df[df['table_id'] == table_id].apply(lambda row: f"{row['drink_name']} ({row['size']})", axis=1).tolist()
# Return the table orders
return ____