@rohaquinlop/pi-subagents 0.2.0 → 0.3.0

package/README.md

@@ -159,6 +159,27 @@ Built-in tools (`read`, `write`, `edit`, `bash`, `grep`, `find`, `ls`) work auto
 
 The `subagent` tool itself is listed in `CUSTOM_TOOL_EXTENSIONS` pointing back to this extension's own `index.ts` — that's how an agent like `worker` can recursively spawn other agents. Recursion is bounded only by each agent's `subagent_agents` allowlist (e.g. worker can spawn scout/researcher, neither of which declares the `subagent` tool, so the chain stops at depth 2).
 
+### 4. Model-specific extensions (no manual mapping)
+
+Extensions that apply to specific models (not tools) can declare `appliesToModels`
+in their `package.json` and auto-load without any `CUSTOM_TOOL_EXTENSIONS` mapping:
+
+```json
+{
+  "pi": {
+    "extensions": ["./extensions/index.ts"],
+    "appliesToModels": ["deepseek-*"]
+  }
+}
+```
+
+When a subagent's configured model matches one of the patterns, the extension is
+loaded via `--extension` in the child process. Glob patterns (`*` wildcard) match
+the **model ID** (the portion after the last `/`, e.g. `deepseek-v4-flash` in
+`nan/deepseek-v4-flash`). Plain strings without wildcards match the **provider
+name** (the portion before the `/`, e.g. `nan` in `nan/deepseek-v4-flash`).
+Matching is case-insensitive.
+
 ## Structure
 
 ```

package/index.ts

@@ -156,6 +156,43 @@ function buildCustomToolExtensions(): Record<string, string> {
 
 const CUSTOM_TOOL_EXTENSIONS: Record<string, string> = buildCustomToolExtensions();
 
+interface ModelExtension {
+	patterns: string[];
+	path: string;
+}
+
+function buildModelExtensions(): ModelExtension[] {
+	const result: ModelExtension[] = [];
+	try {
+		const entries = fs.readdirSync(EXT_BASE, { withFileTypes: true });
+		for (const entry of entries) {
+			if (!entry.isDirectory()) continue;
+			const pkgPath = path.join(EXT_BASE, entry.name, "package.json");
+			if (!fs.existsSync(pkgPath)) continue;
+			try {
+				const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
+				const patterns: unknown = pkg?.pi?.appliesToModels;
+				const extEntries: unknown = pkg?.pi?.extensions;
+				if (!Array.isArray(patterns) || patterns.length === 0) continue;
+				if (!patterns.every((p: unknown): p is string => typeof p === "string")) continue;
+				if (!Array.isArray(extEntries) || extEntries.length === 0) continue;
+				if (!extEntries.every((e: unknown): e is string => typeof e === "string")) continue;
+				const extPath = path.join(EXT_BASE, entry.name, extEntries[0]);
+				if (fs.existsSync(extPath)) {
+					result.push({ patterns, path: extPath });
+				}
+			} catch {
+				// skip corrupted package.json
+			}
+		}
+	} catch {
+		// skip if EXT_BASE doesn't exist
+	}
+	return result;
+}
+
+const MODEL_EXTENSIONS: ModelExtension[] = buildModelExtensions();
+
 // ── Agent Discovery & Registration ────────────────────────────────────
 
 let agents: AgentConfig[] = [];
@@ -300,6 +337,29 @@ function truncLine(text: string, maxWidth: number): string {
 
 // ── Subagent Execution ────────────────────────────────────────────────
 
+/**
+ * Match a model string against a glob-style pattern (supports only `*` as wildcard).
+ * Pattern "deepseek-*" matches "nan/deepseek-v4-flash" and "deepseek-v4-pro".
+ * Plain strings without wildcards match the provider name (portion before `/`).
+ */
+function matchModelPattern(model: string, pattern: string): boolean {
+	const slashIdx = model.indexOf("/");
+	const hasProvider = slashIdx !== -1;
+	const modelId = hasProvider ? model.slice(slashIdx + 1) : model;
+	const provider = hasProvider ? model.slice(0, slashIdx) : "";
+
+	// Pattern without wildcards: match against provider name
+	if (!pattern.includes("*") && hasProvider) {
+		return provider.toLowerCase() === pattern.toLowerCase();
+	}
+
+	// Glob pattern: match against model ID
+	// Escape regex-special characters, then convert escaped * back to .*
+	const escaped = pattern.replace(/[.*+?^${}()|[\]\\]/g, "\\$&").replace(/\\\*/g, ".*");
+	const regex = new RegExp("^" + escaped + "$", "i");
+	return regex.test(modelId);
+}
+
 async function buildPiArgs(
 	agent: AgentConfig,
 	task: string,
@@ -348,6 +408,16 @@ async function buildPiArgs(
 		args.push("--no-tools");
 	}
 
+	// Auto-load model-specific extensions (e.g. deepseek-cache)
+	for (const me of MODEL_EXTENSIONS) {
+		for (const pattern of me.patterns) {
+			if (matchModelPattern(agent.model, pattern)) {
+				extensionPaths.add(me.path);
+				break;
+			}
+		}
+	}
+
 	for (const extPath of extensionPaths) {
 		args.push("--extension", extPath);
 	}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rohaquinlop/pi-subagents",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Pi extension for delegating tasks to subagents — parallel execution, agent discovery, and TUI rendering",
   "keywords": [
     "pi-package",