pi-subagents-lite 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-subagents-lite",
-  "version": "1.0.1",
+  "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-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/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/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,7 @@ export async function executeAgentTool(
     maxTurns,
     thinkingLevel,
     modelKey,
-    invocation: modelName ? { modelName } : undefined,
+    invocation: { modelName },
     graceTurns: __config.agent.graceTurns,
   };
 
@@ -319,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;
     }
   }