Add the missing duckdukgo function

agent.py

@@ -1,5 +1,8 @@
 from typing import TypedDict, Annotated
 import os
+import requests
+from urllib.parse import urlparse
+from pathlib import Path
 from langchain_community.tools import DuckDuckGoSearchRun
 from langchain_community.document_loaders import WikipediaLoader, YoutubeLoader
 from langchain_community.document_loaders.youtube import TranscriptFormat
@@ -57,49 +60,6 @@ def analyze_youtube_video(video_url: str) -> str:
     Returns:
         str: The transcript content of the YouTube video
     """
-    # try:
-        # # Method 1: Try with basic YoutubeLoader first
-        # try:
-        #     loader = YoutubeLoader.from_youtube_url(
-        #         video_url,
-        #         add_video_info=True,
-        #         language=["en", "en-US", "en-GB"]  # Try multiple English variants
-        #     )
-        #     docs = loader.load()
-            
-        #     if docs:
-        #         content = ""
-        #         for doc in docs:
-        #             title = doc.metadata.get('title', 'Unknown Video')
-        #             author = doc.metadata.get('author', 'Unknown Author')
-        #             length = doc.metadata.get('length', 'Unknown')
-                    
-        #             content += f"Video Title: {title}\n"
-        #             content += f"Author: {author}\n"
-        #             content += f"Length: {length} seconds\n"
-        #             content += f"Transcript:\n{doc.page_content}\n\n"
-                
-        #         return content
-        # except Exception as e1:
-        #     print(f"Method 1 failed: {e1}")
-            
-            # Method 2: Try without video info
-            # try:
-            #     loader = YoutubeLoader.from_youtube_url(
-            #         video_url,
-            #         add_video_info=False,
-            #         language=["en"]
-            #     )
-            #     docs = loader.load()
-                
-            #     if docs:
-            #         content = f"Video URL: {video_url}\n"
-            #         content += f"Transcript:\n{docs[0].page_content}\n\n"
-            #         return content
-            # except Exception as e2:
-            #     print(f"Method 2 failed: {e2}")
-                
-            #     # Method 3: Try with chunked format
     try:
         loader = YoutubeLoader.from_youtube_url(
             video_url,
@@ -118,6 +78,70 @@ def analyze_youtube_video(video_url: str) -> str:
     except Exception as e:
         print(f"Analyze video failed: {e}")
 
+# Create file download tool
+@tool
+def download_file(url: str, filename: str = None, download_dir: str = "downloaded_files") -> str:
+    """Download a file from a URL and save it locally.
+    
+    Args:
+        url: The URL of the file to download
+        filename: Optional custom filename. If not provided, will extract from URL
+        download_dir: Directory to save the file (default: 'downloaded_files')
+        
+    Returns:
+        str: Success message with file path or error message
+    """
+    try:
+        # Create download directory if it doesn't exist
+        Path(download_dir).mkdir(exist_ok=True)
+        
+        # Parse URL to get filename if not provided
+        if not filename:
+            parsed_url = urlparse(url)
+            filename = os.path.basename(parsed_url.path)
+            if not filename:
+                filename = "downloaded_file"
+        
+        # Full file path
+        file_path = os.path.join(download_dir, filename)
+        
+        # Download the file
+        print(f"Downloading file from: {url}")
+        response = requests.get(url, stream=True, timeout=30)
+        response.raise_for_status()
+        
+        # Get file size for progress tracking
+        total_size = int(response.headers.get('content-length', 0))
+        
+        # Write file in chunks
+        with open(file_path, 'wb') as file:
+            downloaded = 0
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:
+                    file.write(chunk)
+                    downloaded += len(chunk)
+        
+        # Get file info
+        file_size = os.path.getsize(file_path)
+        file_size_mb = file_size / (1024 * 1024)
+        
+        return f"""File downloaded successfully!
+        
+File Details:
+- URL: {url}
+- Saved as: {file_path}
+- File size: {file_size_mb:.2f} MB
+- Content type: {response.headers.get('content-type', 'Unknown')}
+
+You can now access the file at: {os.path.abspath(file_path)}"""
+        
+    except requests.exceptions.RequestException as e:
+        return f"Error downloading file from {url}: Network error - {str(e)}"
+    except OSError as e:
+        return f"Error saving file: File system error - {str(e)}"
+    except Exception as e:
+        return f"Error downloading file from {url}: {str(e)}"
+
 # Initialize Langfuse CallbackHandler globally
 def get_langfuse_handler():
     """Get configured Langfuse handler"""

requirements.txt

@@ -6,4 +6,5 @@ langgraph
 langfuse
 langchain-openai
 youtube-transcript-api
-pytube
+pytube
+duckduckgo-search