Modified files

agent.py

@@ -9,7 +9,6 @@ from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, No
 from dotenv import load_dotenv
 from requests.exceptions import RequestException
 import wikipedia
-from llama_index.core.schema import Document
 from llama_index.readers.web import BeautifulSoupWebReader
 
 from smolagents import (
@@ -73,7 +72,6 @@ def normalize_answer_format(answer: str) -> str:
     answer = answer.strip().rstrip('.')
     is_list = ',' in answer and len(answer.split(',')) > 1
     try:
-        # Check if it can be a number, ignoring commas for list check
         is_numeric = not is_list and float(answer.replace(',', '')) is not None
     except ValueError:
         is_numeric = False
@@ -113,9 +111,9 @@ def initialize_agent():
 
     # --- Tool Definitions for the Agent ---
 
-    @tool
     @retry
     @lru_cache(maxsize=128)
+    @tool
     def get_webpage_content(url: str) -> str:
         """
         Extracts the text content from a single webpage.
@@ -128,12 +126,11 @@ def initialize_agent():
         docs = loader.load_data(urls=[url])
         if not docs or not docs[0].text:
             raise ValueError(f"No content could be extracted from {url}")
-        # Return up to the first 15,000 characters to avoid overwhelming the context window.
         return docs[0].text[:15000]
 
-    @tool
     @retry
     @lru_cache(maxsize=128)
+    @tool
     def get_youtube_transcript(video_url: str) -> str:
         """
         Fetches the full transcript of a YouTube video as a single string.
@@ -150,15 +147,14 @@ def initialize_agent():
         try:
             transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
             transcript_text = ' '.join([t['text'] for t in transcript_list])
-            # Return up to the first 15,000 characters
             return transcript_text[:15000]
         except (TranscriptsDisabled, NoTranscriptFound) as e:
             logging.error(f"Could not retrieve transcript for {video_url}: {e}")
             raise YouTubeTranscriptApiError(f"Transcript not available for video {video_id}.") from e
             
-    @tool
     @retry
     @lru_cache(maxsize=32)
+    @tool
     def wikipedia_search(query: str) -> str:
         """
         Searches Wikipedia for a given query and returns a summary.