Update tools/web_search_tools.py

tools/web_search_tools.py

@@ -6,6 +6,15 @@ from langchain.tools import tool
 class WebSearchTools:
     @tool("Search the internet")
     def search_internet(query):
+        """
+        Search the internet for the given query and return a list of search results with title, link, and snippet.
+        
+        Args:
+            query (str): The search query.
+        
+        Returns:
+            list: A list of dictionaries containing the title, link, and snippet of each search result.
+        """
         search_url = f"https://www.google.com/search?q={query}&tbm=nws"
         headers = {
             "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
@@ -26,6 +35,15 @@ class WebSearchTools:
 
     @tool("Scrape website content")
     def scrape_and_summarize_website(url):
+        """
+        Scrape the given website URL and return the extracted text content. Download images and save them locally.
+        
+        Args:
+            url (str): The website URL to scrape.
+        
+        Returns:
+            str: The extracted text content from the website.
+        """
         response = requests.get(url)
         soup = BeautifulSoup(response.text, 'html.parser')
         # Extract and summarize text
@@ -45,8 +63,16 @@ class WebSearchTools:
 
     @tool("Download image")
     def download_image(image_url, save_path):
+        """
+        Download an image from the given URL and save it to the specified path.
+        
+        Args:
+            image_url (str): The URL of the image to download.
+            save_path (str): The path where the image will be saved.
+        
+        Returns:
+            None
+        """
         img_data = requests.get(image_url).content
         with open(save_path, 'wb') as img_file:
-            img_file.write(img_data)
-
-
+            img_file.write(img_data)