跳到主要内容

CSVSearchTool

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

描述​

该工具用于在 CSV 文件内容中执行 RAG（检索增强生成）搜索。它允许用户在指定的 CSV 文件内容中语义搜索查询。这个功能特别适用于从大型 CSV 数据集中提取信息，传统的搜索方法可能效率低下。所有带有“搜索”名称的工具，包括 CSVSearchTool，在设计上都是用于搜索不同数据源的 RAG 工具。

安装​

安装 crewai_tools 包

pip install 'crewai[tools]'

示例​

from crewai_tools import CSVSearchTool

# 使用特定的 CSV 文件初始化工具。这样设置允许代理只搜索给定的 CSV 文件。
tool = CSVSearchTool(csv='path/to/your/csvfile.csv')

# 或者

# 在没有特定 CSV 文件的情况下初始化工具。代理将需要在运行时提供 CSV 路径。
tool = CSVSearchTool()

参数​

  • csv：要搜索的 CSV 文件的路径。如果工具在没有特定 CSV 文件的情况下初始化，则这是一个必需的参数；否则，它是可选的。

自定义模型和嵌入​

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

tool = CSVSearchTool(
    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",
            ),
        ),
    )
)