temporal-ai-agent/tools/search_flights.py

def search_flights(args: dict) -> dict:
    """
    Example function for searching flights.
    Currently just prints/returns the passed args,
    but you can add real flight search logic later.
    """
    date_depart = args.get("dateDepart")
    date_return = args.get("dateReturn")
    origin = args.get("origin")
    destination = args.get("destination")

    print(f"Searching flights from {origin} to {destination}")
    print(f"Depart: {date_depart}, Return: {date_return}")

    # Return a mock result so you can verify it
    return {
        "tool": "SearchFlights",
        "searchResults": [
            "Mock flight 1",
            "Mock flight 2",
        ],
        "status": "search-complete",
        "args": args,
    }