跳到主要内容

YoutubeVideoSearchTool

!!! note "实验性质" 我们仍在努力改进工具，因此未来可能会出现意外行为或更改。

描述​

该工具是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内容中进行通用搜索，因此代理可以在其操作期间学习到任何Youtube视频内容的URL
tool = YoutubeVideoSearchTool()

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

参数​

YoutubeVideoSearchTool接受以下初始化参数：

  • youtube_video_url：初始化时的可选参数，但如果要针对特定的Youtube视频进行搜索，则为必需参数。它指定您要在其中进行搜索的Youtube视频URL路径。

自定义模型和嵌入​

默认情况下，该工具使用OpenAI进行嵌入和摘要生成。要自定义模型，您可以使用以下配置字典：

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",
            ),
        ),
    )
)