Strategies

Execution strategies control how an agent iterates between LLM calls and tool execution. The default is ReActStrategy. Swap to a different strategy to change reasoning behaviour without touching agent code.

from sagewai import UniversalAgent, TreeOfThoughtsStrategy

agent = UniversalAgent(
    name="reasoner",
    model="gpt-4o",
    strategy=TreeOfThoughtsStrategy(branches=3, max_depth=2),
)

ExecutionStrategy (Protocol)

The protocol all strategies must satisfy. Implement this to create custom strategies.

from sagewai import ExecutionStrategy

class MyStrategy:
    async def execute(
        self,
        agent: BaseAgent,
        messages: list[ChatMessage],
        tools: list[ToolSpec],
        max_iterations: int,
    ) -> ChatMessage:
        ...

Required Method

ReActStrategy

Reason-Act-Observe loop. The default strategy for all agents.

1. Call the LLM with the current conversation and available tools.
2. If the response contains tool calls, execute them and loop back.
3. If the response is pure text, return it.
4. If max_iterations is exhausted, return a guard message.

from sagewai import ReActStrategy

strategy = ReActStrategy(
    max_tool_calls_per_name=3,
    max_error_streak=2,
)

Constructor

TreeOfThoughtsStrategy

Parallel branch exploration with self-evaluation scoring and pruning.

At each depth level: generate parallel reasoning paths, score via LLM self-evaluation, prune low-scoring branches, and continue the best path.

from sagewai import TreeOfThoughtsStrategy

strategy = TreeOfThoughtsStrategy(
    branches=3,
    max_depth=2,
    top_k=1,
)

Constructor

LATSStrategy

Language Agent Tree Search -- MCTS-inspired search over agent reasoning trajectories with tool use, LLM self-evaluation, and reflective backtracking.

At each step: Select the most promising node via UCT, Expand by generating candidate actions, Evaluate with LLM scoring, and Backpropagate scores up the tree.

from sagewai import LATSStrategy

strategy = LATSStrategy(
    n_samples=3,
    max_depth=4,
    max_iterations=8,
)

Constructor

PlanningStrategy

Decompose goals into subtasks, then execute step by step.

Two modes:

• plan_then_act: Generate plan once, execute all steps sequentially.
• plan_act_reflect: After each step, optionally revise the remaining plan.

from sagewai import PlanningStrategy

strategy = PlanningStrategy(
    mode="plan_act_reflect",
    max_steps=10,
)

Constructor

SelfCorrectionStrategy

Error recovery with failure exemplars. Wraps a base strategy and intercepts validation errors to re-prompt the LLM with correction context (PALADIN-style 1-shot correction).

from sagewai import SelfCorrectionStrategy

strategy = SelfCorrectionStrategy(
    max_corrections=2,
)

Constructor

Related Classes

OutputValidator -- validates LLM outputs against expected schemas:

from sagewai.core.self_correction import OutputValidator

validator = OutputValidator()
validator.add_json_validator(required_fields=["title", "body"])
validator.add_custom_validator("length", lambda text: None if len(text) > 10 else ValueError("Too short"))
errors = validator.validate(text)  # Returns list[str], empty = valid

ExemplarStore -- stores failure-correction pairs for 1-shot prompting:

from sagewai.core.self_correction import ExemplarStore, FailureExemplar

store = ExemplarStore()
store.add(FailureExemplar(
    error_type="invalid_json",
    bad_output="not json",
    correction_prompt="Respond with valid JSON",
    corrected_output='{"title": "Fixed"}',
))

RoutingStrategy

Classify user intent and dispatch to a specialist agent.

Two routing methods:

• "heuristic": keyword matching (cheap, deterministic).
• "llm": asks the host agent's LLM to classify the intent.

from sagewai import RoutingStrategy, UniversalAgent

greeter = UniversalAgent(name="greeter", model="gpt-4o-mini")
researcher = UniversalAgent(name="researcher", model="gpt-4o")

strategy = RoutingStrategy(
    routes={"greet": greeter, "research": researcher},
    fallback=greeter,
    method="heuristic",
    keywords={"greet": ["hello", "hi", "hey"]},
)

Constructor