Spaces:
Sleeping
Sleeping
| from typing import Any, Optional | |
| from smolagents.tools import Tool | |
| from duckduckgo_search import DDGS | |
| class SearchAccommodationsTool(Tool): | |
| name = "search_accommodations" | |
| description = "Searches for available accommodations at a travel destination with customizable filters like budget, style, and location." | |
| inputs = { | |
| 'destination': {'type': 'string', 'description': 'The destination city or region (e.g., "Tokyo", "Bali", "Paris")'}, | |
| 'budget': {'type': 'string', 'description': 'Price range (e.g., "budget", "mid-range", "luxury")', 'nullable': True}, | |
| 'style': {'type': 'string', 'description': 'Type of accommodation (e.g., "hotel", "hostel", "apartment", "resort")', 'nullable': True}, | |
| 'location': {'type': 'string', 'description': 'Preferred area or neighborhood', 'nullable': True} | |
| } | |
| output_type = "string" | |
| def __init__(self, max_results=8): | |
| super().__init__() | |
| try: | |
| from duckduckgo_search import DDGS | |
| except ImportError as e: | |
| raise ImportError( | |
| "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`." | |
| ) from e | |
| self.ddgs = DDGS() | |
| self.max_results = max_results | |
| def forward(self, destination: str, budget: Optional[str] = None, style: Optional[str] = None, location: Optional[str] = None) -> str: | |
| try: | |
| # Construct a search query based on the parameters | |
| query = f"accommodations {destination}" | |
| if budget: | |
| query += f" {budget} price" | |
| if style: | |
| query += f" {style}" | |
| if location: | |
| query += f" in {location}" | |
| # Execute the search | |
| results = self.ddgs.text(query, max_results=self.max_results) | |
| if not results: | |
| return f"No accommodation results found for {destination}. Try adjusting your search parameters." | |
| # Format the results | |
| formatted_results = f"🏨 **Accommodation Options in {destination}**\n\n" | |
| # Add search parameters if provided | |
| search_params = [] | |
| if budget: | |
| search_params.append(f"Budget: {budget}") | |
| if style: | |
| search_params.append(f"Type: {style}") | |
| if location: | |
| search_params.append(f"Area: {location}") | |
| if search_params: | |
| formatted_results += "Search filters: " + ", ".join(search_params) + "\n\n" | |
| # Format each result | |
| for i, result in enumerate(results, 1): | |
| title = result.get('title', 'Unnamed Accommodation') | |
| url = result.get('href', '#') | |
| snippet = result.get('body', 'No description available') | |
| formatted_results += f"### {i}. {title}\n" | |
| formatted_results += f"{snippet}\n" | |
| formatted_results += f"[View Details]({url})\n\n" | |
| formatted_results += "Note: These are search results. For accurate pricing and availability, check the official websites or booking platforms." | |
| return formatted_results | |
| except Exception as e: | |
| return f"Error searching for accommodations: {str(e)}" | |