pi-subagents-lite 1.0.0 → 1.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-subagents-lite",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "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/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/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

@@ -208,6 +208,7 @@ export async function executeAgentTool(
     thinkingLevel,
     modelKey,
     invocation: modelName ? { modelName } : undefined,
+    graceTurns: __config.agent.graceTurns,
   };
 
   if (runInBackground || __config.agent.forceBackground) {
@@ -239,7 +240,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);