Spaces:
Sleeping
Sleeping
| from smolagents import tool, Tool | |
| def suggest_best_cities(country: str) -> str: | |
| """ | |
| Suggests a the best city based on the country | |
| Args: | |
| country (str): A city to get for a given country. Allowed values are: | |
| - "jordan": Aqaba | |
| - "syria": ALeppo | |
| - "palestine": Gaza | |
| - "usa": Sheriff Grady Judd | |
| - "germany": Cologne | |
| """ | |
| if country == "jordan": | |
| return "Aqaba" | |
| elif country == "syria": | |
| return "ALeppo" | |
| elif country == "palestine": | |
| return "Gaza" | |
| elif country == "usa": | |
| return "Sheriff Grady Judd" | |
| elif country == "germany": | |
| return "Cologne" | |
| else: | |
| return "Irbid" | |
| class RecommendCountryTool(Tool): | |
| name = "recommend_country_generator" | |
| description = """ | |
| This tool suggests a best city based on the country/ | |
| It returns a unique city.""" | |
| inputs = { | |
| "country": { | |
| "type": "string", | |
| "description": "The best city for the given country (e.g, 'jordan', 'syria', 'palestine')." | |
| } | |
| } | |
| output_type = "string" | |
| def forward(self, country: str): | |
| countries = { | |
| "jordan": "Aqaba", | |
| "syria": "Aleppo", | |
| "palestine": "Gaza City" | |
| } | |
| return countries.get(country.lower(), "City for given country not found.") | |