A conta, por favor! Dando acesso dos agentes aos dados
No AgentsCafé, todos os pedidos de bebidas são salvos em um arquivo chamado orders.csv, que inclui as colunas: table_id, drink_name e size.
Em vez de percorrer o arquivo manualmente toda vez que um cliente pedir a conta, você vai criar uma ferramenta que busca todos os pedidos atuais de uma mesa específica.
Observação: tool e pandas já foram importados para você.
Este exercício faz parte do curso
Agentes de IA com Hugging Face smolagents
Instruções do exercício
- Use
table_idcomo parâmetro da função para que o agente saiba de qual mesa recuperar os pedidos. - Leia o arquivo
orders.csv, que contém todos os pedidos de bebidas. - Retorne a lista de pedidos de bebidas da mesa.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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 ____