@gotgenes/pi-subagents 13.0.0 → 13.2.0

package/src/lifecycle/agent-runner.ts

@@ -1,464 +0,0 @@
-/**
- * agent-runner.ts - Core execution engine: creates sessions, runs agents, collects results.
- */
-
-import type { Model } from "@earendil-works/pi-ai";
-import {
-  type AgentSession,
-  type AgentSessionEvent,
-  type SettingsManager,
-} from "@earendil-works/pi-coding-agent";
-import type { AgentConfigLookup } from "#src/config/agent-types";
-import type { ChildLifecyclePublisher } from "#src/lifecycle/child-lifecycle";
-import type { ParentSnapshot } from "#src/lifecycle/parent-snapshot";
-import { extractAssistantContent } from "#src/session/content-items";
-import { extractText } from "#src/session/context";
-import type { EnvInfo } from "#src/session/env";
-import { type AssemblerIO, assembleSessionConfig } from "#src/session/session-config";
-import type { ParentSessionInfo, ShellExec, SubagentType, ThinkingLevel } from "#src/types";
-
-/** Names of tools registered by this extension that subagents must NOT inherit. */
-const EXCLUDED_TOOL_NAMES = ["subagent", "get_subagent_result", "steer_subagent"];
-
-/**
- * Filter the session's active tool names: remove recursion-guard tools.
- *
- * Run once after `bindExtensions` so extension-registered tools (added during
- * `bindExtensions`) are also covered by the guard.
- *
- * @param activeTools  Names currently active on the session.
- */
-function filterActiveTools(activeTools: string[]): string[] {
-  return activeTools.filter((t) => !EXCLUDED_TOOL_NAMES.includes(t));
-}
-
-/** Normalize max turns. undefined or 0 = unlimited, otherwise minimum 1. */
-export function normalizeMaxTurns(n: number | undefined): number | undefined {
-  if (n == null || n === 0) return undefined;
-  return Math.max(1, n);
-}
-
-// ── IO boundary ───────────────────────────────────────────────────────────────
-
-/** Minimal resource-loader contract used by the runner. */
-export interface ResourceLoaderLike {
-  reload(): Promise<void>;
-}
-
-/** Minimal session-manager contract used by the runner. */
-export interface SessionManagerLike {
-  newSession(opts: { parentSession?: string }): void;
-  getSessionFile(): string | undefined;
-}
-
-/** Options passed to EnvironmentIO/SessionFactoryIO methods. */
-export interface ResourceLoaderOptions {
-  cwd: string;
-  agentDir: string;
-  noPromptTemplates?: boolean;
-  noThemes?: boolean;
-  noContextFiles?: boolean;
-  systemPromptOverride?: () => string;
-  /** Override the append system prompt. Receives the current base value; return the replacement. */
-  appendSystemPromptOverride?: (base: string[]) => string[];
-}
-
-/** Options passed to SessionFactoryIO.createSession. */
-export interface CreateSessionOptions {
-  cwd: string;
-  agentDir: string;
-  sessionManager: SessionManagerLike;
-  settingsManager: SettingsManager;
-  modelRegistry: unknown;
-  model?: unknown;
-  tools: string[];
-  resourceLoader: ResourceLoaderLike;
-  thinkingLevel?: ThinkingLevel;
-}
-
-/**
- * Environment discovery - detect runtime context and resolve directories.
- *
- * Decouples the runner from direct process/SDK reads so each can be stubbed
- * independently in tests.
- */
-export interface EnvironmentIO {
-  detectEnv: (exec: ShellExec, cwd: string) => Promise<EnvInfo>;
-  getAgentDir: () => string;
-  deriveSessionDir: (parentSessionFile: string | undefined, effectiveCwd: string) => string;
-}
-
-/**
- * Session factory - create SDK objects for a child agent session.
- *
- * Decouples the runner from direct Pi SDK imports and sibling-module IO,
- * making it testable via plain stub objects without vi.mock().
- */
-export interface SessionFactoryIO {
-  createResourceLoader: (opts: ResourceLoaderOptions) => ResourceLoaderLike;
-  createSessionManager: (cwd: string, sessionDir: string) => SessionManagerLike;
-  createSettingsManager: (cwd: string, agentDir: string) => SettingsManager;
-  createSession: (opts: CreateSessionOptions) => Promise<{ session: AgentSession }>;
-  assemblerIO: AssemblerIO;
-}
-
-/**
- * IO boundary injected into runAgent().
- *
- * Backward-compatible intersection of EnvironmentIO and SessionFactoryIO.
- * Callers that previously constructed a RunnerIO object continue to satisfy
- * both sub-interfaces via TypeScript's structural typing.
- */
-export type RunnerIO = EnvironmentIO & SessionFactoryIO;
-
-/**
- * Dependencies owned by the runner — injected at construction time.
- *
- * Groups the IO boundary with the two static domain deps (exec, registry)
- * that every run() call needs but that do not vary per call.
- */
-export interface RunnerDeps {
-  io: RunnerIO;
-  exec: ShellExec;
-  registry: AgentConfigLookup;
-  /** Publishes the child-execution lifecycle so consumers can observe it. */
-  lifecycle: ChildLifecyclePublisher;
-}
-
-// ── Public interfaces ─────────────────────────────────────────────────────────
-
-/**
- * Per-call execution context — fields that vary per spawn.
- *
- * Static dependencies (exec, registry) live on RunnerDeps; this interface
- * carries only the two per-call fields that AgentManager supplies at spawn time.
- */
-export interface RunContext {
-  /** Override working directory (e.g. for worktree isolation). */
-  cwd?: string;
-  /** Parent session identity (file path + session ID). */
-  parentSession?: ParentSessionInfo;
-}
-
-export interface RunOptions {
-  /** Parent execution context - where/who is running. */
-  context: RunContext;
-  model?: Model<any>;
-  maxTurns?: number;
-  signal?: AbortSignal;
-  thinkingLevel?: ThinkingLevel;
-  /** Called once after session creation - session delivery mechanism. */
-  onSessionCreated?: (session: AgentSession) => void;
-  /**
-   * Default max turns from runtime config. Falls back to the module-scope
-   * `defaultMaxTurns` during the lift-and-shift migration; superseded by
-   * per-call `maxTurns` and per-agent `agentConfig.maxTurns`.
-   */
-  defaultMaxTurns?: number;
-  /**
-   * Grace turns after the soft-limit steer message. Falls back to the
-   * module-scope `graceTurns` during migration.
-   */
-  graceTurns?: number;
-}
-
-export interface RunResult {
-  responseText: string;
-  session: AgentSession;
-  /** True if the agent was hard-aborted (max_turns + grace exceeded). */
-  aborted: boolean;
-  /** True if the agent was steered to wrap up (hit soft turn limit) but finished in time. */
-  steered: boolean;
-  /** Path to the persisted session JSONL file, if the session was persisted. */
-  sessionFile?: string;
-}
-
-/** Options for resuming an existing agent session. */
-export interface ResumeOptions {
-  signal?: AbortSignal;
-}
-
-/**
- * Execution boundary: decouples AgentManager (lifecycle management) from the
- * SDK session orchestration in runAgent/resumeAgent.
- */
-export interface AgentRunner {
-  run(snapshot: ParentSnapshot, type: SubagentType, prompt: string, options: RunOptions): Promise<RunResult>;
-  resume(session: AgentSession, prompt: string, options?: ResumeOptions): Promise<string>;
-}
-
-/**
- * Concrete AgentRunner backed by RunnerDeps.
- *
- * Captures IO, exec, and registry at construction time so AgentManager
- * remains unaware of runner-internal dependencies.
- */
-export class ConcreteAgentRunner implements AgentRunner {
-  constructor(private readonly deps: RunnerDeps) {}
-
-  run(snapshot: ParentSnapshot, type: SubagentType, prompt: string, options: RunOptions): Promise<RunResult> {
-    return runAgent(snapshot, type, prompt, options, this.deps);
-  }
-
-  resume(session: AgentSession, prompt: string, options?: ResumeOptions): Promise<string> {
-    return resumeAgent(session, prompt, options);
-  }
-}
-
-
-// ── Private helpers ───────────────────────────────────────────────────────────
-
-/**
- * Subscribe to a session and collect the last assistant message text.
- * Returns an object with a `getText()` getter and an `unsubscribe` function.
- */
-function collectResponseText(session: AgentSession) {
-  let text = "";
-  const unsubscribe = session.subscribe((event: AgentSessionEvent) => {
-    if (event.type === "message_start") {
-      text = "";
-    }
-    if (
-      event.type === "message_update" &&
-      event.assistantMessageEvent.type === "text_delta"
-    ) {
-      text += event.assistantMessageEvent.delta;
-    }
-  });
-  return { getText: () => text, unsubscribe };
-}
-
-/** Get the last assistant text from the completed session history. */
-function getLastAssistantText(session: AgentSession): string {
-  for (let i = session.messages.length - 1; i >= 0; i--) {
-    const msg = session.messages[i];
-    if (msg.role !== "assistant") continue;
-    const text = extractText(msg.content).trim();
-    if (text) return text;
-  }
-  return "";
-}
-
-/**
- * Wire an AbortSignal to abort a session.
- * Returns a cleanup function to remove the listener.
- */
-function forwardAbortSignal(
-  session: AgentSession,
-  signal?: AbortSignal,
-): () => void {
-  if (!signal) return () => {};
-  const onAbort = (): void => { void session.abort(); };
-  signal.addEventListener("abort", onAbort, { once: true });
-  return () => signal.removeEventListener("abort", onAbort);
-}
-
-// ── Public functions ──────────────────────────────────────────────────────────
-
-export async function runAgent(
-  snapshot: ParentSnapshot,
-  type: SubagentType,
-  prompt: string,
-  options: RunOptions,
-  deps: RunnerDeps,
-): Promise<RunResult> {
-  const parentSessionId = options.context.parentSession?.parentSessionId;
-  deps.lifecycle.spawning({ agentName: type, parentSessionId });
-
-  // Resolve working directory upfront - needed for detectEnv before assembly.
-  const effectiveCwd = options.context.cwd ?? snapshot.cwd;
-  const env = await deps.io.detectEnv(deps.exec, effectiveCwd);
-
-  // Assemble session configuration (synchronous, no SDK objects).
-  const cfg = assembleSessionConfig(
-    type,
-    {
-      cwd: snapshot.cwd,
-      parentSystemPrompt: snapshot.systemPrompt,
-      parentModel: snapshot.model,
-      modelRegistry: snapshot.modelRegistry,
-    },
-    {
-      cwd: options.context.cwd,
-      model: options.model,
-      thinkingLevel: options.thinkingLevel,
-    },
-    env,
-    deps.registry,
-    deps.io.assemblerIO,
-  );
-
-  const agentDir = deps.io.getAgentDir();
-
-  // Children always load the parent's extensions and skills.
-  // Suppress AGENTS.md/CLAUDE.md and APPEND_SYSTEM.md - upstream's
-  // buildSystemPrompt() re-appends both AFTER systemPromptOverride, which
-  // would defeat prompt_mode: replace. Parent context, if
-  // wanted, reaches the subagent via prompt_mode: append (parentSystemPrompt
-  // is embedded in systemPromptOverride) or inherit_context (conversation).
-  const loader = deps.io.createResourceLoader({
-    cwd: cfg.effectiveCwd,
-    agentDir,
-    noPromptTemplates: true,
-    noThemes: true,
-    noContextFiles: true,
-    systemPromptOverride: () => cfg.systemPrompt,
-    appendSystemPromptOverride: () => [],
-  });
-  await loader.reload();
-
-  // Create a persisted SessionManager so transcripts are written in Pi's
-  // official JSONL format. Falls back to a temp directory when the parent
-  // session is not persisted (e.g. headless/API mode).
-  const sessionDir = deps.io.deriveSessionDir(options.context.parentSession?.parentSessionFile, cfg.effectiveCwd);
-  const sessionManager = deps.io.createSessionManager(cfg.effectiveCwd, sessionDir);
-  sessionManager.newSession({ parentSession: options.context.parentSession?.parentSessionId });
-
-  const { session } = await deps.io.createSession({
-    cwd: cfg.effectiveCwd,
-    agentDir,
-    sessionManager,
-    settingsManager: deps.io.createSettingsManager(cfg.effectiveCwd, agentDir),
-    modelRegistry: snapshot.modelRegistry,
-    model: cfg.model,
-    tools: cfg.toolNames,
-    resourceLoader: loader,
-    thinkingLevel: cfg.thinkingLevel,
-  });
-
-  // Publish session-created before bindExtensions() so observers (e.g. the
-  // permission system) can register the child synchronously and have their
-  // entry in place for the first permission check during child extension
-  // initialization. The event bus dispatches synchronously, so a synchronous
-  // subscriber completes before this returns. Paired with disposed() in the
-  // finally block below to guarantee cleanup on both success and error paths.
-  deps.lifecycle.sessionCreated({ sessionDir, agentName: type, parentSessionId });
-
-  // Bind extensions so that session_start fires and extensions can initialize
-  // (e.g. loading credentials, setting up state). Placed after tool filtering
-  // so extension-provided skills/prompts from extendResourcesFromExtensions()
-  // respect the active tool set. All ExtensionBindings fields are optional.
-  await session.bindExtensions({});
-
-  // Apply recursion guard: remove our own tools from the child's active set.
-  // Runs after bindExtensions so extension-registered tools are included in the
-  // post-bind active set. Unconditional: children always load the parent's
-  // extensions, so the guard must always strip our dispatch tools.
-  const filtered = filterActiveTools(session.getActiveToolNames());
-  session.setActiveToolsByName(filtered);
-
-  options.onSessionCreated?.(session);
-
-  // Track turns for graceful max_turns enforcement
-  let turnCount = 0;
-  const maxTurns = normalizeMaxTurns(
-    options.maxTurns ?? cfg.agentMaxTurns ?? options.defaultMaxTurns,
-  );
-  let softLimitReached = false;
-  let aborted = false;
-
-  const unsubTurns = session.subscribe((event: AgentSessionEvent) => {
-    if (event.type === "turn_end") {
-      turnCount++;
-      if (maxTurns != null) {
-        if (!softLimitReached && turnCount >= maxTurns) {
-          softLimitReached = true;
-          void session.steer(
-            "You have reached your turn limit. Wrap up immediately - provide your final answer now.",
-          );
-        } else if (softLimitReached && turnCount >= maxTurns + (options.graceTurns ?? 5)) {
-          aborted = true;
-          void session.abort();
-        }
-      }
-    }
-  });
-
-  const collector = collectResponseText(session);
-  const cleanupAbort = forwardAbortSignal(session, options.signal);
-
-  // Prepend parent context if it was captured at spawn time
-  let effectivePrompt = prompt;
-  if (snapshot.parentContext) {
-    effectivePrompt = snapshot.parentContext + prompt;
-  }
-
-  try {
-    await session.prompt(effectivePrompt);
-    deps.lifecycle.completed({ sessionDir, agentName: type, aborted, steered: softLimitReached });
-  } finally {
-    unsubTurns();
-    collector.unsubscribe();
-    cleanupAbort();
-    deps.lifecycle.disposed({ sessionDir });
-  }
-
-  const responseText =
-    collector.getText().trim() || getLastAssistantText(session);
-  return {
-    responseText,
-    session,
-    aborted,
-    steered: softLimitReached,
-    sessionFile: sessionManager.getSessionFile(),
-  };
-}
-
-/**
- * Send a new prompt to an existing session (resume).
- */
-export async function resumeAgent(
-  session: AgentSession,
-  prompt: string,
-  options: ResumeOptions = {},
-): Promise<string> {
-  const collector = collectResponseText(session);
-  const cleanupAbort = forwardAbortSignal(session, options.signal);
-
-  try {
-    await session.prompt(prompt);
-  } finally {
-    collector.unsubscribe();
-    cleanupAbort();
-  }
-
-  return collector.getText().trim() || getLastAssistantText(session);
-}
-
-/**
- * Get the subagent's conversation messages as formatted text.
- */
-export function getAgentConversation(session: AgentSession): string {
-  const parts: string[] = [];
-
-  for (const msg of session.messages) {
-    if (msg.role === "user") {
-      const text =
-        typeof msg.content === "string"
-          ? msg.content
-          : extractText(msg.content);
-      if (text.trim()) parts.push(`[User]: ${text.trim()}`);
-    } else if (msg.role === "assistant") {
-      const { textParts, toolNames } = extractAssistantContent(msg.content);
-      const attribution = formatAttribution(msg);
-      if (textParts.length > 0)
-        parts.push(`[Assistant${attribution}]: ${textParts.join("\n")}`);
-      if (toolNames.length > 0)
-        parts.push(`[Tool Calls]:\n${toolNames.map((n) => `  Tool: ${n}`).join("\n")}`);
-    } else if (msg.role === "toolResult") {
-      const text = extractText(msg.content);
-      const truncated = text.length > 200 ? text.slice(0, 200) + "..." : text;
-      parts.push(`[Tool Result (${msg.toolName})]: ${truncated}`);
-    }
-  }
-
-  return parts.join("\n\n");
-}
-
-/** Build a `(provider/model)` attribution suffix for assistant messages. */
-function formatAttribution(msg: { provider?: string; model?: string }): string {
-  const { provider, model } = msg;
-  if (!provider && !model) return "";
-  if (provider && model) return ` (${provider}/${model})`;
-  return ` (${provider ?? model})`;
-}

package/src/lifecycle/execution-state.ts

@@ -1,17 +0,0 @@
-/**
- * execution-state.ts — ExecutionState: execution-phase state for a running agent.
- *
- * Constructed and attached to Agent when onSessionCreated fires inside startAgent().
- * Contains the session and output file — the two fields that become known once the
- * runner creates the session. promise stays as a separate Agent field because
- * it is set at a different moment (after runner.run() returns).
- */
-
-import type { AgentSession } from "@earendil-works/pi-coding-agent";
-
-export interface ExecutionState {
-	/** The active agent session — available from the moment the session is created. */
-	readonly session: AgentSession;
-	/** Path to the agent's session JSONL file, or undefined if not yet available. */
-	readonly outputFile: string | undefined;
-}