@by-lua/lspec-subagents 1.0.1

package/README.md

@@ -0,0 +1,123 @@
+# lspec-subagents
+
+> **9 specialized sub-agents for PI.dev** — extensão standalone com modelo centralizado de configuração (inspirado no oh-my-opencode-slim). Fork do `@tintinweb/pi-subagents`.
+
+9 agentes especializados para o fluxo L-Spec (Lua Spec-Driven Development). Os 3 agentes padrão do original (general-purpose, Explore, Plan) foram **completamente removidos** — só os 9 L-Spec existem no código.
+
+## Agentes
+
+- **orchestrator** — Coordenador central do fluxo L-Spec (`claude-sonnet-4`)
+- **explorer** — Navegação rápida no codebase, read-only (`gpt-4o-mini`)
+- **librarian** — Pesquisa de docs externas e APIs (`claude-sonnet-4`)
+- **oracle** — Arquiteto sênior, code review, read-only (`claude-opus-4`)
+- **designer** — UI/UX specialist (`claude-sonnet-4`)
+- **fixer** — Implementador rápido de tarefas definidas (`claude-sonnet-4`)
+- **observer** — Análise visual (imagens, PDFs, screenshots) (`gpt-4o-mini`, vision-capable)
+- **council** — Consenso multi-modelo (spawna councillors) (`claude-opus-4`)
+- **councillor** — Membro individual do conselho (`gpt-4o-mini`)
+
+## Configuração Centralizada de Modelos
+
+Modelos são **placeholders** (`{{model:orchestrator}}`) resolvidos em runtime via JSON.
+
+### Como configurar
+
+Crie `lspec-model-config.json` em:
+- **Global**: `~/.pi/agent/lspec-model-config.json`
+- **Projeto**: `.pi/lspec-model-config.json` (sobrescreve o global)
+
+```json
+{
+  "agents": {
+    "orchestrator": "claude-sonnet-4",
+    "explorer": "gpt-4o-mini",
+    "librarian": "claude-sonnet-4",
+    "oracle": "claude-opus-4",
+    "designer": "claude-sonnet-4",
+    "fixer": "claude-sonnet-4",
+    "observer": "gpt-4o-mini",
+    "council": "claude-opus-4",
+    "councillor": "gpt-4o-mini"
+  }
+}
+```
+
+Pode usar qualquer modelo disponível no seu provedor. O observer precisa de um modelo com **visão**.
+
+### Ordem de resolução
+
+1. Projeto (`.pi/lspec-model-config.json`) — **maior prioridade**
+2. Global (`~/.pi/agent/lspec-model-config.json`)
+3. Defaults embutidos (claude-sonnet-4, gpt-4o-mini, claude-opus-4)
+
+## Instalação
+
+```bash
+curl -fsSL https://raw.githubusercontent.com/by-lua/lspec-subagents/main/install.sh | bash
+```
+
+O instalador faz 2 passos:
+1. `pi install git:github.com/by-lua/lspec-subagents` — instala a extensão standalone
+2. Copia os 9 agent `.md` files pra `~/.pi/agents/` + cria `lspec-model-config.json` (se não existir)
+
+Requer: `git`, `pi` (PI.dev).
+
+### Dependências Opcionais (Recomendadas)
+
+Os agentes usam estas ferramentas PI quando disponíveis:
+
+```bash
+pi install npm:pi-cymbal              # Code navigation (explorer, oracle, fixer, orchestrator)
+pi install npm:@davehardy20/pi-lsp-tools  # LSP-powered code tools (explorer, fixer, oracle)
+pi install npm:pi-mermaid             # Mermaid diagrams (orchestrator, designer)
+```
+
+Cada agente usa as ferramentas disponíveis com fallback para grep/read se o pacote não estiver instalado.
+
+### Atualizar
+
+```bash
+curl -fsSL https://raw.githubusercontent.com/by-lua/lspec-subagents/main/update.sh | bash
+```
+
+### Desinstalar
+
+```bash
+curl -fsSL https://raw.githubusercontent.com/by-lua/lspec-subagents/main/uninstall.sh | bash
+```
+
+Remove extensão + 9 agent .md files. Preserva `lspec-model-config.json`.
+
+### Instalação manual
+
+```bash
+pi install git:github.com/by-lua/lspec-subagents          # extensão standalone
+git clone --depth 1 https://github.com/by-lua/lspec-subagents.git /tmp/lspec-sub
+cp /tmp/lspec-sub/.pi/agents/*.md ~/.pi/agents/          # .md overrides
+cp /tmp/lspec-sub/lspec-model-config.example.json ~/.pi/agent/lspec-model-config.json
+rm -rf /tmp/lspec-sub
+```
+
+## Como usar
+
+No PI, use o tool `Agent` com o tipo do agente:
+
+```
+Agent(agent_type="explorer", prompt="Find all React components in src/")
+Agent(agent_type="orchestrator", prompt="Run L-Spec Discovery for the auth feature")
+```
+
+Ou customize agentes `.md` em `.pi/agents/` para sobrescrever os defaults.
+
+## Diferenças do pi-subagents original
+
+- ✅ **9 agentes L-Spec** embutidos no código (não override por .md)
+- ✅ **Zero agentes padrão** — general-purpose/Explore/Plan removidos do source
+- ✅ **Placeholders de modelo** (`{{model:orchestrator}}`)
+- ✅ **Config JSON centralizada** (projeto + global + defaults)
+- ✅ **Fallback** mudou de `general-purpose` pra `orchestrator`
+- ✅ **Extensão standalone** — não depende do pacote npm @tintinweb
+
+## Licença
+
+MIT — fork de `@tintinweb/pi-subagents` (MIT).

package/dist/agent-manager.d.ts

@@ -0,0 +1,108 @@
+/**
+ * agent-manager.ts — Tracks agents, background execution, resume support.
+ *
+ * Background agents are subject to a configurable concurrency limit (default: 4).
+ * Excess agents are queued and auto-started as running agents complete.
+ * Foreground agents bypass the queue (they block the parent anyway).
+ */
+import type { Model } from "@mariozechner/pi-ai";
+import type { AgentSession, ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { type ToolActivity } from "./agent-runner.js";
+import type { AgentInvocation, AgentRecord, IsolationMode, SubagentType, ThinkingLevel } from "./types.js";
+export type OnAgentComplete = (record: AgentRecord) => void;
+export type OnAgentStart = (record: AgentRecord) => void;
+export type OnAgentCompact = (record: AgentRecord, info: CompactionInfo) => void;
+export type CompactionInfo = {
+    reason: "manual" | "threshold" | "overflow";
+    tokensBefore: number;
+};
+interface SpawnOptions {
+    description: string;
+    model?: Model<any>;
+    maxTurns?: number;
+    isolated?: boolean;
+    inheritContext?: boolean;
+    thinkingLevel?: ThinkingLevel;
+    isBackground?: boolean;
+    /**
+     * Skip the maxConcurrent queue check for this spawn — start immediately even
+     * if the configured concurrency limit would otherwise queue it. Used by the
+     * scheduler so a fired job can't be deferred past its trigger window.
+     */
+    bypassQueue?: boolean;
+    /** Isolation mode — "worktree" creates a temp git worktree for the agent. */
+    isolation?: IsolationMode;
+    /** Resolved invocation snapshot captured for UI display. */
+    invocation?: AgentInvocation;
+    /** Parent abort signal — when aborted, the subagent is also stopped. */
+    signal?: AbortSignal;
+    /** Called on tool start/end with activity info (for streaming progress to UI). */
+    onToolActivity?: (activity: ToolActivity) => void;
+    /** Called on streaming text deltas from the assistant response. */
+    onTextDelta?: (delta: string, fullText: string) => void;
+    /** Called when the agent session is created (for accessing session stats). */
+    onSessionCreated?: (session: AgentSession) => void;
+    /** Called at the end of each agentic turn with the cumulative count. */
+    onTurnEnd?: (turnCount: number) => void;
+    /** Called once per assistant message_end with that message's usage delta. */
+    onAssistantUsage?: (usage: {
+        input: number;
+        output: number;
+        cacheWrite: number;
+    }) => void;
+    /** Called when the session successfully compacts. */
+    onCompaction?: (info: CompactionInfo) => void;
+}
+export declare class AgentManager {
+    private agents;
+    private cleanupInterval;
+    private onComplete?;
+    private onStart?;
+    private onCompact?;
+    private maxConcurrent;
+    /** Queue of background agents waiting to start. */
+    private queue;
+    /** Number of currently running background agents. */
+    private runningBackground;
+    constructor(onComplete?: OnAgentComplete, maxConcurrent?: number, onStart?: OnAgentStart, onCompact?: OnAgentCompact);
+    /** Update the max concurrent background agents limit. */
+    setMaxConcurrent(n: number): void;
+    getMaxConcurrent(): number;
+    /**
+     * Spawn an agent and return its ID immediately (for background use).
+     * If the concurrency limit is reached, the agent is queued.
+     */
+    spawn(pi: ExtensionAPI, ctx: ExtensionContext, type: SubagentType, prompt: string, options: SpawnOptions): string;
+    /** Actually start an agent (called immediately or from queue drain). */
+    private startAgent;
+    /** Start queued agents up to the concurrency limit. */
+    private drainQueue;
+    /**
+     * Spawn an agent and wait for completion (foreground use).
+     * Foreground agents bypass the concurrency queue.
+     */
+    spawnAndWait(pi: ExtensionAPI, ctx: ExtensionContext, type: SubagentType, prompt: string, options: Omit<SpawnOptions, "isBackground">): Promise<AgentRecord>;
+    /**
+     * Resume an existing agent session with a new prompt.
+     */
+    resume(id: string, prompt: string, signal?: AbortSignal): Promise<AgentRecord | undefined>;
+    getRecord(id: string): AgentRecord | undefined;
+    listAgents(): AgentRecord[];
+    abort(id: string): boolean;
+    /** Dispose a record's session and remove it from the map. */
+    private removeRecord;
+    private cleanup;
+    /**
+     * Remove all completed/stopped/errored records immediately.
+     * Called on session start/switch so tasks from a prior session don't persist.
+     */
+    clearCompleted(): void;
+    /** Whether any agents are still running or queued. */
+    hasRunning(): boolean;
+    /** Abort all running and queued agents immediately. */
+    abortAll(): number;
+    /** Wait for all running and queued agents to complete (including queued ones). */
+    waitForAll(): Promise<void>;
+    dispose(): void;
+}
+export {};

package/dist/agent-manager.js

@@ -0,0 +1,391 @@
+/**
+ * agent-manager.ts — Tracks agents, background execution, resume support.
+ *
+ * Background agents are subject to a configurable concurrency limit (default: 4).
+ * Excess agents are queued and auto-started as running agents complete.
+ * Foreground agents bypass the queue (they block the parent anyway).
+ */
+import { randomUUID } from "node:crypto";
+import { resumeAgent, runAgent } from "./agent-runner.js";
+import { addUsage } from "./usage.js";
+import { cleanupWorktree, createWorktree, pruneWorktrees, } from "./worktree.js";
+/** Default max concurrent background agents. */
+const DEFAULT_MAX_CONCURRENT = 4;
+export class AgentManager {
+    agents = new Map();
+    cleanupInterval;
+    onComplete;
+    onStart;
+    onCompact;
+    maxConcurrent;
+    /** Queue of background agents waiting to start. */
+    queue = [];
+    /** Number of currently running background agents. */
+    runningBackground = 0;
+    constructor(onComplete, maxConcurrent = DEFAULT_MAX_CONCURRENT, onStart, onCompact) {
+        this.onComplete = onComplete;
+        this.onStart = onStart;
+        this.onCompact = onCompact;
+        this.maxConcurrent = maxConcurrent;
+        // Cleanup completed agents after 10 minutes (but keep sessions for resume)
+        this.cleanupInterval = setInterval(() => this.cleanup(), 60_000);
+        this.cleanupInterval.unref();
+    }
+    /** Update the max concurrent background agents limit. */
+    setMaxConcurrent(n) {
+        this.maxConcurrent = Math.max(1, n);
+        // Start queued agents if the new limit allows
+        this.drainQueue();
+    }
+    getMaxConcurrent() {
+        return this.maxConcurrent;
+    }
+    /**
+     * Spawn an agent and return its ID immediately (for background use).
+     * If the concurrency limit is reached, the agent is queued.
+     */
+    spawn(pi, ctx, type, prompt, options) {
+        const id = randomUUID().slice(0, 17);
+        const abortController = new AbortController();
+        const record = {
+            id,
+            type,
+            description: options.description,
+            status: options.isBackground ? "queued" : "running",
+            toolUses: 0,
+            startedAt: Date.now(),
+            abortController,
+            lifetimeUsage: { input: 0, output: 0, cacheWrite: 0 },
+            compactionCount: 0,
+            invocation: options.invocation,
+        };
+        this.agents.set(id, record);
+        const args = { pi, ctx, type, prompt, options };
+        if (options.isBackground && !options.bypassQueue && this.runningBackground >= this.maxConcurrent) {
+            // Queue it — will be started when a running agent completes
+            this.queue.push({ id, args });
+            return id;
+        }
+        // startAgent can throw (e.g. strict worktree-isolation failure) — clean
+        // up the record so callers don't see an orphan in `listAgents()`.
+        try {
+            this.startAgent(id, record, args);
+        }
+        catch (err) {
+            this.agents.delete(id);
+            throw err;
+        }
+        return id;
+    }
+    /** Actually start an agent (called immediately or from queue drain). */
+    startAgent(id, record, { pi, ctx, type, prompt, options }) {
+        // Worktree isolation: try to create a temporary git worktree. Strict —
+        // fail loud if not possible (no silent fallback to main tree). Done
+        // BEFORE state mutation so a throw doesn't leave the record half-running.
+        let worktreeCwd;
+        if (options.isolation === "worktree") {
+            const wt = createWorktree(ctx.cwd, id);
+            if (!wt) {
+                throw new Error('Cannot run with isolation: "worktree" — not a git repo, no commits yet, or `git worktree add` failed. ' +
+                    'Initialize git and commit at least once, or omit `isolation`.');
+            }
+            record.worktree = wt;
+            worktreeCwd = wt.path;
+        }
+        record.status = "running";
+        record.startedAt = Date.now();
+        if (options.isBackground)
+            this.runningBackground++;
+        this.onStart?.(record);
+        // Wire parent abort signal to stop the subagent when the parent is interrupted
+        let detachParentSignal;
+        if (options.signal) {
+            const onParentAbort = () => this.abort(id);
+            options.signal.addEventListener("abort", onParentAbort, { once: true });
+            detachParentSignal = () => options.signal.removeEventListener("abort", onParentAbort);
+        }
+        const detach = () => { detachParentSignal?.(); detachParentSignal = undefined; };
+        const promise = runAgent(ctx, type, prompt, {
+            pi,
+            agentId: id,
+            model: options.model,
+            maxTurns: options.maxTurns,
+            isolated: options.isolated,
+            inheritContext: options.inheritContext,
+            thinkingLevel: options.thinkingLevel,
+            cwd: worktreeCwd,
+            signal: record.abortController.signal,
+            onToolActivity: (activity) => {
+                if (activity.type === "end")
+                    record.toolUses++;
+                options.onToolActivity?.(activity);
+            },
+            onTurnEnd: options.onTurnEnd,
+            onTextDelta: options.onTextDelta,
+            onAssistantUsage: (usage) => {
+                addUsage(record.lifetimeUsage, usage);
+                options.onAssistantUsage?.(usage);
+            },
+            onCompaction: (info) => {
+                record.compactionCount++;
+                this.onCompact?.(record, info);
+                options.onCompaction?.(info);
+            },
+            onSessionCreated: (session) => {
+                record.session = session;
+                // Flush any steers that arrived before the session was ready
+                if (record.pendingSteers?.length) {
+                    for (const msg of record.pendingSteers) {
+                        session.steer(msg).catch(() => { });
+                    }
+                    record.pendingSteers = undefined;
+                }
+                options.onSessionCreated?.(session);
+            },
+        })
+            .then(({ responseText, session, aborted, steered }) => {
+            // Don't overwrite status if externally stopped via abort()
+            if (record.status !== "stopped") {
+                record.status = aborted ? "aborted" : steered ? "steered" : "completed";
+            }
+            record.result = responseText;
+            record.session = session;
+            record.completedAt ??= Date.now();
+            detach();
+            // Final flush of streaming output file
+            if (record.outputCleanup) {
+                try {
+                    record.outputCleanup();
+                }
+                catch { /* ignore */ }
+                record.outputCleanup = undefined;
+            }
+            // Clean up worktree if used
+            if (record.worktree) {
+                const wtResult = cleanupWorktree(ctx.cwd, record.worktree, options.description);
+                record.worktreeResult = wtResult;
+                if (wtResult.hasChanges && wtResult.branch) {
+                    record.result = (record.result ?? "") +
+                        `\n\n---\nChanges saved to branch \`${wtResult.branch}\`. Merge with: \`git merge ${wtResult.branch}\``;
+                }
+            }
+            if (options.isBackground) {
+                this.runningBackground--;
+                try {
+                    this.onComplete?.(record);
+                }
+                catch { /* ignore completion side-effect errors */ }
+                this.drainQueue();
+            }
+            return responseText;
+        })
+            .catch((err) => {
+            // Don't overwrite status if externally stopped via abort()
+            if (record.status !== "stopped") {
+                record.status = "error";
+            }
+            record.error = err instanceof Error ? err.message : String(err);
+            record.completedAt ??= Date.now();
+            detach();
+            // Final flush of streaming output file on error
+            if (record.outputCleanup) {
+                try {
+                    record.outputCleanup();
+                }
+                catch { /* ignore */ }
+                record.outputCleanup = undefined;
+            }
+            // Best-effort worktree cleanup on error
+            if (record.worktree) {
+                try {
+                    const wtResult = cleanupWorktree(ctx.cwd, record.worktree, options.description);
+                    record.worktreeResult = wtResult;
+                }
+                catch { /* ignore cleanup errors */ }
+            }
+            if (options.isBackground) {
+                this.runningBackground--;
+                this.onComplete?.(record);
+                this.drainQueue();
+            }
+            return "";
+        });
+        record.promise = promise;
+    }
+    /** Start queued agents up to the concurrency limit. */
+    drainQueue() {
+        while (this.queue.length > 0 && this.runningBackground < this.maxConcurrent) {
+            const next = this.queue.shift();
+            const record = this.agents.get(next.id);
+            if (!record || record.status !== "queued")
+                continue;
+            try {
+                this.startAgent(next.id, record, next.args);
+            }
+            catch (err) {
+                // Late failure (e.g. strict worktree-isolation) — surface on the record
+                // so the user/agent can see it via /agents, then keep draining.
+                record.status = "error";
+                record.error = err instanceof Error ? err.message : String(err);
+                record.completedAt = Date.now();
+                this.onComplete?.(record);
+            }
+        }
+    }
+    /**
+     * Spawn an agent and wait for completion (foreground use).
+     * Foreground agents bypass the concurrency queue.
+     */
+    async spawnAndWait(pi, ctx, type, prompt, options) {
+        const id = this.spawn(pi, ctx, type, prompt, { ...options, isBackground: false });
+        const record = this.agents.get(id);
+        await record.promise;
+        return record;
+    }
+    /**
+     * Resume an existing agent session with a new prompt.
+     */
+    async resume(id, prompt, signal) {
+        const record = this.agents.get(id);
+        if (!record?.session)
+            return undefined;
+        record.status = "running";
+        record.startedAt = Date.now();
+        record.completedAt = undefined;
+        record.result = undefined;
+        record.error = undefined;
+        try {
+            const responseText = await resumeAgent(record.session, prompt, {
+                onToolActivity: (activity) => {
+                    if (activity.type === "end")
+                        record.toolUses++;
+                },
+                onAssistantUsage: (usage) => {
+                    addUsage(record.lifetimeUsage, usage);
+                },
+                onCompaction: (info) => {
+                    record.compactionCount++;
+                    this.onCompact?.(record, info);
+                },
+                signal,
+            });
+            record.status = "completed";
+            record.result = responseText;
+            record.completedAt = Date.now();
+        }
+        catch (err) {
+            record.status = "error";
+            record.error = err instanceof Error ? err.message : String(err);
+            record.completedAt = Date.now();
+        }
+        return record;
+    }
+    getRecord(id) {
+        return this.agents.get(id);
+    }
+    listAgents() {
+        return [...this.agents.values()].sort((a, b) => b.startedAt - a.startedAt);
+    }
+    abort(id) {
+        const record = this.agents.get(id);
+        if (!record)
+            return false;
+        // Remove from queue if queued
+        if (record.status === "queued") {
+            this.queue = this.queue.filter(q => q.id !== id);
+            record.status = "stopped";
+            record.completedAt = Date.now();
+            return true;
+        }
+        if (record.status !== "running")
+            return false;
+        record.abortController?.abort();
+        record.status = "stopped";
+        record.completedAt = Date.now();
+        return true;
+    }
+    /** Dispose a record's session and remove it from the map. */
+    removeRecord(id, record) {
+        record.session?.dispose?.();
+        record.session = undefined;
+        this.agents.delete(id);
+    }
+    cleanup() {
+        const cutoff = Date.now() - 10 * 60_000;
+        for (const [id, record] of this.agents) {
+            if (record.status === "running" || record.status === "queued")
+                continue;
+            if ((record.completedAt ?? 0) >= cutoff)
+                continue;
+            this.removeRecord(id, record);
+        }
+    }
+    /**
+     * Remove all completed/stopped/errored records immediately.
+     * Called on session start/switch so tasks from a prior session don't persist.
+     */
+    clearCompleted() {
+        for (const [id, record] of this.agents) {
+            if (record.status === "running" || record.status === "queued")
+                continue;
+            this.removeRecord(id, record);
+        }
+    }
+    /** Whether any agents are still running or queued. */
+    hasRunning() {
+        return [...this.agents.values()].some(r => r.status === "running" || r.status === "queued");
+    }
+    /** Abort all running and queued agents immediately. */
+    abortAll() {
+        let count = 0;
+        // Clear queued agents first
+        for (const queued of this.queue) {
+            const record = this.agents.get(queued.id);
+            if (record) {
+                record.status = "stopped";
+                record.completedAt = Date.now();
+                count++;
+            }
+        }
+        this.queue = [];
+        // Abort running agents
+        for (const record of this.agents.values()) {
+            if (record.status === "running") {
+                record.abortController?.abort();
+                record.status = "stopped";
+                record.completedAt = Date.now();
+                count++;
+            }
+        }
+        return count;
+    }
+    /** Wait for all running and queued agents to complete (including queued ones). */
+    async waitForAll() {
+        // Loop because drainQueue respects the concurrency limit — as running
+        // agents finish they start queued ones, which need awaiting too.
+        while (true) {
+            this.drainQueue();
+            const pending = [...this.agents.values()]
+                .filter(r => r.status === "running" || r.status === "queued")
+                .map(r => r.promise)
+                .filter(Boolean);
+            if (pending.length === 0)
+                break;
+            await Promise.allSettled(pending);
+        }
+    }
+    dispose() {
+        clearInterval(this.cleanupInterval);
+        // Clear queue
+        this.queue = [];
+        for (const record of this.agents.values()) {
+            record.session?.dispose();
+        }
+        this.agents.clear();
+        // Prune any orphaned git worktrees (crash recovery)
+        try {
+            pruneWorktrees(process.cwd());
+        }
+        catch { /* ignore */ }
+    }
+}