Update tools/recipe_tools.py

tools/recipe_tools.py

@@ -1,43 +1,54 @@
 from typing import Dict, List, Optional
 from smolagents.tools import Tool, tool
-from smolagents import DuckDuckGoSearchTool
-import requests
 from bs4 import BeautifulSoup
 import re
 import json
 
 @tool
-def search_recipe(query: str) -> List[Dict[str, str]]:
-    """Searches for recipes using DuckDuckGo
+def search_recipe(query: str, web_search, visit_webpage) -> List[Dict[str, str]]:
+    """Searches for recipes using web search and retrieves recipe details
     Args:
         query: search query for the recipe (e.g., 'chicken pasta recipe')
+        web_search: DuckDuckGoSearchTool instance
+        visit_webpage: VisitWebPageTool instance
     Returns:
-        List of dictionaries containing recipe titles and URLs
+        List of dictionaries containing recipe information
     """
-    search_tool = DuckDuckGoSearchTool()
-    results = search_tool.search(f"{query} recipe")
+    # Use the web_search tool to find recipes
+    search_results = web_search.forward(f"{query} recipe best rated")
     
-    # Filter and format results
+    # Parse the markdown formatted results to extract URLs
     recipes = []
-    for result in results[:5]:  # Limit to top 5 results
-        if any(domain in result['link'] for domain in ['allrecipes.com', 'food.com', 'foodnetwork.com']):
-            recipes.append({
-                'title': result['title'],
-                'url': result['link']
-            })
-    return recipes
+    lines = search_results.split('\n')
+    for line in lines:
+        if line.startswith('[') and '](' in line and ')' in line:
+            # Extract title and URL from markdown link
+            title = line[1:line.index(']')]
+            url = line[line.index('(')+1:line.index(')')]
+            if any(domain in url.lower() for domain in ['allrecipes.com', 'food.com', 'foodnetwork.com']):
+                recipes.append({
+                    'title': title,
+                    'url': url
+                })
+    
+    # Limit to top 3 recipes
+    return recipes[:3]
 
 @tool
-def parse_recipe(url: str) -> Dict:
+def parse_recipe(url: str, visit_webpage) -> Dict:
     """Extracts recipe information from a webpage
     Args:
         url: URL of the recipe page
+        visit_webpage: VisitWebPageTool instance
     Returns:
         Dictionary containing recipe details (ingredients, instructions)
     """
     try:
-        response = requests.get(url, headers={'User-Agent': 'Mozilla/5.0'})
-        soup = BeautifulSoup(response.text, 'html.parser')
+        # Use visit_webpage tool to get the page content
+        page_content = visit_webpage.forward(url)
+        
+        # Parse the HTML content
+        soup = BeautifulSoup(page_content, 'html.parser')
         
         # Initialize recipe data
         recipe_data = {
@@ -63,7 +74,7 @@ def parse_recipe(url: str) -> Dict:
         return recipe_data
         
     except Exception as e:
-        return {'error': str(e)}
+        return {'error': f"Error parsing recipe: {str(e)}"}
 
 @tool
 def generate_shopping_list(ingredients: List[str]) -> Dict[str, List[str]]: