Rachunek poproszę! Daj agentowi dostęp do danych
W AgentsCafé wszystkie zamówienia na napoje są zapisane w pliku orders.csv, który zawiera kolumny: table_id, drink_name i size.
Zamiast ręcznie przeszukiwać plik za każdym razem, gdy klient prosi o rachunek, zbudujesz narzędzie, które wyszukuje wszystkie bieżące zamówienia dla konkretnego stolika.
Uwaga: tool i pandas zostały już zaimportowane.
To ćwiczenie jest częścią kursu
Agenty AI z Hugging Face smolagents
Instrukcje do ćwiczenia
- Użyj
table_idjako parametru funkcji, żeby agent wiedział, dla którego stolika pobrać zamówienia. - Wczytaj plik
orders.csv, który zawiera wszystkie zamówienia na napoje. - Zwróć listę zamówień dla danego stolika.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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 ____