Create a YouTube Tool node

[shhlife]

It'd be great to have a YouTube tool that would let you get information about youtube videos.

from typing import cast

import yt_dlp
from attrs import define
from griptape.artifacts import AudioArtifact
from griptape.structures import Pipeline
from griptape.tasks import AudioTranscriptionTask, BaseTextInputTask


@define
class YoutubeAudioTask(BaseTextInputTask):
    def try_run(self) -> AudioArtifact:
        ydl_opts = {
            "outtmpl": "downloaded_audio.%(ext)s",
        }

        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
            ydl.download([self.input.value])

        with open("downloaded_audio.webm", "rb") as f:
            audio_data = f.read()

        return AudioArtifact(audio_data, format="webm")


pipeline = Pipeline(
    tasks=[
        YoutubeAudioTask(id="youtube"),
        AudioTranscriptionTask(
            lambda task: cast(AudioArtifact, task.parent_outputs["youtube"])
        ),
    ]
)

pipeline.run("https://www.youtube.com/watch?v=TQn2hJeHQbM")