fix(run): address gadfly review of the phases PR
executus CI / test (pull_request) Successful in 48s

Real findings from the consensus review (37 raw; many devstral dups/noise):

- Optional/budget-salvage branches no longer swallow a context
  cancellation / deadline / critic-kill: such errors return immediately so
  the run is classified cancelled/timeout/killed, not "ok" with a fallback.
  (the most serious finding — an Optional final phase could mask a killed run)
- IsRunFunc bare phase now feeds the SHARED step observer (not just the
  audit recorder), so the critic's activity clock + Result.Steps see it —
  a long synthesize phase no longer looks idle to the critic.
- phaseModel returns the resolver's enriched (usage-attribution) context and
  the phase's calls use it, mirroring the single-loop path (non-base-tier
  phases were mis-attributed).
- salvagePhaseTranscript trims the tail on a rune boundary (was a raw byte
  slice that could split a UTF-8 rune); maxSalvage is now a named const with
  rationale.
- expandPhaseTemplate logs a WARN on parse/execute failure instead of
  silently returning the unexpanded template; documented the phase-name
  identifier requirement + the "Query" shadow.
- removed the dead phaseDeps.baseTier field.
- extracted multimodalUserMessage, shared by runAgent + the phase runner
  (was duplicated image-folding).
- aggregated phase usage is stamped onto the result even on a hard-error
  return; TrimSpace computed once; filterToolbox returns the base toolbox
  as-is for the empty-names (full-palette) case instead of copying;
  phaseModel WARN no longer prints error=<nil>.

New test: Optional phase does not swallow a cancellation. Full ./... green.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-06-29 15:44:04 -04:00
parent 30b79a330f
commit 0dd2ced717
3 changed files with 151 additions and 87 deletions
+10 -17
View File
@@ -4,7 +4,6 @@ import (
"context" "context"
"errors" "errors"
"fmt" "fmt"
"strings"
"time" "time"
"gitea.stevedudenhoeffer.com/steve/majordomo/agent" "gitea.stevedudenhoeffer.com/steve/majordomo/agent"
@@ -344,13 +343,13 @@ func (e *Executor) Run(ctx context.Context, ra RunnableAgent, inv tool.Invocatio
// (Per-phase step caps are fixed — the critic's dynamic ceiling is not // (Per-phase step caps are fixed — the critic's dynamic ceiling is not
// propagated to phases — but its steer + hard deadline still apply.) // propagated to phases — but its steer + hard deadline still apply.)
runRes, runErr = e.runPhases(runCtx, ra, phaseDeps{ runRes, runErr = e.runPhases(runCtx, ra, phaseDeps{
baseModel: model, baseModel: model,
baseTier: tier, baseToolbox: toolbox,
baseToolbox: toolbox, baseMaxIter: maxIter,
baseMaxIter: maxIter, sharedOpts: sharedOpts,
sharedOpts: sharedOpts, stepObserver: stepObserver,
steer: steer, steer: steer,
rec: rec, rec: rec,
}, input, inv.Images) }, input, inv.Images)
} }
@@ -492,15 +491,9 @@ func runAgent(ctx context.Context, ag *agent.Agent, input string, images []llm.I
if len(images) == 0 { if len(images) == 0 {
return ag.Run(ctx, input, opts...) return ag.Run(ctx, input, opts...)
} }
parts := make([]llm.Part, 0, len(images)+1)
if strings.TrimSpace(input) != "" {
parts = append(parts, llm.Text(input))
}
for _, img := range images {
parts = append(parts, img)
}
// Copy opts before appending so a caller-supplied backing array is never // Copy opts before appending so a caller-supplied backing array is never
// mutated/aliased (the variadic slice can have spare capacity). // mutated/aliased (the variadic slice can have spare capacity). The multimodal
opts = append(opts[:len(opts):len(opts)], agent.WithHistory([]llm.Message{llm.UserParts(parts...)})) // opening turn (text + image parts) is built by the shared helper.
opts = append(opts[:len(opts):len(opts)], agent.WithHistory([]llm.Message{multimodalUserMessage(input, images)}))
return ag.Run(ctx, "", opts...) return ag.Run(ctx, "", opts...)
} }
+116 -70
View File
@@ -8,6 +8,7 @@ import (
"log/slog" "log/slog"
"strings" "strings"
"text/template" "text/template"
"unicode/utf8"
"gitea.stevedudenhoeffer.com/steve/majordomo/agent" "gitea.stevedudenhoeffer.com/steve/majordomo/agent"
"gitea.stevedudenhoeffer.com/steve/majordomo/llm" "gitea.stevedudenhoeffer.com/steve/majordomo/llm"
@@ -24,17 +25,17 @@ import (
// carries RunnableAgent.Phases as a DTO — actually EXECUTES them (it previously // carries RunnableAgent.Phases as a DTO — actually EXECUTES them (it previously
// ignored the slice and ran a single loop with the base prompt). It reuses the // ignored the slice and ran a single loop with the base prompt). It reuses the
// shared run machinery built once in Run: the same stepObserver (so audit/steps/ // shared run machinery built once in Run: the same stepObserver (so audit/steps/
// critic accumulate across every phase), the same critic steer, and the same // critic-activity accumulate across every phase, including IsRunFunc bare calls),
// compaction option. // the same critic steer, and the same compaction option.
// //
// Semantics preserved from mort's pipeline: // Semantics preserved from mort's pipeline:
// - phases run sequentially; ctx cancellation between phases aborts the run. // - phases run sequentially; ctx cancellation/deadline/critic-kill aborts the
// run (even mid-phase and even for an Optional phase).
// - IsRunFunc = one bare LLM call, no tools, no loop. // - IsRunFunc = one bare LLM call, no tools, no loop.
// - Optional phases swallow errors and substitute FallbackMessage. // - Optional phases swallow NON-context errors and substitute FallbackMessage.
// - a non-optional phase that merely exhausts its step/tool budget is NOT fatal: // - a non-optional phase that merely exhausts its step/tool budget is NOT fatal:
// its partial transcript is salvaged and the pipeline continues (so a long // its partial transcript is salvaged and the pipeline continues — EXCEPT a
// verify phase never discards every earlier phase's work). A hard error // final phase that salvaged nothing, which is a genuine empty-result failure.
// (cancellation, model failure) still aborts.
// - per-phase ModelTier resolve failures fall back to the base model with a WARN. // - per-phase ModelTier resolve failures fall back to the base model with a WARN.
// //
// Deliberately NOT carried over (kernel is leaner than mort's legacy pipeline): // Deliberately NOT carried over (kernel is leaner than mort's legacy pipeline):
@@ -42,35 +43,56 @@ import (
// no-tool-call-is-final-answer termination, like its single-loop path), and the // no-tool-call-is-final-answer termination, like its single-loop path), and the
// critic's dynamic iteration ceiling (per-phase caps are fixed at phase start — // critic's dynamic iteration ceiling (per-phase caps are fixed at phase start —
// the run-level critic's steer + hard deadline still apply across phases). // the run-level critic's steer + hard deadline still apply across phases).
//
// NOTE on phase names: {{.<PhaseName>}} resolves a map key, so a phase whose name
// is not a Go-template identifier (hyphens, spaces, leading digit) cannot be
// referenced as {{.my-phase}} — authors must use {{index . "my-phase"}}. A
// template that fails to parse/execute is logged (WARN) and passed through
// unchanged rather than silently dropped (see expandPhaseTemplate). Avoid naming
// a phase "Query" — it shadows the original-input variable.
// phaseDeps carries the per-run state the phase runner shares with Run: the base // phaseDeps carries the per-run state the phase runner shares with Run: the base
// model + tier, the full decorated toolbox (filtered per phase), the base step // model, the full decorated toolbox (filtered per phase), the base step cap, the
// cap, the shared agent options (tool-error limits + step observer + compactor), // shared agent options (tool-error limits + step observer + compactor), the
// the critic/session steer, and the audit recorder (phase events). // shared step observer (also fed by IsRunFunc bare calls), the critic/session
// steer, and the audit recorder (phase events).
type phaseDeps struct { type phaseDeps struct {
baseModel llm.Model baseModel llm.Model
baseTier string baseToolbox *llm.Toolbox
baseToolbox *llm.Toolbox baseMaxIter int
baseMaxIter int sharedOpts []agent.Option
sharedOpts []agent.Option stepObserver func(agent.Step)
steer func() []llm.Message steer func() []llm.Message
rec RunRecorder rec RunRecorder
} }
// runPhases executes ra.Phases sequentially and returns a synthetic agent.Result // runPhases executes ra.Phases sequentially and returns a synthetic agent.Result
// whose Output is the final phase's output, with Usage aggregated across phases // whose Output is the final phase's output, with Usage aggregated across phases
// and Messages set to the last phase's transcript (for the PostRun hook). A hard // and Messages set to the last phase's transcript (for the PostRun hook). A hard
// (non-optional, non-budget) phase failure returns the error. // (non-optional, non-budget) phase failure — and any context cancellation/
// deadline/critic-kill — returns the error.
func (e *Executor) runPhases(runCtx context.Context, ra RunnableAgent, deps phaseDeps, query string, images []llm.ImagePart) (*agent.Result, error) { func (e *Executor) runPhases(runCtx context.Context, ra RunnableAgent, deps phaseDeps, query string, images []llm.ImagePart) (*agent.Result, error) {
outputs := make(map[string]string, len(ra.Phases)) outputs := make(map[string]string, len(ra.Phases))
var lastResult *agent.Result var lastResult *agent.Result
var lastOutput string var lastOutput string
var totalUsage llm.Usage var totalUsage llm.Usage
// finish stamps the aggregated usage + final output onto the synthetic result.
finish := func(err error) (*agent.Result, error) {
if lastResult == nil {
lastResult = &agent.Result{}
}
lastResult.Usage = totalUsage
if err == nil {
lastResult.Output = lastOutput
}
return lastResult, err
}
for i, phase := range ra.Phases { for i, phase := range ra.Phases {
// A killed/timed-out/cancelled run must not start its next phase. // A killed/timed-out/cancelled run must not start its next phase.
if err := runCtx.Err(); err != nil { if err := runCtx.Err(); err != nil {
return lastResult, err return finish(err)
} }
instructions := expandPhaseTemplate(phase.SystemPrompt, query, outputs) instructions := expandPhaseTemplate(phase.SystemPrompt, query, outputs)
@@ -84,7 +106,14 @@ func (e *Executor) runPhases(runCtx context.Context, ra RunnableAgent, deps phas
totalUsage = addUsage(totalUsage, res.Usage) totalUsage = addUsage(totalUsage, res.Usage)
} }
if err != nil { if err != nil {
// A context cancellation / deadline / critic-kill is NEVER swallowed by
// the Optional or budget-salvage branches — the run genuinely ended and
// must surface as cancelled/timeout/killed (statusFor classifies it).
if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
return finish(err)
}
isLast := i == len(ra.Phases)-1 isLast := i == len(ra.Phases)-1
trimmed := strings.TrimSpace(output)
switch { switch {
case phase.Optional: case phase.Optional:
output = phase.FallbackMessage output = phase.FallbackMessage
@@ -97,13 +126,14 @@ func (e *Executor) runPhases(runCtx context.Context, ra RunnableAgent, deps phas
deps.rec.LogEvent("phase_failed_optional", map[string]any{"phase": phase.Name, "error": err.Error()}) deps.rec.LogEvent("phase_failed_optional", map[string]any{"phase": phase.Name, "error": err.Error()})
} }
case isPhaseBudgetExhaustion(err) && (!isLast || strings.TrimSpace(output) != ""): case isPhaseBudgetExhaustion(err) && (!isLast || trimmed != ""):
// Soft stop: the phase ran out of its step/tool budget before // Soft stop: the phase ran out of its step/tool budget before
// composing a final answer. Not fatal — it did real work // composing a final answer. Not fatal — it did real work (runOnePhase
// (runOnePhase salvaged its partial transcript into output), and // salvaged its partial transcript into output), and aborting would
// aborting would discard every completed phase before it. Degrade // discard every completed phase before it. Degrade and continue.
// gracefully and continue. // (A FINAL phase that salvaged nothing falls through to the hard error
if strings.TrimSpace(output) == "" { // below: there is no result to return.)
if trimmed == "" {
output = fmt.Sprintf("(Phase %q reached its step budget before producing a consolidated result; continuing with its partial findings.)", phase.Name) output = fmt.Sprintf("(Phase %q reached its step budget before producing a consolidated result; continuing with its partial findings.)", phase.Name)
} else { } else {
output += fmt.Sprintf("\n\n(Note: phase %q reached its step budget before fully completing; the above is its partial output.)", phase.Name) output += fmt.Sprintf("\n\n(Note: phase %q reached its step budget before fully completing; the above is its partial output.)", phase.Name)
@@ -115,7 +145,7 @@ func (e *Executor) runPhases(runCtx context.Context, ra RunnableAgent, deps phas
} }
default: default:
return lastResult, fmt.Errorf("pipeline phase %q: %w", phase.Name, err) return finish(fmt.Errorf("pipeline phase %q: %w", phase.Name, err))
} }
} }
@@ -123,14 +153,7 @@ func (e *Executor) runPhases(runCtx context.Context, ra RunnableAgent, deps phas
lastOutput = output lastOutput = output
} }
// Synthesize the run result: the final phase's output, usage aggregated over return finish(nil)
// all phases, and the last phase's transcript for the PostRun hook.
if lastResult == nil {
lastResult = &agent.Result{}
}
lastResult.Output = lastOutput
lastResult.Usage = totalUsage
return lastResult, nil
} }
// runOnePhase runs a single phase: a bare LLM call for IsRunFunc phases, a fresh // runOnePhase runs a single phase: a bare LLM call for IsRunFunc phases, a fresh
@@ -138,21 +161,23 @@ func (e *Executor) runPhases(runCtx context.Context, ra RunnableAgent, deps phas
// failed bare call), and any error. On a budget-exhaustion error the loop's // failed bare call), and any error. On a budget-exhaustion error the loop's
// partial transcript is salvaged into the returned output. // partial transcript is salvaged into the returned output.
func (e *Executor) runOnePhase(runCtx context.Context, ra RunnableAgent, deps phaseDeps, phase Phase, instructions, query string, images []llm.ImagePart) (string, *agent.Result, error) { func (e *Executor) runOnePhase(runCtx context.Context, ra RunnableAgent, deps phaseDeps, phase Phase, instructions, query string, images []llm.ImagePart) (string, *agent.Result, error) {
model := e.phaseModel(runCtx, deps, ra, phase) phaseCtx, model := e.phaseModel(runCtx, deps, ra, phase)
// The phase's expanded instructions are the system prompt (with the platform // The phase's expanded instructions are the system prompt (with the platform
// header so tools keep their run ids); the original query is the user message. // header so tools keep their run ids); the original query is the user message.
system := e.systemPromptWithBody(instructions) system := e.systemPromptWithBody(instructions)
if phase.IsRunFunc { if phase.IsRunFunc {
// Bare LLM call: no tool loop, no tools array (some models 400 on an empty // Bare LLM call: no tool loop, no tools array (some models 400 on an empty
// tools list). The response still lands in the audit token tally. // tools list). The response is fed through the SAME step observer as a loop
msgs := []llm.Message{phaseUserMessage(query, images)} // step so the audit token tally, Result.Steps, AND the critic's activity
resp, err := model.Generate(runCtx, llm.Request{System: system, Messages: msgs}) // clock all see it (a long synthesize phase must not look idle to the critic).
msgs := []llm.Message{multimodalUserMessage(query, images)}
resp, err := model.Generate(phaseCtx, llm.Request{System: system, Messages: msgs})
if err != nil { if err != nil {
return "", nil, fmt.Errorf("phase %q model call: %w", phase.Name, err) return "", nil, fmt.Errorf("phase %q model call: %w", phase.Name, err)
} }
if deps.rec != nil { if deps.stepObserver != nil {
deps.rec.OnStep(1, resp) deps.stepObserver(agent.Step{Index: 0, Response: resp})
} }
return resp.Text(), &agent.Result{ return resp.Text(), &agent.Result{
Output: resp.Text(), Output: resp.Text(),
@@ -175,7 +200,7 @@ func (e *Executor) runOnePhase(runCtx context.Context, ra RunnableAgent, deps ph
}, deps.sharedOpts...) }, deps.sharedOpts...)
ag := agent.New(model, system, opts...) ag := agent.New(model, system, opts...)
res, runErr := runAgent(runCtx, ag, query, images, agent.WithSteer(deps.steer)) res, runErr := runAgent(phaseCtx, ag, query, images, agent.WithSteer(deps.steer))
output := "" output := ""
if res != nil { if res != nil {
output = res.Output output = res.Output
@@ -190,19 +215,26 @@ func (e *Executor) runOnePhase(runCtx context.Context, ra RunnableAgent, deps ph
return output, res, runErr return output, res, runErr
} }
// phaseModel resolves the phase's model tier, falling back to the base model on // phaseModel resolves the phase's model tier, returning the resolver's enriched
// an empty tier or a resolution failure (WARN — visible, non-fatal). // context (usage attribution) alongside the model. An empty tier or a resolution
func (e *Executor) phaseModel(ctx context.Context, deps phaseDeps, ra RunnableAgent, phase Phase) llm.Model { // failure falls back to the base model + the run context (WARN — visible, not
// fatal). Returning the enriched ctx mirrors the single-loop path, which adopts
// ctx = modelCtx, so a non-base-tier phase's calls are attributed correctly.
func (e *Executor) phaseModel(ctx context.Context, deps phaseDeps, ra RunnableAgent, phase Phase) (context.Context, llm.Model) {
if phase.ModelTier == "" { if phase.ModelTier == "" {
return deps.baseModel return ctx, deps.baseModel
} }
_, m, err := e.cfg.Models(ctx, phase.ModelTier) modelCtx, m, err := e.cfg.Models(ctx, phase.ModelTier)
if err != nil || m == nil { if err != nil || m == nil {
reason := "resolver returned a nil model"
if err != nil {
reason = err.Error()
}
slog.Warn("run: pipeline phase model resolve failed; using base model", slog.Warn("run: pipeline phase model resolve failed; using base model",
"agent", ra.Name, "phase", phase.Name, "tier", phase.ModelTier, "error", err) "agent", ra.Name, "phase", phase.Name, "tier", phase.ModelTier, "reason", reason)
return deps.baseModel return ctx, deps.baseModel
} }
return m return modelCtx, m
} }
// isPhaseBudgetExhaustion reports whether err is a soft budget/guard stop (the // isPhaseBudgetExhaustion reports whether err is a soft budget/guard stop (the
@@ -212,9 +244,15 @@ func isPhaseBudgetExhaustion(err error) bool {
return errors.Is(err, agent.ErrMaxSteps) || errors.Is(err, agent.ErrToolLoop) return errors.Is(err, agent.ErrMaxSteps) || errors.Is(err, agent.ErrToolLoop)
} }
// maxSalvageBytes bounds a salvaged partial transcript so a long phase's narrated
// reasoning doesn't blow up the next phase's prompt (the tail is the most recent,
// most relevant reasoning). Matches mort's pipeline cap.
const maxSalvageBytes = 8000
// salvagePhaseTranscript reconstructs a best-effort phase output from a loop that // salvagePhaseTranscript reconstructs a best-effort phase output from a loop that
// ended without a final answer: the assistant's narrated text across every step, // ended without a final answer: the assistant's narrated text across every step,
// tail-trimmed to a bound. Returns "" when the model wrote no prose. // tail-trimmed to maxSalvageBytes on a rune boundary. Returns "" when the model
// wrote no prose.
func salvagePhaseTranscript(res *agent.Result) string { func salvagePhaseTranscript(res *agent.Result) string {
if res == nil { if res == nil {
return "" return ""
@@ -232,22 +270,27 @@ func salvagePhaseTranscript(res *agent.Result) string {
} }
} }
out := strings.TrimSpace(b.String()) out := strings.TrimSpace(b.String())
const maxSalvage = 8000 if len(out) > maxSalvageBytes {
if len(out) > maxSalvage { tail := out[len(out)-maxSalvageBytes:]
out = "...(earlier reasoning trimmed)...\n" + out[len(out)-maxSalvage:] // Advance to the next rune boundary so the cut never splits a UTF-8 rune.
for len(tail) > 0 && !utf8.RuneStart(tail[0]) {
tail = tail[1:]
}
out = "...(earlier reasoning trimmed)...\n" + tail
} }
return out return out
} }
// phaseUserMessage builds a phase's user message: the original query text plus // multimodalUserMessage builds a user message from text + inline images. Shared
// any inline images. Mirrors the single-loop multimodal seeding in runAgent. // by the phase runner and runAgent so the image-folding lives in one place.
func phaseUserMessage(query string, images []llm.ImagePart) llm.Message { // Empty text with images yields an image-only message (no empty text part).
func multimodalUserMessage(text string, images []llm.ImagePart) llm.Message {
if len(images) == 0 { if len(images) == 0 {
return llm.UserText(query) return llm.UserText(text)
} }
parts := make([]llm.Part, 0, len(images)+1) parts := make([]llm.Part, 0, len(images)+1)
if strings.TrimSpace(query) != "" { if strings.TrimSpace(text) != "" {
parts = append(parts, llm.Text(query)) parts = append(parts, llm.Text(text))
} }
for _, img := range images { for _, img := range images {
parts = append(parts, img) parts = append(parts, img)
@@ -257,11 +300,13 @@ func phaseUserMessage(query string, images []llm.ImagePart) llm.Message {
// expandPhaseTemplate applies Go text/template substitution to a phase prompt, // expandPhaseTemplate applies Go text/template substitution to a phase prompt,
// replacing {{.Query}} with the original query and {{.<PhaseName>}} with a prior // replacing {{.Query}} with the original query and {{.<PhaseName>}} with a prior
// phase's output. Returns the original string unchanged if parsing or execution // phase's output. On a parse/execute error it logs a WARN and returns the
// fails (best-effort, not fatal). // template unchanged (best-effort, non-fatal) so a misconfigured prompt is
// visible rather than silently masked.
func expandPhaseTemplate(tmpl, query string, priorOutputs map[string]string) string { func expandPhaseTemplate(tmpl, query string, priorOutputs map[string]string) string {
t, err := template.New("phase").Option("missingkey=zero").Parse(tmpl) t, err := template.New("phase").Option("missingkey=zero").Parse(tmpl)
if err != nil { if err != nil {
slog.Warn("run: pipeline phase template parse failed; using it unexpanded", "error", err)
return tmpl return tmpl
} }
data := map[string]string{"Query": query} data := map[string]string{"Query": query}
@@ -270,22 +315,21 @@ func expandPhaseTemplate(tmpl, query string, priorOutputs map[string]string) str
} }
var buf bytes.Buffer var buf bytes.Buffer
if err := t.Execute(&buf, data); err != nil { if err := t.Execute(&buf, data); err != nil {
slog.Warn("run: pipeline phase template execute failed; using it unexpanded", "error", err)
return tmpl return tmpl
} }
return buf.String() return buf.String()
} }
// filterToolbox returns a new toolbox restricted to the named tools (preserving // filterToolbox returns a toolbox restricted to the named tools (preserving
// palette order). Empty names = the full palette. Unknown names are skipped with // palette order). Empty names = the full palette (the base toolbox is returned
// a WARN — a typo'd phase tool list should not abort a run mid-pipeline. // as-is — it is read-only during a run, like the single-loop path). Unknown names
// are skipped with a WARN — a typo'd phase tool list should not abort a run.
func filterToolbox(box *llm.Toolbox, names []string) *llm.Toolbox { func filterToolbox(box *llm.Toolbox, names []string) *llm.Toolbox {
out := llm.NewToolbox(box.Name())
if len(names) == 0 { if len(names) == 0 {
for _, t := range box.Tools() { return box
_ = out.Add(t)
}
return out
} }
out := llm.NewToolbox(box.Name())
for _, name := range names { for _, name := range names {
t, ok := box.Get(name) t, ok := box.Get(name)
if !ok { if !ok {
@@ -300,7 +344,9 @@ func filterToolbox(box *llm.Toolbox, names []string) *llm.Toolbox {
} }
// addUsage sums two llm.Usage tallies field-by-field so a phased run reports the // addUsage sums two llm.Usage tallies field-by-field so a phased run reports the
// total tokens across all phases. // total tokens across all phases. NOTE: if llm.Usage gains a field, add it here
// too — the audit recorder (rec) is the authoritative per-run token source, this
// is the secondary Result.Usage roll-up.
func addUsage(a, b llm.Usage) llm.Usage { func addUsage(a, b llm.Usage) llm.Usage {
a.InputTokens += b.InputTokens a.InputTokens += b.InputTokens
a.OutputTokens += b.OutputTokens a.OutputTokens += b.OutputTokens
+25
View File
@@ -105,6 +105,31 @@ func TestPhases_OptionalFailureSubstitutesFallback(t *testing.T) {
} }
} }
// TestPhases_OptionalDoesNotSwallowCancellation: an Optional phase that fails
// with a context cancellation must NOT be swallowed into its FallbackMessage —
// the run genuinely ended (cancel/deadline/critic-kill) and must surface the
// error so the run is classified cancelled/timeout/killed, not "ok".
func TestPhases_OptionalDoesNotSwallowCancellation(t *testing.T) {
models, _ := phaseProvider(t, fake.Fail(context.Canceled))
ex := New(Config{Registry: tool.NewRegistry(), Models: models})
ra := RunnableAgent{
Name: "pipeline",
ModelTier: "test-model",
Phases: []Phase{
// IsRunFunc so the cancellation surfaces directly wrapped (%w).
{Name: "a", SystemPrompt: "Phase A", IsRunFunc: true, Optional: true, FallbackMessage: "FB"},
},
}
res := ex.Run(context.Background(), ra, tool.Invocation{RunID: "r", CallerID: "c"}, "Q")
if !errors.Is(res.Err, context.Canceled) {
t.Fatalf("Optional phase must NOT swallow a cancellation; res.Err = %v", res.Err)
}
if res.Output == "FB" {
t.Error("a cancelled run must not report the fallback message as output")
}
}
// TestPhases_HardErrorAborts: a NON-optional phase that hits a hard error (not a // TestPhases_HardErrorAborts: a NON-optional phase that hits a hard error (not a
// budget/step exhaustion) aborts the pipeline; later phases do not run. // budget/step exhaustion) aborts the pipeline; later phases do not run.
func TestPhases_HardErrorAborts(t *testing.T) { func TestPhases_HardErrorAborts(t *testing.T) {