¡La cuenta, por favor! Dar acceso a datos a los agentes
En AgentsCafé, todos los pedidos de bebidas se guardan en un archivo llamado orders.csv, que incluye las columnas: table_id, drink_name y size.
En lugar de revisar el archivo manualmente cada vez que un cliente pide la cuenta, vas a crear una herramienta que busque todos los pedidos actuales de una mesa concreta.
Nota: tool y pandas ya se han importado por ti.
Este ejercicio forma parte del curso
Agentes de IA con Hugging Face smolagents
Instrucciones del ejercicio
- Usa
table_idcomo parámetro de la función para que el agente sepa de qué mesa recuperar los pedidos. - Lee el archivo
orders.csv, que contiene todos los pedidos de bebidas. - Devuelve la lista de pedidos de bebidas de la mesa.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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 ____