XMLSearchTool

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

描述

XMLSearchTool 是一款前沿的 RAG 工具,专为在 XML 文件中进行语义搜索而设计。 对于需要高效解析和提取 XML 内容信息的用户来说,此工具非常理想,它支持输入搜索查询和可选的 XML 文件路径。 通过指定 XML 路径,用户可以更精确地将搜索目标锁定在该文件的内容上,从而获得更相关的搜索结果。

安装

要开始使用XMLSearchTool,您首先需要安装crewai_tools包。这可以通过以下命令轻松完成:

pip install 'crewai[tools]'

示例

以下是两个演示如何使用XMLSearchTool的示例。 第一个示例展示了在特定XML文件中进行搜索,而第二个示例则说明了如何在不预定义XML路径的情况下启动搜索,从而提供搜索范围的灵活性。

Code
from crewai_tools import XMLSearchTool

# Allow agents to search within any XML file's content 
#as it learns about their paths during execution
tool = XMLSearchTool()

# OR

# Initialize the tool with a specific XML file path 
#for exclusive search within that document
tool = XMLSearchTool(xml='path/to/your/xmlfile.xml')

参数

  • xml: 这是您希望搜索的XML文件的路径。 在工具初始化期间,这是一个可选参数,但必须在初始化时或作为run方法参数的一部分提供,以执行搜索。

自定义模型和嵌入

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

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

这个页面有帮助吗?