agents.planning.p_and_e.enhanced_multi_agent

Enhanced Multi-Agent Base for Plan and Execute patterns.

This module provides an enhanced version of MultiAgent that allows for cleaner configuration with agents, state schema, and branches passed directly.

Classes

EnhancedMultiAgent

Enhanced MultiAgent that accepts agents, state schema, and branches directly.

PlanAndExecuteMultiAgent

Plan and Execute multi-agent using enhanced base.

Module Contents

class agents.planning.p_and_e.enhanced_multi_agent.EnhancedMultiAgent(agents, state_schema=None, branches=None, schema_composition_method='smart', execution_mode='conditional', **kwargs)

Bases: haive.agents.multi.base.MultiAgent

Enhanced MultiAgent that accepts agents, state schema, and branches directly.

Initialize enhanced multi-agent with direct configuration.

Parameters:

  • agents (list[Any]) – List of agents to orchestrate

  • state_schema (type[haive.core.schema.state_schema.StateSchema] | None) – Optional state schema override

  • branches (dict[str, dict[str, Any]] | None) – Branch configuration for conditional routing

  • schema_composition_method (str) – How to compose schemas (“smart”, “shared”, “namespaced”)

  • execution_mode (str) – Execution pattern

  • **kwargs – Additional MultiAgent arguments

setup_multi_agent()

Override to use state schema override if provided.

Return type:

EnhancedMultiAgent

class agents.planning.p_and_e.enhanced_multi_agent.PlanAndExecuteMultiAgent(agents, state_schema=None, **kwargs)

Bases: EnhancedMultiAgent

Plan and Execute multi-agent using enhanced base.

Initialize Plan and Execute multi-agent.

Parameters:

  • agents (list[Any]) – List of [planner, executor, replanner] agents

  • state_schema (type[haive.core.schema.state_schema.StateSchema] | None) – Optional state schema (defaults to PlanExecuteState)

  • **kwargs – Additional arguments

build_custom_graph(graph)

Build the Plan and Execute workflow graph.

Parameters:

graph (haive.core.graph.state_graph.base_graph2.BaseGraph)

Return type:

haive.core.graph.state_graph.base_graph2.BaseGraph