@cr1ms0n/pi-subagent 0.8.8 → 0.8.9

package/src/extension.ts

@@ -22,15 +22,9 @@ import {
 import { createGetPiCommand, getLaunchResolution } from "./launch.js";
 import { abortAsPromise } from "./maintenance.js";
 import { sweepSessionsLifecycle } from "./distill.js";
-import { runTasksWithContextPolicy } from "./orchestrator.js";
+import { runTasks } from "./orchestrator.js";
 import { OutputManager } from "./output.js";
-import {
-  CONTEXT_MANAGEMENT_TOOL_NAMES,
-  readContextManagementPolicy,
-  synthesisToolsForModel,
-  type ContextManagementPolicy,
-} from "./context-policy.js";
-import { parseDepth, parseSpawnPolicy, SPAWNS_ENV_VAR, validateSubagentRequest, type ResolvedTask } from "./policy.js";
+import { CONTEXT_MANAGEMENT_TOOLS, parseDepth, parseSpawnPolicy, SPAWNS_ENV_VAR, validateSubagentRequest, type ResolvedTask } from "./policy.js";
 import type { ChildRunner } from "./runner.js";
 import { ProcessLockManager } from "./process-lock.js";
 import { SessionScopedRunRegistry, snapshotFromLiveRun } from "./registry.js";
@@ -529,13 +523,7 @@ async function runSynthesis(
   runtime: SessionRuntime,
   instruction: string,
   results: TaskResult[],
-  options: {
-    runId: string;
-    modelPolicy: ModelPolicySnapshot;
-    contextPolicy: ContextManagementPolicy;
-    parentContextTools: readonly string[];
-    signal: AbortSignal;
-  },
+  options: { runId: string; modelPolicy: ModelPolicySnapshot; signal: AbortSignal },
 ): Promise<TaskResult | undefined> {
   const sections = results.map((result, index) => {
     // Typed handoff: validated structured results feed the synthesis child
@@ -582,9 +570,7 @@ async function runSynthesis(
         label: "synthesis",
         profile: "review",
         canWrite: false,
-        // Same exact target-model gate as ordinary children: a disallowed
-        // synthesis model receives no context-manager tools at all.
-        tools: synthesisToolsForModel(options.contextPolicy, approved.route.model, options.parentContextTools),
+        tools: ["read", ...CONTEXT_MANAGEMENT_TOOLS],
         model: approved.route.model,
         fallbackModels: [...approved.route.fallbackModels],
         thinking: approved.route.thinking ?? "low",
@@ -990,19 +976,12 @@ export default function registerSubagent(pi: ExtensionAPI): void {
       // restarting the parent session; no provider catalog or credentials are
       // loaded by this path.
       const dispatchConfig = loadConfig(await readConfigFile());
-      // One Remote Context snapshot per dispatch: the same operator-owned
-      // gateway allowlist gates initial validation, every fallback attempt,
-      // and the internally constructed synthesis child. Missing/invalid/off
-      // toolkit configuration fails closed to an empty allowlist.
-      const contextPolicy = await readContextManagementPolicy();
-      const parentToolNames = pi.getAllTools().map((tool) => tool.name);
-      const parentContextTools = CONTEXT_MANAGEMENT_TOOL_NAMES.filter((tool) => parentToolNames.includes(tool));
       const model = ctx.model;
       const validated = validateSubagentRequest(params, {
         cwd: ctx.cwd,
         model: model ? `${model.provider}/${model.id}` : undefined,
         thinking: pi.getThinkingLevel() as TaskSpec["thinking"],
-        availableTools: parentToolNames,
+        availableTools: pi.getAllTools().map((tool) => tool.name),
         activeTools: pi.getActiveTools(),
         depth: parseDepth(),
         sessionFile: ctx.sessionManager.getSessionFile() ?? undefined,
@@ -1014,7 +993,6 @@ export default function registerSubagent(pi: ExtensionAPI): void {
         agents: agentCatalog(runtime),
         modelPolicy: dispatchConfig.modelPolicy,
         modelPolicyError: dispatchConfig.modelPolicyError,
-        contextPolicy,
       });
       if (!validated.ok) fail(validated.error);
 
@@ -1196,7 +1174,6 @@ export default function registerSubagent(pi: ExtensionAPI): void {
       }
 
       const specs: TaskSpec[] = validated.tasks.map((task: ResolvedTask) => ({
-        backend: task.backend,
         task: task.task,
         label: task.label,
         systemPrompt: task.systemPrompt,
@@ -1285,7 +1262,7 @@ export default function registerSubagent(pi: ExtensionAPI): void {
 
       const work = (async () => {
         try {
-          const result = await runTasksWithContextPolicy(specs, {
+          const result = await runTasks(specs, {
             semaphore: runtime.semaphore,
             getPiCommand: runtime.getPiCommand,
             sessionDir: runtime.config.sessionDir,
@@ -1299,8 +1276,6 @@ export default function registerSubagent(pi: ExtensionAPI): void {
             stallAfterMs: runtime.config.stallAfterMs,
             stallKillAfterMs: runtime.config.stallKillAfterMs,
             maxRetries: runtime.config.maxRetries,
-            contextPolicy,
-            parentContextTools,
             onRunnerCreated: (index, runner) => {
               let runners = runtime.liveRunners.get(runId);
               if (!runners) runtime.liveRunners.set(runId, (runners = new Map()));
@@ -1347,8 +1322,6 @@ export default function registerSubagent(pi: ExtensionAPI): void {
             const synthesized = await runSynthesis(runtime, validated.synthesis, result.results, {
               runId,
               modelPolicy: dispatchConfig.modelPolicy!,
-              contextPolicy,
-              parentContextTools,
               signal: controller.signal,
             });
             if (synthesized) result.results = [synthesized, ...result.results];