|
|
from transformers.tools import tool |
|
|
import random |
|
|
|
|
|
|
|
|
destinations = [ |
|
|
{ |
|
|
"destination_name": "Kyoto", |
|
|
"country": "Japan", |
|
|
"best_time_to_visit": "March to May, October to November", |
|
|
"climate": "Temperate", |
|
|
"currency": "Japanese Yen", |
|
|
"fun_facts": "Famous for temples and cherry blossoms.", |
|
|
"top_attractions": ["Fushimi Inari Shrine", "Kinkaku-ji", "Arashiyama Bamboo Grove"], |
|
|
"continent": "Asia", |
|
|
"budget_friendly": False |
|
|
}, |
|
|
{ |
|
|
"destination_name": "Lisbon", |
|
|
"country": "Portugal", |
|
|
"best_time_to_visit": "March to October", |
|
|
"climate": "Warm Mediterranean", |
|
|
"currency": "Euro", |
|
|
"fun_facts": "One of the oldest cities in Western Europe.", |
|
|
"top_attractions": ["Belém Tower", "Alfama district", "Lisbon Oceanarium"], |
|
|
"continent": "Europe", |
|
|
"budget_friendly": True |
|
|
}, |
|
|
{ |
|
|
"destination_name": "Cape Town", |
|
|
"country": "South Africa", |
|
|
"best_time_to_visit": "March to May, September to November", |
|
|
"climate": "Mediterranean", |
|
|
"currency": "South African Rand", |
|
|
"fun_facts": "Known for Table Mountain and coastal scenery.", |
|
|
"top_attractions": ["Table Mountain", "Cape Point", "Robben Island"], |
|
|
"continent": "Africa", |
|
|
"budget_friendly": True |
|
|
}, |
|
|
|
|
|
] |
|
|
|
|
|
@tool |
|
|
def random_travel_destination( |
|
|
continent: str = None, |
|
|
climate: str = None, |
|
|
budget_friendly: bool = None, |
|
|
) -> dict: |
|
|
""" |
|
|
Returns a random travel destination. You can optionally filter by continent, climate, and budget-friendliness. |
|
|
""" |
|
|
filtered = destinations |
|
|
|
|
|
if continent: |
|
|
filtered = [d for d in filtered if d["continent"].lower() == continent.lower()] |
|
|
if climate: |
|
|
filtered = [d for d in filtered if d["climate"].lower() == climate.lower()] |
|
|
if budget_friendly is not None: |
|
|
filtered = [d for d in filtered if d["budget_friendly"] == budget_friendly] |
|
|
|
|
|
if not filtered: |
|
|
return {"error": "No destinations match the selected filters."} |
|
|
|
|
|
return random.choice(filtered) |
|
|
|
