Update app.py

Updated search tool due to errors calling the DuckDuckGo api

app.py

@@ -20,22 +20,38 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def search_duckduckgo_documents(query: str, max_results: int = 10) -> List[Dict]:
+def search_duckduckgo_documents(query: str, max_results: int = 10) -> str: 
     """
-    Search DuckDuckGo and retrieve up to 10 documents/results.
+    Search DuckDuckGo and retrieve top N search results.
     
     Args:
         query (str): The search query
         max_results (int): Maximum number of results to return (default: 10)
     
     Returns:
-        List[Dict]: List of search results with title, url, and snippet
+        str: Formatted search results as a string (limited to max_results)
     """
     # Initialize the DuckDuckGo search tool
     search_tool = DuckDuckGoSearchTool()
     
-    # Perform the search
-    results = search_tool(query, max_results=max_results)
+    # Perform the search (DuckDuckGoSearchTool doesn't accept max_results parameter)
+    results = search_tool(query)
+    
+    # Manually limit results by parsing the string output
+    if isinstance(results, str):
+        # Split by lines and try to limit results
+        lines = results.split('\n')
+        print(len(lines))
+        result_count = 0
+        limited_lines = []
+        
+        for line in lines:
+            if result_count >= max_results:
+                break
+            result_count += 1
+            limited_lines.append(line)
+            
+        results = '\n'.join(limited_lines)
     
     return results