MDX RAG 搜索

`MDXSearchTool`

描述

MDX Search Tool 是 crewai_tools 包中的一个组件，旨在帮助进行高级 Markdown 语言提取。它让用户能够通过基于查询的搜索，高效地在 MD 文件中查找并提取相关信息。这个工具对于数据分析、信息管理和研究任务都非常有价值，可简化在大型文档集合中查找特定信息的过程。

安装

在使用 MDX Search Tool 之前，请确保已经安装了 crewai_tools 包。如果尚未安装，可以使用以下命令：

pip install 'crewai[tools]'

使用示例

要使用 MDX Search Tool，你首先需要设置必要的环境变量。然后，将该工具集成到你的 crewAI 项目中即可开始市场研究。下面是一个基本示例：

from crewai_tools import MDXSearchTool

# 初始化工具，以搜索执行过程中了解到的任何 MDX 内容
tool = MDXSearchTool()

# 或者

# 使用特定的 MDX 文件路径初始化工具，
# 以便仅在该文档内进行专属搜索
tool = MDXSearchTool(mdx='path/to/your/document.mdx')

参数

mdx：可选。指定用于搜索的 MDX 文件路径。可在初始化时提供。

模型与嵌入自定义

默认情况下，该工具使用 OpenAI 进行嵌入和摘要。若要自定义，可以按如下方式使用配置字典：

from chromadb.config import Settings

tool = MDXSearchTool(
    config={
        "embedding_model": {
            "provider": "openai",
            "config": {
                "model": "text-embedding-3-small",
                # "api_key": "sk-...",
            },
        },
        "vectordb": {
            "provider": "chromadb",  # 或 "qdrant"
            "config": {
                # "settings": Settings(persist_directory="/content/chroma", allow_reset=True, is_persistent=True),
                # from qdrant_client.models import VectorParams, Distance
                # "vectors_config": VectorParams(size=384, distance=Distance.COSINE),
            }
        },
    }
)