pi-subagents-lite 1.3.0 → 1.4.1

package/src/agent-types.ts

@@ -1,184 +0,0 @@
-/**
- * agent-types.ts — Unified agent type registry.
- *
- * Merges embedded default agents with user-defined agents from .pi/agents/*.md.
- * User agents override defaults with the same name. Disabled agents are kept but excluded from spawning.
- */
-
-import { scanAgentFilesInDir, mergeAgents } from "./agent-discovery.js";
-import { DEFAULT_AGENTS } from "./default-agents.js";
-import type { AgentConfig } from "./types.js";
-
-/**
- * All tool names that Pi can provide to a session.
- *
- * Note: only `read`, `bash`, `edit`, `write` are active by default.
- * `grep` must be explicitly activated via setActiveToolsByName().
- * `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", "find"];
-
-/** Unified runtime registry of all agents (defaults + user-defined). */
-const agents = new Map<string, AgentConfig>();
-
-/**
- * Directories to scan for agent .md files at startup and on-demand.
- * Set by setAgentScanDirs() during session_start.
- */
-let userAgentDir = "";
-let projectAgentDir = "";
-
-/**
- * Register agents into the unified registry.
- * Starts with DEFAULT_AGENTS, then overlays user agents (overrides defaults with same name).
- * Hidden agents (hidden === true) are kept in the registry but excluded from spawning.
- */
-export function registerAgents(userAgents: Map<string, AgentConfig>): void {
-  agents.clear();
-
-  // Start with defaults
-  for (const [name, config] of DEFAULT_AGENTS) {
-    agents.set(name, config);
-  }
-
-  // Overlay user agents (overrides defaults with same name)
-  for (const [name, config] of userAgents) {
-    agents.set(name, config);
-  }
-}
-
-/**
- * Set the agent scan directories for on-demand discovery.
- * Called during session_start alongside scanAndRegisterAgents.
- */
-export function setAgentScanDirs(userDir: string, projectDir: string): void {
-  userAgentDir = userDir;
-  projectAgentDir = projectDir;
-}
-
-/**
- * Scan the known agent directories and register any newly discovered agents
- * that aren't already in the registry. Returns the number of new agents added.
- *
- * @param worktreeDir - Optional absolute path to a worktree's `.pi/agents/` directory.
- *   When set, agents from this directory are also scanned and added to the registry.
- *   Worktree-local types use "project" source attribution and follow the same
- *   parsing and name-uniqueness rules as the parent's project scan.
- */
-export async function discoverNewAgents(worktreeDir?: string): Promise<number> {
-  const [userAgents, projectAgents] = await Promise.all([
-    scanAgentFilesInDir(userAgentDir, "user"),
-    scanAgentFilesInDir(projectAgentDir, "project"),
-  ]);
-
-  const merged = mergeAgents(DEFAULT_AGENTS, userAgents, projectAgents);
-
-  let count = 0;
-  for (const [name, config] of merged) {
-    if (!agents.has(name)) {
-      agents.set(name, config);
-      count++;
-    }
-  }
-
-  // Scan worktree-local agents (only when worktreeDir is provided)
-  if (worktreeDir) {
-    const worktreeAgents = await scanAgentFilesInDir(worktreeDir, "project");
-    // Use mergeAgents to convert AgentConfigFromMd to AgentConfig (applies fromMd
-    // and BASE_DEFAULTS), then add only names not already in the registry.
-    const wtMerged = mergeAgents(new Map(), [], worktreeAgents);
-    for (const [name, config] of wtMerged) {
-      if (!agents.has(name)) {
-        agents.set(name, config);
-        count++;
-      }
-    }
-  }
-
-  return count;
-}
-
-/** Resolve a type name case-insensitively. Also matches displayName. Returns the canonical key or undefined. */
-export function resolveType(name: string): string | undefined {
-  if (!name) return undefined;
-  if (agents.has(name)) return name;
-  const lower = name.toLowerCase();
-  for (const [key, config] of agents.entries()) {
-    if (key.toLowerCase() === lower) return key;
-    if ((config.displayName ?? '').toLowerCase() === lower) return key;
-  }
-  return undefined;
-}
-
-/** Get the agent config for a type (case-insensitive). */
-export function getAgentConfig(name: string): AgentConfig | undefined {
-  const key = resolveType(name);
-  return key ? agents.get(key) : undefined;
-}
-
-/** Get all visible type names (for spawning and tool descriptions). */
-export function getAvailableTypes(): string[] {
-  return [...agents.entries()]
-    .filter(([_, config]) => config.hidden !== true)
-    .map(([name]) => name);
-}
-
-/** Get all type names including hidden (for UI listing). */
-export function getAllTypes(): string[] {
-  return [...agents.keys()];
-}
-
-/** Get built-in tool names for a type (case-insensitive). */
-export function getToolNamesForType(type: string): string[] {
-  const config = getAgentConfig(type);
-  return config?.registeredTools?.length
-    ? config.registeredTools
-    : [...BUILTIN_TOOL_NAMES];
-}
-
-/** Resolved config shape returned by getConfig. */
-export interface ResolvedAgentConfig {
-  displayName: string;
-  description: string;
-  registeredTools: string[];
-  /** Controls tool schema visibility. true = all, string[] = listed, false = none. */
-  tools?: true | string[] | false;
-  extensions: true | string[] | false;
-  skills: true | string[] | false;
-}
-
-function toResolved(config: AgentConfig): ResolvedAgentConfig {
-  return {
-    displayName: config.displayName ?? config.name,
-    description: config.description,
-    registeredTools: config.registeredTools ?? BUILTIN_TOOL_NAMES,
-    tools: config.tools,
-    extensions: config.extensions,
-    skills: config.skills,
-  };
-}
-
-/** Get config for a type (case-insensitive). Falls back to general-purpose. */
-export function getConfig(type: string): ResolvedAgentConfig {
-  const resolvedKey = resolveType(type);
-  const config = resolvedKey ? agents.get(resolvedKey) : undefined;
-
-  // If config exists and is not hidden, use it; otherwise fall back to general-purpose
-  const activeConfig = config?.hidden !== true
-    ? config
-    : agents.get("general-purpose");
-
-  if (activeConfig && activeConfig.hidden !== true) {
-    return toResolved(activeConfig);
-  }
-
-  // Absolute fallback — general-purpose was hidden or missing
-  return {
-    displayName: "Agent",
-    description: "General-purpose agent for complex, multi-step tasks",
-    registeredTools: BUILTIN_TOOL_NAMES,
-    extensions: true,
-    skills: true,
-  };
-}

package/src/config-mutator.ts

@@ -1,183 +0,0 @@
-/**
- * config-mutator.ts — Typed setters for all __config mutations.
- *
- * Every setter saves (saveConfigAtomic) and syncs internally.
- * menus.ts calls setters instead of directly mutating __config.
- *
- * Sync responsibilities:
- *   - Widget settings (compact, maxLines, shortcut) → syncWidgetSettings
- *   - Cost display → setShowCostEnabled (syncs to widget)
- *   - Agent bulk replace → syncWidgetSettings
- *   - Concurrency → getManager().setConcurrency()
- *   - All others → saveConfigAtomic only
- */
-
-import {
-  __config,
-  getManager,
-  setShowCostEnabled,
-  syncWidgetSettings,
-} from "./state.js";
-import { saveConfigAtomic, DEFAULT_CONFIG } from "./config-io.js";
-import { CONFIG_AGENT_NON_MODEL_KEYS } from "./types.js";
-
-// ============================================================================
-// Local helpers
-// ============================================================================
-
-/**
- * Persist concurrency config to disk and apply to the running manager.
- * Defined locally so concurrency setters don't double-save.
- */
-function applyConcurrencyConfig(): void {
-  saveConfigAtomic(__config);
-  getManager()?.setConcurrency(__config.concurrency);
-}
-
-// ============================================================================
-// Model override setters
-// ============================================================================
-
-/** Set or update a model override for a type (or "default" for global). */
-export function setModelOverride(type: string, value: string | null): void {
-  __config.agent[type] = value;
-  saveConfigAtomic(__config);
-}
-
-/** Set the global default model. */
-export function setDefaultModel(value: string | null): void {
-  __config.agent.default = value;
-  saveConfigAtomic(__config);
-}
-
-/** Clear a single per-type model override. */
-export function clearModelOverride(type: string): void {
-  delete __config.agent[type];
-  saveConfigAtomic(__config);
-}
-
-/** Clear all model overrides, preserving non-model settings. */
-export function clearAllModelOverrides(): void {
-  const preserved: Record<string, unknown> = {};
-  for (const key of CONFIG_AGENT_NON_MODEL_KEYS) {
-    const val = __config.agent[key];
-    if (val != null || key === "default" || key === "forceBackground") {
-      preserved[key] = val;
-    }
-  }
-  __config.agent = preserved as typeof __config.agent;
-  saveConfigAtomic(__config);
-  syncWidgetSettings();
-}
-
-// ============================================================================
-// Simple agent settings
-// ============================================================================
-
-/** Toggle force-background mode. */
-export function setForceBackground(enabled: boolean): void {
-  __config.agent.forceBackground = enabled;
-  saveConfigAtomic(__config);
-}
-
-/** Set the cost display toggle (syncs to widget via setShowCostEnabled). */
-export function setShowCost(enabled: boolean): void {
-  setShowCostEnabled(enabled);
-  saveConfigAtomic(__config);
-}
-
-/** Set grace turns (number of turns after timeout before hard kill). */
-export function setGraceTurns(n: number): void {
-  __config.agent.graceTurns = n;
-  saveConfigAtomic(__config);
-}
-
-// ============================================================================
-// Widget settings (sync via syncWidgetSettings)
-// ============================================================================
-
-/** Toggle force-compact widget mode. */
-export function setWidgetCompact(enabled: boolean): void {
-  __config.agent.widgetCompact = enabled;
-  saveConfigAtomic(__config);
-  syncWidgetSettings();
-}
-
-/**
- * Set max lines for full widget mode.
- * Auto-derives widgetMaxLinesCompact if not explicitly set.
- */
-export function setWidgetMaxLines(lines: number): void {
-  __config.agent.widgetMaxLines = lines;
-  if (__config.agent.widgetMaxLinesCompact === undefined) {
-    __config.agent.widgetMaxLinesCompact = Math.floor(lines / 2);
-  }
-  saveConfigAtomic(__config);
-  syncWidgetSettings();
-}
-
-/** Set max lines for compact widget mode. */
-export function setWidgetMaxLinesCompact(lines: number): void {
-  __config.agent.widgetMaxLinesCompact = lines;
-  saveConfigAtomic(__config);
-  syncWidgetSettings();
-}
-
-/** Toggle ctrl+o widget shortcut. */
-export function setWidgetShortcut(enabled: boolean): void {
-  __config.agent.widgetShortcut = enabled;
-  saveConfigAtomic(__config);
-}
-
-/** Replace the entire agent config object (used by "clear all overrides"). */
-export function setAgent(agent: typeof __config.agent): void {
-  __config.agent = agent;
-  saveConfigAtomic(__config);
-  syncWidgetSettings();
-}
-
-// ============================================================================
-// Concurrency setters (save + sync via getManager().setConcurrency)
-// ============================================================================
-
-/** Set the global concurrency default. */
-export function setConcurrencyDefault(n: number): void {
-  __config.concurrency.default = n;
-  applyConcurrencyConfig();
-}
-
-/** Set or update a per-provider concurrency limit. */
-export function setConcurrencyProvider(key: string, n: number): void {
-  const current = __config.concurrency.providers ?? {};
-  __config.concurrency.providers = { ...current, [key]: n };
-  applyConcurrencyConfig();
-}
-
-/** Set or update a per-model concurrency limit. */
-export function setConcurrencyModel(key: string, n: number): void {
-  const current = __config.concurrency.models ?? {};
-  __config.concurrency.models = { ...current, [key]: n };
-  applyConcurrencyConfig();
-}
-
-/** Remove a per-provider concurrency limit. */
-export function removeConcurrencyProvider(key: string): void {
-  if (__config.concurrency.providers) {
-    delete __config.concurrency.providers[key];
-  }
-  applyConcurrencyConfig();
-}
-
-/** Remove a per-model concurrency limit. */
-export function removeConcurrencyModel(key: string): void {
-  if (__config.concurrency.models) {
-    delete __config.concurrency.models[key];
-  }
-  applyConcurrencyConfig();
-}
-
-/** Reset all concurrency settings to defaults. */
-export function resetConcurrency(): void {
-  __config.concurrency = { ...DEFAULT_CONFIG.concurrency };
-  applyConcurrencyConfig();
-}