LLamaIndex 多Agent工作流配合使用

在此之前一直在使用创建单个代理。但也旨在支持多代理系统，其中多个代理协作完成您的任务，并根据需要将控制权移交给彼此。AgentWorkflow

我们的系统将有三个代理：

• A 将在 Web 上搜索有关给定主题的信息。ResearchAgent
• A 将使用 ResearchAgent 找到的信息编写报告。WriteAgent
• A 将审阅报告并提供反馈。ReviewAgent

 创建一个多代理系统，该系统将按顺序执行这些代理。有很多方法可以构建一个系统来执行这项任务。在此示例中，将使用一些工具来帮助研究和编写过程。

• 一个在 Web 上搜索有关给定主题的信息的工具（我们将使用 Tavily，就像我们在前面的示例中所做的那样）web_search
• 一个将在 Web 上找到的研究保存到 state 以便其他工具可以使用它的工具（参见 state management 来提醒自己这是如何运作的）record_notes
• 一个工具，用于使用write_reportResearchAgent
• 用于查看报告和提供反馈的工具。review_report

利用 Context 类，我们可以在 Agent 之间传递状态，并且每个 Agent 都可以访问系统的当前状态。

创建第一个代理，它将使用该工具在 Web 上搜索信息，并使用该工具将这些注释保存到状态以供其他代理使用。这里需要注意的关键语法元素是： * 的 ，用于向其他代理标识代理，我们稍后会看到 * 其他代理使用它来决定将控制权移交给下一个谁 * 的 ，用于定义代理的行为 * 是代理可以将其控制权交给的代理名称的可选列表。默认情况下，它将能够将控制权交给任何其他代理。

# 定义一个函数，模拟搜索函数
async def search_web(query: str) -> dict[str, str]:
    """根据查询返回相关信息。"""
    print(f"调用search: {query}")
    return {
        "query": query,
        "result": """
        Python 是一种解释型、面向对象、动态数据类型的高级程序设计语言。
        Python 由 Guido van Rossum 于 1989 年底发明，第一个公开发行版发行于 1991 年。
        像 Perl 语言一样, Python 源代码同样遵循 GPL(GNU General Public License) 协议。
        官方宣布，2020 年 1 月 1 日， 停止 Python 2 的更新。
        Python 2.7 被确定为最后一个 Python 2.x 版本。
        """
    }

 

research_agent = FunctionAgent(
        name="ResearchAgent",