跳转至

Youtube视频搜索工具

实验性

我们仍在改进工具,因此未来可能会有意外行为或变化。

描述

此工具是crewai_tools包的一部分,旨在利用检索增强生成(RAG)技术在Youtube视频内容中进行语义搜索。它是包中利用RAG针对不同来源的几个“搜索”工具之一。YoutubeVideoSearchTool允许在搜索中具有灵活性;用户可以在不指定视频URL的情况下搜索任何Youtube视频内容,或者通过提供其URL来针对特定Youtube视频进行搜索。

安装

要使用YoutubeVideoSearchTool,您必须首先安装crewai_tools包。此包包含YoutubeVideoSearchTool以及其他旨在增强您的数据分析和处理任务的实用工具。通过在终端执行以下命令来安装该包:

pip install 'crewai[tools]'

示例

要将YoutubeVideoSearchTool集成到您的Python项目中,请按照以下示例操作。这展示了如何使用该工具进行通用Youtube内容搜索以及针对特定视频内容的定向搜索。

from crewai_tools import YoutubeVideoSearchTool

# 在不指定视频URL的情况下进行通用搜索,因此代理可以在其操作过程中了解到的任何Youtube视频内容中进行搜索
tool = YoutubeVideoSearchTool()

# 针对特定Youtube视频内容的定向搜索
tool = YoutubeVideoSearchTool(youtube_video_url='https://youtube.com/watch?v=example')

参数

YoutubeVideoSearchTool接受以下初始化参数:

  • youtube_video_url:初始化时的可选参数,但如果针对特定Youtube视频则必需。它指定您想要在其中搜索的Youtube视频URL路径。
    tool = YoutubeVideoSearchTool(
        config=dict(
            llm=dict(
                provider="ollama", # 或 google, openai, anthropic, llama2, ...
                config=dict(
                    model="llama2",
                    # temperature=0.5,
                    # top_p=1,
                    # stream=true,
                ),
            ),
            embedder=dict(
                provider="google", # 或 openai, ollama, ...
                config=dict(
                    model="models/embedding-001",
                    task_type="retrieval_document",
                    # title="Embeddings",
                ),
            ),
        )
    )