tools.py

from rich import print as rprint
import random


def search_flights(departure_city, destination_city, travel_date):
    rprint("[green]Running search_flights[/green]")
    return [
        {
            "flight_number": "AA1422",
            "departure_time": "8:00 AM",
            travel_date: travel_date,
            departure_city: departure_city,
            destination_city: destination_city,
        },
        {
            "flight_number": "DL1089",
            "departure_time": "10:00 AM",
            travel_date: travel_date,
            departure_city: departure_city,
            destination_city: destination_city,
        },
    ]


def book_flight(flight_number, travel_date):
    rprint("[green]Running book_flight[/green]")
    seat_number = f"{random.randint(1, 30)}A"
    return {
        "status": "confirmed",
        "flight_number": flight_number,
        "travel_date": travel_date,
        "seat_number": seat_number,
    }