Update app.py
Browse files
app.py
CHANGED
@@ -22,14 +22,16 @@ import re
|
|
22 |
|
23 |
from smolagents import Tool
|
24 |
|
|
|
|
|
25 |
class YouTubeTranscriptTool(Tool):
|
26 |
name = "youtube_transcript"
|
27 |
description = "Extracts transcript text from a YouTube video URL"
|
28 |
inputs = {"url": str}
|
29 |
outputs = {"transcript": str}
|
|
|
30 |
|
31 |
def __call__(self, url: str) -> dict:
|
32 |
-
# Example logic (replace with real code)
|
33 |
from youtube_transcript_api import YouTubeTranscriptApi
|
34 |
video_id = url.split("v=")[-1].split("&")[0]
|
35 |
transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
|
|
|
22 |
|
23 |
from smolagents import Tool
|
24 |
|
25 |
+
from smolagents import Tool
|
26 |
+
|
27 |
class YouTubeTranscriptTool(Tool):
|
28 |
name = "youtube_transcript"
|
29 |
description = "Extracts transcript text from a YouTube video URL"
|
30 |
inputs = {"url": str}
|
31 |
outputs = {"transcript": str}
|
32 |
+
output_type = dict # ✅ This line is required!
|
33 |
|
34 |
def __call__(self, url: str) -> dict:
|
|
|
35 |
from youtube_transcript_api import YouTubeTranscriptApi
|
36 |
video_id = url.split("v=")[-1].split("&")[0]
|
37 |
transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
|