pi-subagents-lite 1.0.0 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-subagents-lite",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Lightweight sub-agents for pi — spawn specialized agents with isolated sessions, tools, and models.",
   "keywords": [
     "pi-package",

package/src/agent-manager.ts

@@ -115,6 +115,8 @@ export interface SpawnOptions {
   onAssistantUsage?: (usage: LifetimeUsage) => void;
   /** Called when the session successfully compacts. */
   onCompaction?: (info: CompactionInfo) => void;
+  /** Grace turns: extra turns allowed after hitting maxTurns. */
+  graceTurns?: number;
 }
 
 export class AgentManager {
@@ -304,6 +306,7 @@ export class AgentManager {
       model: options.model,
       maxTurns: options.maxTurns,
       thinkingLevel: options.thinkingLevel,
+      graceTurns: options.graceTurns,
       signal: record.abortController!.signal,
       ...this.createRecordCallbacks(record, options),
       onTurnEnd: (turnCount) => {

package/src/agent-runner.ts

@@ -29,8 +29,8 @@ import { type CompactionInfo, type EnvInfo, SHORT_ID_LENGTH, type SubagentType,
 /** Names of tools registered by this extension that subagents must NOT inherit. */
 const EXCLUDED_TOOL_NAMES = ["Agent"];
 
-/** Additional turns allowed after the soft limit steer message. */
-const GRACE_TURNS = 5;
+/** Default grace turns when not specified in config. */
+const DEFAULT_GRACE_TURNS = 6;
 
 /** Timeout for quick git commands (branch detection, repo check). */
 const GIT_EXEC_TIMEOUT_MS = 5000;
@@ -76,6 +76,8 @@ interface RunOptions {
    * pre-compaction context size estimate. Aborted compactions don't fire.
    */
   onCompaction?: (info: CompactionInfo) => void;
+  /** Grace turns: extra turns allowed after hitting maxTurns. Defaults to 6. */
+  graceTurns?: number;
 }
 
 interface RunResult {
@@ -558,6 +560,7 @@ export async function runAgent(
   const maxTurns = normalizeMaxTurns(options.maxTurns ?? agentConfig?.maxTurns);
   let softLimitReached = false;
   let aborted = false;
+  const graceTurns = options.graceTurns ?? DEFAULT_GRACE_TURNS;
 
   const unsubEvents = subscribeToSessionEvents(session, options);
 
@@ -569,7 +572,7 @@ export async function runAgent(
       if (!softLimitReached && turnCount >= maxTurns) {
         softLimitReached = true;
         session.steer("You have reached your turn limit. Wrap up immediately — provide your final answer now.");
-      } else if (softLimitReached && turnCount >= maxTurns + GRACE_TURNS) {
+      } else if (softLimitReached && turnCount >= maxTurns + graceTurns) {
         aborted = true;
         session.abort();
       }

package/src/agent-types.ts

@@ -17,7 +17,7 @@ import type { AgentConfig } from "./types.js";
  * `find` and `ls` were removed — they're thin wrappers over bash commands
  * that add ~180 tokens/turn with no real benefit.
  */
-export const BUILTIN_TOOL_NAMES: string[] = ["read", "bash", "edit", "write", "grep"];
+export const BUILTIN_TOOL_NAMES: string[] = ["read", "bash", "edit", "write", "grep", "find"];
 
 /** Unified runtime registry of all agents (defaults + user-defined). */
 const agents = new Map<string, AgentConfig>();

package/src/config-io.ts

@@ -14,7 +14,7 @@ const CONFIG_PATH = path.join(CONFIG_DIR, "subagents-lite.json");
 
 /** Default configuration — used when config file doesn't exist or is invalid. */
 export const DEFAULT_CONFIG: SubagentsConfig = {
-  agent: { default: null, forceBackground: false },
+  agent: { default: null, forceBackground: false, graceTurns: 6 },
   concurrency: { default: 4 },
 };
 

package/src/default-agents.ts

@@ -7,7 +7,7 @@
 
 import type { AgentConfig } from "./types.js";
 
-const READ_ONLY_TOOLS = ["read", "bash", "grep"];
+const READ_ONLY_TOOLS = ["read", "bash", "grep", "find"];
 
 export const DEFAULT_AGENTS: Map<string, AgentConfig> = new Map([
   [

package/src/index.ts

@@ -136,6 +136,13 @@ async function loadConfigAndRegisterAgents(ctx: ExtensionContext): Promise<void>
 // UI helpers — stats card rendering (shared by renderResult and message renderer)
 // ============================================================================
 
+/** Format agent display name with optional model: "Agent (mimo-v2.5-pro)" or "Agent". */
+function agentNameLabel(d: Record<string, unknown>, theme: Theme): string {
+  const typeName = getDisplayName((d.type as string) || "");
+  const modelName = d.modelName as string | undefined;
+  return modelName ? `${theme.bold(typeName)} (${modelName})` : theme.bold(typeName);
+}
+
 /** Build the stats line for an agent result card. Used by both renderers. */
 function buildStatsLine(d: Record<string, unknown>, theme: Theme): string {
   const parts = buildStatsParts({
@@ -205,8 +212,9 @@ function registerAgentTool(pi: ExtensionAPI): void {
       const desc = (d?.description as string) || "";
 
       if (d && d.turnCount != null) {
+        const namePart = agentNameLabel(d, theme);
         const statsLine = buildStatsLine(d, theme);
-        let lines = `${icon} ${statsLine}\n  ${theme.fg("text", desc)}`;
+        let lines = `${icon} ${namePart}·${statsLine}\n  ${theme.fg("text", desc)}`;
         if (expanded && text) {
           lines += "\n" + text.split("\n").map(l => `  ${l}`).join("\n");
         }
@@ -265,11 +273,9 @@ export default function (pi: ExtensionAPI) {
     if (d && d.turnCount != null) {
       const isError = d.status === "error" || d.status === "aborted" || d.status === "stopped";
       const icon = isError ? theme.fg("error", "✗") : theme.fg("success", "✓");
-      const typeName = getDisplayName((d.type as string) || "");
-      const modelName = d.modelName as string | undefined;
       const desc = (d.description as string) || "";
 
-      const namePart = modelName ? `${theme.bold(typeName)} (${modelName})` : theme.bold(typeName);
+      const namePart = agentNameLabel(d, theme);
       const statsLine = buildStatsLine(d, theme);
       let headerLine = `${icon} ${namePart}·${statsLine}\n  ${theme.fg("text", desc)}`;
       if ((d.outputFile as string)) {
@@ -283,13 +289,10 @@ export default function (pi: ExtensionAPI) {
         inner.addChild(new Text(resultLines, 0, 0));
       }
     } else {
-      const typeName = getDisplayName((d?.type as string) || "");
-      const modelName = d?.modelName as string | undefined;
       const desc = (d?.description as string) || "";
       let line = `${theme.fg("success", "✓")}`;
-      if (typeName) {
-        const namePart = modelName ? `${theme.bold(typeName)} (${modelName})` : theme.bold(typeName);
-        line += ` ${namePart}`;
+      if (d?.type) {
+        line += ` ${agentNameLabel(d, theme)}`;
       }
       if (desc) line += `\n  ${theme.fg("text", desc)}`;
       if (d?.outputFile) {

package/src/menus.ts

@@ -301,6 +301,26 @@ export async function showModelSettingsMenu(
       );
     });
 
+    // Grace turns setting
+    const graceTurns = __config.agent.graceTurns ?? 6;
+    items.push(`Grace turns · ${graceTurns}`);
+    actions.push(async () => {
+      const input = await ctx.ui.input("Grace turns (≥ 0)", String(graceTurns));
+      if (input === undefined) return;
+      const parsed = parseInt(input.trim(), 10);
+      if (isNaN(parsed)) {
+        ctx.ui.notify("Invalid value — must be a number", "error");
+        return;
+      }
+      if (parsed < 0) {
+        ctx.ui.notify("Invalid value — must be ≥ 0", "error");
+        return;
+      }
+      __config.agent.graceTurns = parsed;
+      saveConfigAtomic(__config);
+      ctx.ui.notify(`Grace turns set to ${parsed}`, "info");
+    });
+
     items.push("");
     actions.push(async () => {});
     items.push("─── per-type overrides ───");
@@ -375,13 +395,20 @@ export async function showModelSettingsMenu(
     items.push("Clear all overrides");
     actions.push(async () => {
       const hasOverrides = Object.entries(__config.agent).some(
-        ([k, v]) => k !== "default" && k !== "forceBackground" && v != null,
+        ([k, v]) => k !== "default" && k !== "forceBackground" && k !== "graceTurns" && v != null,
       );
       if (!hasOverrides && __config.agent.default === null) {
         ctx.ui.notify("No overrides to clear", "info");
         return;
       }
-      __config.agent = { default: __config.agent.default, forceBackground: __config.agent.forceBackground };
+      const preserved: Record<string, unknown> = {
+        default: __config.agent.default,
+        forceBackground: __config.agent.forceBackground,
+      };
+      if (__config.agent.graceTurns != null) {
+        preserved.graceTurns = __config.agent.graceTurns;
+      }
+      __config.agent = preserved as typeof __config.agent;
       saveConfigAtomic(__config);
       ctx.ui.notify("All model overrides cleared", "info");
     });

package/src/model-precedence.ts

@@ -17,7 +17,8 @@ export interface SubagentsConfig {
   agent: {
     default: string | null;
     forceBackground: boolean;
-    [agentType: string]: string | null | undefined | boolean;
+    graceTurns?: number;
+    [agentType: string]: string | null | undefined | boolean | number;
   };
   concurrency: {
     default: number;
@@ -60,10 +61,11 @@ export function resolveModel(options: ResolveModelOptions): string {
   const { subagentType, agentConfig, config, parentModelId, sessionOverrides } = options;
 
   // Precedence chain: session > config > frontmatter > parent
+  // Cast agent values: index signature includes number (graceTurns), but models are always strings
   const candidates: Array<string | boolean | null | undefined> = [
     sessionOverrides?.[subagentType],
     sessionOverrides?.["default"],
-    config.agent[subagentType],
+    config.agent[subagentType] as string | null | undefined,
     config.agent["default"],
     agentConfig?.model,
     parentModelId, // final fallback (always a valid string)

package/src/tool-execution.ts

@@ -191,11 +191,8 @@ export async function executeAgentTool(
   const model = findModelInRegistry(modelStr, ctx.modelRegistry, ctx.model);
   const modelKey = model ? `${model.provider}/${model.id}` : undefined;
 
-  // Determine modelName for invocation (only when different from parent)
-  const parentModelId = ctx.model ? `${ctx.model.provider}/${ctx.model.id}` : "";
-  const modelName = (modelKey && modelKey !== parentModelId)
-    ? parseModelKey(modelKey)?.modelId
-    : undefined;
+  // Determine modelName for invocation (always capture for display)
+  const modelName = model?.id;
 
   // Resolve thinking: explicit param > agent config (frontmatter) > undefined (inherit)
   const thinkingLevel = parseThinkingLevel(params.thinking as string | undefined)
@@ -207,7 +204,8 @@ export async function executeAgentTool(
     maxTurns,
     thinkingLevel,
     modelKey,
-    invocation: modelName ? { modelName } : undefined,
+    invocation: { modelName },
+    graceTurns: __config.agent.graceTurns,
   };
 
   if (runInBackground || __config.agent.forceBackground) {
@@ -239,7 +237,7 @@ async function executeSpawnBackground(
 
   const record = manager.getRecord(agentId)!;
   const details: Record<string, unknown> = { type: resolvedType, description: spawnOptions.description };
-  const suffix = `A notification will arrive when done - User asks you not to poll or duplicate the delegated work.\n\nAgent ID: ${agentId}`;
+  const suffix = `A notification will arrive when done - User asks you not to poll, check status or duplicate the delegated work.\n\nAgent ID: ${agentId}`;
   const label = record.status === "queued" ? "Agent queued" : "Agent running";
 
   return successResult(`[${label}] ${suffix}`, details);
@@ -318,12 +316,10 @@ export async function toolCallListener(
 
   if (effectiveModel) {
     input.model = effectiveModel;
-    // Inject _modelOverride for renderCall when model differs from parent
-    if (effectiveModel !== parentModelId) {
-      const parsed = parseModelKey(effectiveModel);
-      if (parsed) {
-        input._modelOverride = parsed.modelId;
-      }
+    // Always inject _modelOverride for renderCall
+    const parsed = parseModelKey(effectiveModel);
+    if (parsed) {
+      input._modelOverride = parsed.modelId;
     }
   }