Spaces:
Sleeping
Sleeping
File size: 3,463 Bytes
9ef3f6c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
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)}"
|