add youtube transcript tool

app.py

@@ -6,6 +6,7 @@ from smolagents import CodeAgent, OpenAIServerModel, HfApiModel
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
+from tools.youtube_transcript import YoutubeTranscriptTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -19,15 +20,16 @@ class BasicAgent:
         final_answer = FinalAnswerTool()
         visit_webpage = VisitWebpageTool()
         web_search = DuckDuckGoSearchTool()
+        youtube_transcript = YoutubeTranscriptTool()
         # model = OpenAIServerModel(model_id="gpt-4.1", api_key=os.getenv("OPENAI_API_KEY"))
         model = HfApiModel(
             max_tokens=2096,
             temperature=0.5,
-            model_id='Qwen/Qwen2.5-Coder-7B-Instruct', custom_role_conversions=None,
+            model_id='Qwen/Qwen2.5-Coder-32B-Instruct', custom_role_conversions=None,
         )
         self.agent = CodeAgent(
             model=model,
-            tools=[visit_webpage, web_search, final_answer],
+            tools=[visit_webpage, web_search, final_answer, youtube_transcript],
             max_steps=5,
             verbosity_level=1,
             add_base_tools=True
@@ -40,7 +42,7 @@ class BasicAgent:
             [YOUR FINAL ANSWER] should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
             If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
             If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
-            If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+            If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string and return a string that contains the comma separated values.
             IMPORTANT NOTES TO LIMIT COSTS AND PREVENT ERRORS:
             - Use web search sparingly and only when absolutely necessary.
             - Limit to 1-2 web searches per question.

requirements.txt

@@ -6,3 +6,4 @@ smolagents[openai]
 requests
 duckduckgo_search
 pandas
+youtube-transcript-api

tools/youtube_transcript.py

@@ -0,0 +1,27 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import duckduckgo_search
+
+class YoutubeTranscriptTool(Tool):
+    name = "youtube_transcript"
+    description = "This tool allows you to retrieve the transcript/subtitles for a given YouTube video."
+    inputs = {'video_id': {'type': 'string', 'description': 'The Id of YouTube video'}}
+    output_type = "list[dict]"
+
+    def __init__(self, max_results=10, **kwargs):
+        super().__init__()
+        self.max_results = max_results
+        try:
+            from youtube_transcript_api import YouTubeTranscriptApi
+        except ImportError as e:
+            raise ImportError(
+                "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
+            ) from e
+        self.ytt_api = YouTubeTranscriptApi()
+
+    def forward(self, video_id: str) -> list[dict] | str:
+        try:
+            result = self.ytt_api.fetch(video_id)
+            return result.to_raw_data()
+        except Exception as e:
+            return f"An unexpected error occurred: {str(e)}"