跳至内容

任务交接

交接功能允许一个代理将任务委托给另一个代理。这在需要不同代理专注于各自专业领域的场景中特别有用。例如,一个客户支持应用可能拥有专门处理订单状态、退款、常见问题等不同任务的代理。

交接过程被表示为LLM的工具。因此,如果有一个名为Refund Agent的代理进行交接,该工具将被命名为transfer_to_refund_agent

创建交接

所有代理都有一个handoffs参数,它可以接收一个Agent直接传入,或者一个自定义交接的Handoff对象。

您可以使用Agents SDK提供的handoff()函数创建交接。该函数允许您指定要交接的代理,以及可选的重写和输入过滤器。

基本用法

以下是创建一个简单交接的方法:

from agents import Agent, handoff

billing_agent = Agent(name="Billing agent")
refund_agent = Agent(name="Refund agent")

# (1)!
triage_agent = Agent(name="Triage agent", handoffs=[billing_agent, handoff(refund_agent)])
  1. 您可以直接使用代理(如billing_agent所示),也可以使用handoff()函数。

通过handoff()函数自定义交接流程

handoff() 函数允许您自定义设置。

  • agent: 这是任务将被移交的代理。
  • tool_name_override: 默认情况下,会使用Handoff.default_tool_name()函数,该函数解析为transfer_to_。您可以覆盖此设置。
  • tool_description_override: 覆盖来自Handoff.default_tool_description()的默认工具描述
  • on_handoff: 当触发转接时执行的回调函数。这在需要立即启动某些数据获取操作时非常有用。该函数接收代理上下文,也可以选择性地接收LLM生成的输入。输入数据由input_type参数控制。
  • input_type: 交接操作期望的输入类型(可选)。
  • input_filter: 这允许您过滤下一个代理接收的输入。详见下文。
from agents import Agent, handoff, RunContextWrapper

def on_handoff(ctx: RunContextWrapper[None]):
    print("Handoff called")

agent = Agent(name="My agent")

handoff_obj = handoff(
    agent=agent,
    on_handoff=on_handoff,
    tool_name_override="custom_handoff_tool",
    tool_description_override="Custom description",
)

交接输入

在某些情况下,您希望LLM在调用转接时提供一些数据。例如,想象一个转接到"升级代理"的场景。您可能希望提供原因,以便可以记录它。

from pydantic import BaseModel

from agents import Agent, handoff, RunContextWrapper

class EscalationData(BaseModel):
    reason: str

async def on_handoff(ctx: RunContextWrapper[None], input_data: EscalationData):
    print(f"Escalation agent called with reason: {input_data.reason}")

agent = Agent(name="Escalation agent")

handoff_obj = handoff(
    agent=agent,
    on_handoff=on_handoff,
    input_type=EscalationData,
)

输入过滤器

当发生交接时,新的代理会接管对话,并能够查看整个先前的对话历史记录。如果您想改变这一点,可以设置一个input_filter。输入过滤器是一个函数,它通过HandoffInputData接收现有输入,并且必须返回一个新的HandoffInputData

有一些常见模式(例如从历史记录中移除所有工具调用),这些已在agents.extensions.handoff_filters中为您实现

from agents import Agent, handoff
from agents.extensions import handoff_filters

agent = Agent(name="FAQ agent")

handoff_obj = handoff(
    agent=agent,
    input_filter=handoff_filters.remove_all_tools, # (1)!
)
  1. 当调用FAQ agent时,这将自动从历史记录中移除所有工具。

为确保LLMs能正确理解任务交接,我们建议在您的agents中包含关于交接的信息。我们在agents.extensions.handoff_prompt.RECOMMENDED_PROMPT_PREFIX中提供了一个建议的前缀,或者您可以调用agents.extensions.handoff_prompt.prompt_with_handoff_instructions来自动将建议的数据添加到您的提示中。

from agents import Agent
from agents.extensions.handoff_prompt import RECOMMENDED_PROMPT_PREFIX

billing_agent = Agent(
    name="Billing agent",
    instructions=f"""{RECOMMENDED_PROMPT_PREFIX}
    <Fill in the rest of your prompt here>.""",
)