Add arg number_of_options with default value to app.py

app.py

@@ -14,7 +14,8 @@ search_tool = DuckDuckGoSearchTool()
 def find_open_restaurants_by_location(location: str, 
                                 cuisine: str = None, 
                                 rating: float = None, 
-                                timezone: str = None
+                                timezone: str = None,
+                                number_of_options: int = 5
                                )-> str: 
     """A tool that searches for restaurants according to location, cuisine, rating and local time. 
     Args:
@@ -22,6 +23,7 @@ def find_open_restaurants_by_location(location: str,
         cuisine: An optional string representing the cuisine type (e.g. Asian, Indian, Italian, breakfast, lunch, etc.)
         rating: An optional float representing the average customer rating of the restaurant from Google, TripAdvisor, or Yelp.
         timezone: An optional string representing a valid timezone (e.g., 'America/New_York').
+        number_of_options: An integer representing the number of return options, default set to 5.
     """
     try:
        # If no timezone is provided, default to UTC
@@ -32,7 +34,8 @@ def find_open_restaurants_by_location(location: str,
         local_time = datetime.datetime.now(tz).strftime("%H:%M")
 
         # Build query dynamically based on provided inputs
-        query_parts = [f'Find a list of 5 restaurants currently open in "{location}".']
+        query_parts = [f"restaurants open in {location}"]
+        
         if cuisine:
             query_parts.append(f"serving {cuisine} cuisine")
         if rating:
@@ -42,12 +45,12 @@ def find_open_restaurants_by_location(location: str,
         search_results = search_tool(query)  
 
         if not search_results:
-            return f"❌ No matching restaurants found in {location} at {local_time}."
+            return f"❌ No matching restaurants found in {location} at {local_time} fulfilling desired requests."
 
         # Ensure search_results is a list before iterating    
         if isinstance(search_results, list) and len(search_results) > 0:
             results_str = "\n".join(
-                [f"{result.get('title', 'Unknown')} - {result.get('url', 'No URL')}" for result in search_results]
+                [f"{result.get('title', 'Unknown')} - {result.get('url', 'No URL')}" for result in search_results[:number_of_options]]
             )
         else:
             results_str = str(search_results)  # Convert to string if it's not a list