Pydantic-AI框架中Agent间消息传递的解决方案

在基于Pydantic-AI框架开发多Agent系统时，开发者经常会遇到Agent间消息传递的问题。本文深入分析这一技术难题，并提供经过验证的解决方案。

问题背景

在多Agent系统架构中，一个Agent（如酒店预订Agent）需要将任务委托给另一个Agent（如报价查询Agent）时，消息历史的传递会出现两个关键问题：

1. 未完成的工具调用链：当主Agent调用工具将任务委托给子Agent时，这个工具调用本身会留在消息历史中，但缺少对应的工具返回响应，导致API调用失败。

2. 系统提示冲突：消息历史中保留的是主Agent的系统提示，这会覆盖子Agent的系统提示，导致子Agent无法按照预期行为执行。

技术分析

通过分析Pydantic-AI框架的消息处理机制，我们发现：

1. 消息历史(ModelMessage列表)会完整记录所有交互，包括未完成的工具调用
2. 系统提示(SystemPromptPart)作为消息的一部分，会随消息历史一起传递
3. OpenAI等模型API对工具调用的完整性有严格校验

解决方案

我们开发了一个通用的消息预处理函数，可以解决上述两个问题：

async def prepare_messages_for_agent_delegation(
    target_agent: Agent,
    messages: List[ModelMessage],
) -> List[ModelMessage]:
    """
    为Agent委托准备消息历史：
    1. 移除未完成的工具调用（最后一条如果是工具调用则删除）
    2. 替换系统提示为目标Agent的系统提示
    """
    if not messages:
        return []
    messages = deepcopy(messages)

    # 移除最后一条未完成的工具调用
    if len(messages) > 1 and messages[-1].parts[-1].part_kind == "tool-call":
        messages = messages[:-1]

    # 替换系统提示
    if isinstance(messages[0], ModelRequest):
        ctx = RunContext(deps=None, model=target_agent.model, usage=Usage(), prompt="required")
        new_sys_parts = await target_agent._sys_parts(ctx)
        
        non_sys_parts = [part for part in messages[0].parts if not isinstance(part, SystemPromptPart)]
        messages[0] = ModelRequest(parts=new_sys_parts + non_sys_parts)

    return messages

使用示例

async def delegate_to_agent2(ctx: RunContext):
    messages = await prepare_messages_for_agent_delegation(
        target_agent=agent2,
        messages=ctx.messages,
    )
    response = await agent2.run(
        user_prompt="",  # 可留空或传入特定提示
        deps=ctx.deps,
        message_history=messages,
    )
    return response

最佳实践

1. 消息历史处理：在Agent委托时，务必处理消息历史，移除未完成的工具调用
2. 系统提示隔离：确保每个Agent使用自己的系统提示
3. 版本兼容性：该解决方案已在pydantic-ai 0.0.21版本验证可用
4. 性能考虑：消息历史的深拷贝操作在大型对话历史中可能影响性能，建议按需使用

总结

Pydantic-AI框架中的Agent间协作是一个强大但需要谨慎处理的功能。通过本文提供的解决方案，开发者可以避免常见的消息传递陷阱，构建更加健壮的多Agent系统。随着框架的迭代，我们期待官方能提供更优雅的内置解决方案。

对于更复杂的场景，建议考虑扩展RunContext类或开发自定义中间件来统一处理Agent间的消息传递问题。