Modified files

agent.py

@@ -119,7 +119,13 @@ def initialize_agent():
     @tool
     @retry
     def query_webpage(url: str, query: str) -> str:
-        """Extracts specific information from a webpage by asking a targeted question."""
+        """
+        Extracts specific information from a webpage by asking a targeted question.
+
+        Args:
+            url (str): The full URL of the webpage to query.
+            query (str): The specific question to ask about the webpage's content.
+        """
         logging.info(f"📄 Querying webpage: {url}")
         loader = download_loader("BeautifulSoupWebReader")()
         docs = loader.load_data(urls=[url])
@@ -132,7 +138,13 @@ def initialize_agent():
     @tool
     @retry
     def query_youtube_video(video_url: str, query: str) -> str:
-        """Extracts specific information from a YouTube video transcript."""
+        """
+        Extracts specific information from a YouTube video transcript.
+
+        Args:
+            video_url (str): The full URL of the YouTube video.
+            query (str): The specific question to ask about the video's content.
+        """
         logging.info(f"🎬 Querying YouTube video: {video_url}")
         video_id_match = re.search(r'(?:v=|\/)([a-zA-Z0-9_-]{11}).*', video_url)
         if not video_id_match: return "Error: Invalid YouTube URL."
@@ -152,7 +164,7 @@ def initialize_agent():
         Searches Wikipedia for a given query and returns a summary.
 
         Args:
-        query (str): The term or question to search for on Wikipedia.
+            query (str): The term or question to search for on Wikipedia.
         """
         try:
             return wikipedia.summary(query, sentences=5)
@@ -175,7 +187,6 @@ def initialize_agent():
 
     google_search_tool = GoogleSearchTool(provider='serpapi', serpapi_api_key=api_keys['serpapi']) if api_keys['serpapi'] else None
 
-    # LOGICAL FIX: Create a single, powerful CodeAgent with all necessary tools.
     tools_list = [tool for tool in [google_search_tool, query_webpage, query_youtube_video, wikipedia_search] if tool]
     
     agent = CodeAgent(
@@ -208,8 +219,9 @@ def initialize_agent():
         *Example 2: Multi-step question involving web search*
         ```python
         # Find the birth date of the author of 'Pride and Prejudice'
-        author_name = GoogleSearchTool(query="author of Pride and Prejudice")
+        author_name_info = GoogleSearchTool(query="author of Pride and Prejudice")
         # Let's assume the tool returns "Jane Austen"
+        # Now get the birth date from Wikipedia
         birth_date_info = wikipedia_search(query="Jane Austen birth date")
         print(birth_date_info)
         ```