Updated tools

agent.py

@@ -5,7 +5,6 @@ import json
 import pandas as pd
 from smolagents import (
     CodeAgent, 
-    YouTubeSearchTool,
     LiteLLMModel, 
     DuckDuckGoSearchTool, 
     FinalAnswerTool,
@@ -75,6 +74,24 @@ def read_python_file(file_name: str) -> str:
     base_path = "data/question_files"
     with open(os.path.join(base_path, file_name), "r") as f:
         return f.read()
+@tool
+def get_youtube_transcript(video_id:str)-> str:
+    """
+    Retrieves the transcript for a given YouTube video.
+
+    Args:
+        video_id: The ID of the YouTube video.
+
+    Returns:
+        A list of dictionaries, where each dictionary represents a transcript segment
+        and contains 'text' and 'start' keys, or None if no transcript is found.
+    """
+    try:
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+        return transcript
+    except Exception as e:
+        # print(f"Error getting transcript: {e}")
+        return f"Error getting transcript: {e}"
 
 @tool
 def read_excel_file(file_name: str) -> str:
@@ -106,8 +123,11 @@ agent = CodeAgent(
         WikipediaSearchTool(),
         arxiv_search,
         FinalAnswerTool(),
-        YouTubeSearchTool(),
-        extract_text_from_image
+        extract_text_from_image.
+        read_excel_file,
+        read_python_file,
+        get_youtube_transcript,
+    
     ],
     planning_interval=3,
     max_steps=10,