subagent_interrupt.go RawHistoryFolder

1package tools
2
3import (
4	"context"
5	"encoding/json"
6	"fmt"
7	"strings"
8
9	"github.com/bnema/zut/packages/agent/subagents"
10	"github.com/bnema/zut/packages/core"
11)
12
13// SubagentInterruptTool cancels only a sub-agent's running turn. Unlike stop,
14// the child stays alive with its transcript, so a later resume continues with
15// full context.
16type SubagentInterruptTool struct {
17	ResidentManager *subagents.ResidentManager
18	Enabled         func() bool
19}
20
21type subagentInterruptArgs struct {
22	AgentID string `json:"agent_id"`
23}
24
25// Name returns the shared facade name: this type is an internal
26// implementation and is never registered on its own.
27func (t *SubagentInterruptTool) Name() string { return SubagentToolName }
28
29func (t *SubagentInterruptTool) Execute(ctx context.Context, raw json.RawMessage, _ func(string)) (core.ToolResult, error) {
30	if ctx != nil {
31		select {
32		case <-ctx.Done():
33			return core.ToolResult{}, ctx.Err()
34		default:
35		}
36	}
37	prefix := t.Name()
38	if t.ResidentManager == nil {
39		return protocolToolError(prefix + ": subagent runtime not available in this mode")
40	}
41	if t.Enabled == nil || !t.Enabled() {
42		return protocolToolError(prefix + ": subagent management is unavailable in this mode")
43	}
44	var args subagentInterruptArgs
45	if err := decodeSubagentArgs(raw, &args); err != nil {
46		return core.ToolResult{}, err
47	}
48	id := strings.TrimSpace(args.AgentID)
49	if id == "" {
50		return protocolToolError(prefix + ": agent_id is required")
51	}
52	snapshot, ok := findResidentStatusSnapshot(t.ResidentManager.Snapshot(), id)
53	if !ok {
54		return protocolToolError(fmt.Sprintf("%s: no such agent %q", prefix, id))
55	}
56	interrupted, err := t.ResidentManager.Interrupt(ctx, snapshot.ID)
57	if err != nil {
58		return protocolToolError(prefix + ": " + err.Error())
59	}
60	if !interrupted {
61		return protocolToolError(fmt.Sprintf("%s: agent %q has no running turn to interrupt", prefix, snapshot.ID))
62	}
63	if updated, ok := t.ResidentManager.SnapshotFor(snapshot.ID); ok {
64		snapshot = updated
65	}
66	return renderResidentAction("interrupt_requested", publicResidentStatus(snapshot))
67}