YoutubeVideoSearchTool

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

描述

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

安装

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

pip install 'crewai[tools]'

示例

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

Code
from crewai_tools import YoutubeVideoSearchTool

# General search across Youtube content without specifying a video URL, 
# so the agent can search within any Youtube video content 
# it learns about its url during its operation
tool = YoutubeVideoSearchTool()

# Targeted search within a specific Youtube video's content
tool = YoutubeVideoSearchTool(
    youtube_video_url='https://youtube.com/watch?v=example'
)

参数

YoutubeVideoSearchTool 接受以下初始化参数:

  • youtube_video_url: 初始化时的可选参数,但如果目标是特定的Youtube视频,则为必需。它指定您想要搜索的Youtube视频URL路径。

自定义模型和嵌入

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

Code
tool = YoutubeVideoSearchTool(
    config=dict(
        llm=dict(
            provider="ollama", # or google, openai, anthropic, llama2, ...
            config=dict(
                model="llama2",
                # temperature=0.5,
                # top_p=1,
                # stream=true,
            ),
        ),
        embedder=dict(
            provider="google", # or openai, ollama, ...
            config=dict(
                model="models/embedding-001",
                task_type="retrieval_document",
                # title="Embeddings",
            ),
        ),
    )
)

这个页面有帮助吗?