pi-subagents-lite 0.4.1 → 1.0.0

package/src/menus.ts

@@ -453,8 +453,8 @@ async function handleAgentBriefing(ctx: ExtensionCommandContext): Promise<void>
     lines.push(config.description);
     lines.push("");
 
-    if (config.builtinToolNames) {
-      lines.push(`**Tools:** ${config.builtinToolNames.join(", ")}`);
+    if (config.registeredTools) {
+      lines.push(`**Tools:** ${config.registeredTools.join(", ")}`);
     }
     if (config.model) {
       lines.push(`**Default model:** ${config.model}`);
@@ -533,7 +533,7 @@ export async function showConcurrencySettingsMenu(
     items.push(`Default concurrency limit · ${__config.concurrency.default}`);
     actions.push(async () => {
       await promptConcurrencyInput(
-        ctx, "Default concurrency limit", __config.concurrency.default,
+        ctx, "Default limit", __config.concurrency.default,
         (value) => { __config.concurrency.default = value; },
       );
     });
@@ -839,13 +839,13 @@ async function showAgentTypes(ctx: ExtensionCommandContext): Promise<void> {
   for (const name of types) {
     const cfg = getAgentConfig(name);
     if (!cfg) continue;
-    const disabled = cfg.enabled === false ? " [DISABLED]" : "";
+    const hidden = cfg.hidden === true ? " [HIDDEN]" : "";
     const model = cfg.model ? `  Model: ${cfg.model}` : "";
-    const tools = cfg.builtinToolNames
-      ? `  Tools: ${cfg.builtinToolNames.join(", ")}`
+    const tools = cfg.registeredTools
+      ? `  Tools: ${cfg.registeredTools.join(", ")}`
       : "  Tools: all built-in tools";
     const source = cfg.source ? `  Source: ${cfg.source}` : "";
-    lines.push(`  ${name}${disabled}`);
+    lines.push(`  ${name}${hidden}`);
     lines.push(`    ${cfg.description}`);
     if (model) lines.push(model);
     lines.push(tools);

package/src/model-selector.ts

@@ -1,9 +1,6 @@
 /**
  * model-selector.ts — TUI model selection dialog.
  *
- * Ported from subagent-lazy/model-selector.ts verbatim.
- * Used by the /agents menu for model selection.
- *
  * Reuses the same building blocks as pi's ModelSelectorComponent but without
  * the SettingsManager dependency — no side effects, just callbacks.
  */
@@ -18,10 +15,7 @@ import {
   Text,
 } from "@earendil-works/pi-tui";
 import { DynamicBorder } from "@earendil-works/pi-coding-agent";
-
-// Theme type from ctx.ui.custom() callback — avoid deep import that may not resolve
-// in jiti extension loader. The constructor receives the theme instance directly.
-type Theme = any;
+import type { Theme } from "./ui/agent-widget.js";
 
 /* ------------------------------------------------------------------ */
 /*  Types                                                              */

package/src/output-file.ts

@@ -1,12 +1,9 @@
 /**
  * output-file.ts — Human-readable output logging for agent transcripts.
  *
- * Forked from upstream pi-subagents. Key modifications:
- *   - Rewrote from JSONL to human-readable format
- *   - Path changed to /tmp/pi-agent-outputs/<agentId>.log (no CID-encoded nesting)
- *   - Directory created with 0o700 permissions
- *   - Append-only, human-readable, supports `tail -f`
- *   - Lines: [USER], [TOOL], [ASSISTANT], [DONE] with ISO timestamps
+ * Path: /tmp/pi-agent-outputs/<agentId>.log
+ * Append-only, human-readable, supports `tail -f`.
+ * Lines: [USER], [TOOL], [ASSISTANT], [DONE] with ISO timestamps.
  */
 
 import { appendFileSync, mkdirSync, writeFileSync } from "node:fs";

package/src/prompts.ts

@@ -7,9 +7,8 @@
 
 import type { AgentConfig, EnvInfo } from "./types.js";
 import type { SkillMeta } from "./skill-loader.js";
-export type { SkillMeta };
 
-/** Extra sections to inject into the system prompt (skills only — no memoryBlock). */
+/** Extra sections to inject into the system prompt (skills). */
 export interface PromptExtras {
   /** Preloaded skill contents to inject (full content). */
   skillBlocks?: { name: string; content: string }[];
@@ -45,7 +44,7 @@ export function buildAgentPrompt(
   envLines.push(`Platform: ${env.platform}`);
   const envBlock = envLines.join("\n");
 
-  // Build optional extras suffix (skills only — no memoryBlock)
+  // Build optional extras suffix (skills)
   const extraSections: string[] = [];
 
   // Skill metadata whitelist (like Pi's available_skills format)

package/src/result-viewer.ts

@@ -17,10 +17,7 @@ import {
 } from "@earendil-works/pi-tui";
 import { DynamicBorder } from "@earendil-works/pi-coding-agent";
 import { type LifetimeUsage, formatTokens } from "./usage.js";
-import { formatMs } from "./ui/agent-widget.js";
-
-// Theme type from ctx.ui.custom() callback
-type Theme = any;
+import { formatMs, type Theme } from "./ui/agent-widget.js";
 
 /* ------------------------------------------------------------------ */
 /*  Types                                                              */

package/src/skill-loader.ts

@@ -16,8 +16,6 @@
  * SKILL.md is a skill — we don't descend into it (Pi: skills don't nest).
  *
  * Symlinks are rejected for security (deviation from Pi, which follows them).
- *
- * Changed from upstream: imports from ./utils.js instead of ./memory.js.
  */
 
 import type { Dirent } from "node:fs";

package/src/tool-execution.ts

@@ -10,7 +10,7 @@ import type { ExtensionContext, ToolCallEvent } from "@earendil-works/pi-coding-
 import type { AgentRecord } from "./types.js";
 import type { SpawnOptions as AgentManagerSpawnOptions } from "./agent-manager.js";
 import type { AgentActivity } from "./ui/agent-widget.js";
-import { resolveType, getAgentConfig } from "./agent-types.js";
+import { resolveType, getAgentConfig, discoverNewAgents } from "./agent-types.js";
 import { resolveModel } from "./model-precedence.js";
 import { addUsage, getLifetimeTotal, getSessionContextPercent, type LifetimeUsage } from "./usage.js";
 
@@ -173,7 +173,12 @@ export async function executeAgentTool(
   ctx: ExtensionContext,
 ): Promise<any> {
   const type = (params.agent as string) || "general-purpose";
-  const resolvedType = resolveType(type);
+  let resolvedType = resolveType(type);
+  if (!resolvedType) {
+    // Not found in registry — try scanning filesystem for agents added during the session
+    await discoverNewAgents();
+    resolvedType = resolveType(type);
+  }
   if (!resolvedType) {
     return errorResult(`Unknown agent type: ${type}`);
   }
@@ -181,7 +186,6 @@ export async function executeAgentTool(
   const prompt = params.prompt as string;
   const description = params.description as string;
   const runInBackground = params.run_in_background as boolean | undefined;
-  const isolated = params.isolated as boolean | undefined;
   const maxTurns = params.max_turns as number | undefined ?? getAgentConfig(resolvedType)?.maxTurns;
   const modelStr = params.model as string | undefined;
   const model = findModelInRegistry(modelStr, ctx.modelRegistry, ctx.model);
@@ -201,7 +205,6 @@ export async function executeAgentTool(
     description,
     model,
     maxTurns,
-    isolated,
     thinkingLevel,
     modelKey,
     invocation: modelName ? { modelName } : undefined,
@@ -234,10 +237,7 @@ async function executeSpawnBackground(
   widget?.ensureTimer();
   widget?.update();
 
-  const record = manager.getRecord(agentId);
-  if (!record) {
-    return errorResult("Failed to create agent");
-  }
+  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 label = record.status === "queued" ? "Agent queued" : "Agent running";
@@ -327,11 +327,6 @@ export async function toolCallListener(
     }
   }
 
-  // Inject isolated from agent config if not explicitly passed
-  if (input.isolated === undefined && agentConfig?.isolated !== undefined) {
-    input.isolated = agentConfig.isolated;
-  }
-
   // Inject thinking from agent config if not explicitly passed
   if (input.thinking === undefined && agentConfig?.thinking !== undefined) {
     input.thinking = agentConfig.thinking;

package/src/types.ts

@@ -1,11 +1,5 @@
 /**
- * types.ts — Type definitions for the subagent system.
- *
- * Trimmed from upstream: removed ScheduledSubagent, ScheduleStoreData,
- * IsolationMode, MemoryScope, JoinMode.
- * From AgentConfig: removed memory, isolation, inheritContext, runInBackground.
- * From AgentRecord: removed groupId, joinMode, worktree, worktreeResult.
- * From AgentInvocation: removed inheritContext, isolation.
+ * Type definitions for the subagent system.
  */
 
 import type { AgentSession } from "@earendil-works/pi-coding-agent";
@@ -22,11 +16,21 @@ export interface AgentConfig {
   name: string;
   displayName?: string;
   description: string;
-  builtinToolNames?: string[];
-  /** Tool denylist — these tools are removed even if `builtinToolNames` or extensions include them. */
-  disallowedTools?: string[];
-  /** true = inherit all, string[] = only listed, false = none */
+  /** Tools to register with the session (controls availability, not LLM visibility). */
+  registeredTools?: string[];
+  /**
+   * Controls which tool schemas the LLM sees. Can reference built-in tools
+   * and extension tools. true = all, string[] = listed, false = none.
+   * Supports ext/* syntax to include all tools from an extension.
+   * Mutually exclusive with excludeTools.
+   */
+  tools?: true | string[] | false;
+  /** Tool blacklist — all tools except these are visible. Mutually exclusive with tools (when tools is string[]). */
+  excludeTools?: string[];
+  /** true = inherit all, string[] = only listed, false = none. Mutually exclusive with excludeExtensions. */
   extensions: true | string[] | false;
+  /** Extension blacklist — all extensions except these load. Mutually exclusive with extensions (when extensions is string[]). */
+  excludeExtensions?: string[];
   /** Whitelist of allowed skills (metadata only in system prompt). true = all, string[] = listed, false = none */
   skills: true | string[] | false;
   /** Skills to preload with full content into system prompt. string[] = listed, false/undefined = none */
@@ -35,14 +39,13 @@ export interface AgentConfig {
   thinking?: ThinkingLevel;
   maxTurns?: number;
   systemPrompt: string;
-  /** Default for spawn: no extension tools. undefined = caller decides. */
-  isolated?: boolean;
+
   /** true = this is an embedded default agent (informational) */
   isDefault?: boolean;
-  /** false = agent is hidden from the registry */
-  enabled?: boolean;
+  /** true = agent is hidden from the schema enum but can still be called by name. */
+  hidden?: boolean;
   /** Where this agent was loaded from */
-  source?: "default" | "project" | "global";
+  source?: "project" | "global";
 }
 
 export interface AgentRecord {
@@ -87,7 +90,6 @@ export interface AgentInvocation {
   modelName?: string;
   thinking?: ThinkingLevel;
   maxTurns?: number;
-  isolated?: boolean;
   runInBackground?: boolean;
 }
 
@@ -109,8 +111,4 @@ export interface CompactionInfo {
   tokensBefore: number;
 }
 
-/** Parsed "provider/model-id" key. */
-export interface ModelKey {
-  provider: string;
-  modelId: string;
-}
+

package/src/ui/agent-widget.ts

@@ -1,9 +1,5 @@
 /**
  * agent-widget.ts — Persistent widget showing running/completed agents above the editor.
- *
- * Ported from upstream pi-subagents.
- * Import paths use relative imports within our extension.
- * addUsage/getLifetimeTotal/getSessionContextPercent imported from ../usage.js.
  */
 
 import { truncateToWidth } from "@earendil-works/pi-tui";
@@ -62,9 +58,10 @@ const TOOL_DISPLAY: Record<string, string> = {
 
 // ---- Types ----
 
-type Theme = {
+export type Theme = {
   fg(color: string, text: string): string;
   bold(text: string): string;
+  italic?: (text: string) => string;
 };
 
 export type UICtx = {

package/src/utils.ts

@@ -1,9 +1,8 @@
 /**
- * utils.ts — Security helpers: safe file access, name validation + general utilities.
+ * utils.ts — Security helpers and general utilities.
  *
- * Security helpers extracted from upstream memory.ts — pure implementations copied verbatim.
- * General utilities (parseModelKey, findModelInRegistry) moved here so both agent-runner
- * and tool-execution can use them without circular dependencies.
+ * Security helpers (isUnsafeName, isSymlink, safeReadFile) protect against
+ * path traversal and symlink attacks in agent/skill name resolution.
  */
 
 import { lstatSync, readFileSync } from "node:fs";
@@ -35,7 +34,7 @@ export function isSymlink(filePath: string): boolean {
  */
 export function safeReadFile(filePath: string): string | undefined {
   try {
-    if (lstatSync(filePath).isSymbolicLink()) return undefined;
+    if (isSymlink(filePath)) return undefined;
     return readFileSync(filePath, "utf-8");
   } catch {
     return undefined;