@cortexkit/aft-pi 0.58.2 → 0.59.0

package/dist/types.d.ts

@@ -1,5 +1,6 @@
 import type { AftTransportPool } from "@cortexkit/aft-bridge";
 import type { AftConfig } from "./config.js";
+import type { OmpInternalUrlRouter } from "./omp-internal-urls.js";
 /**
  * Shared context passed to every tool wrapper.
  * Bundles the bridge pool, the resolved AFT config, and the storage dir.
@@ -12,9 +13,17 @@ import type { AftConfig } from "./config.js";
 export interface PluginContext {
     pool: AftTransportPool;
     config: AftConfig;
+    /** OMP's process-global internal URL router, absent on upstream Pi. */
+    ompRouter?: OmpInternalUrlRouter;
     /** Whether hashline edit/read mode is active for this plugin registration. */
     hashlineEffective?: boolean;
     /** Absolute path to AFT's data storage dir (e.g. ~/.local/share/cortexkit/aft). */
     storageDir: string;
+    /**
+     * Starts the work the extension defers until a session needs AFT (LSP
+     * discovery and installs, ONNX Runtime preparation, the warmup bridge).
+     * Idempotent; a no-op once the host has shut the session down.
+     */
+    startSessionWork?: () => void;
 }
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAE7C;;;;;;;;GAQG;AACH,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,EAAE,SAAS,CAAC;IAClB,8EAA8E;IAC9E,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,mFAAmF;IACnF,UAAU,EAAE,MAAM,CAAC;CACpB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAEnE;;;;;;;;GAQG;AACH,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,EAAE,SAAS,CAAC;IAClB,uEAAuE;IACvE,SAAS,CAAC,EAAE,oBAAoB,CAAC;IACjC,8EAA8E;IAC9E,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,mFAAmF;IACnF,UAAU,EAAE,MAAM,CAAC;IACnB;;;;OAIG;IACH,gBAAgB,CAAC,EAAE,MAAM,IAAI,CAAC;CAC/B"}

package/dist/workflow-hints.d.ts

@@ -39,13 +39,58 @@ interface ToolSurfaceFlags {
     bashWatch: boolean;
     bashWrite: boolean;
 }
+/**
+ * Name of the prompt section that carries the hints on hosts with structured
+ * system prompt sections. Pi renders it as
+ * `<aft_workflow_hints>\n…\n</aft_workflow_hints>` and requires names to match
+ * `^[a-z][a-z0-9_-]*$`.
+ */
+export declare const WORKFLOW_HINTS_SECTION = "aft_workflow_hints";
+/**
+ * The parts of a `before_agent_start` event the hints handler reads, across
+ * the hosts the plugin runs on:
+ * - upstream Pi 0.86.0 and later expose `systemPromptOptions.sections`, a
+ *   per-run copy that handlers may mutate, and `forceSystemPrompt`, which is
+ *   set once any handler has returned `systemPrompt` for this run;
+ * - older upstream Pi passes `systemPromptOptions` without `sections` (and it
+ *   is the session's base options object, so it must not be mutated);
+ * - oh-my-pi (OMP) passes `systemPrompt` as an array with one entry per
+ *   provider system block and has no `systemPromptOptions`.
+ */
+export interface WorkflowHintsStartEvent {
+    systemPrompt: string | readonly string[];
+    systemPromptOptions?: {
+        sections?: Record<string, string>;
+        forceSystemPrompt?: string;
+    };
+}
+export type WorkflowHintsStartResult = {
+    systemPrompt: string | string[];
+} | undefined;
+/**
+ * Add the hints block to one `before_agent_start` run.
+ *
+ * Where the host has prompt sections, the block goes into its own section and
+ * nothing is returned. Returning `systemPrompt` would force the whole prompt
+ * for the run: Pi then sends one collapsed system message and drops the
+ * transcript-backed mid-conversation tool and prompt changes, so every later
+ * tool activation rewrites the whole prompt cache. As a section, the block is
+ * diffed like any other part of the prompt and, being identical on every run,
+ * is sent once and stays in the cached prefix.
+ *
+ * Hosts without sections only accept a returned replacement prompt, so the
+ * block is appended to the prompt they passed in: as a new last entry when the
+ * prompt is an array of system blocks (OMP), keeping the existing blocks
+ * intact, or after a blank line when it is a single string (older Pi).
+ */
+export declare function applyWorkflowHints(event: WorkflowHintsStartEvent, hintsBlock: string): WorkflowHintsStartResult;
 /**
  * Register the workflow-hints extension on Pi via `before_agent_start`.
  *
- * Pi assembles a fresh system prompt for every turn, then fires
- * `before_agent_start` with the assembled prompt. Our handler appends the
- * AFT workflow hints block to that prompt. If multiple extensions return a
- * `systemPrompt`, Pi chains them — so we always append (never replace).
+ * Pi fires `before_agent_start` once per user prompt with the prompt it
+ * assembled; {@link applyWorkflowHints} adds the hints block to it without
+ * dropping other extensions' contributions. The block is built once here, so
+ * it is byte-identical on every run of the session.
  * Magic Context's short-lived Pi children don't need these workflow hints.
  */
 export declare function registerWorkflowHints(pi: ExtensionAPI, config: AftConfig, surface: ToolSurfaceFlags): void;

package/dist/workflow-hints.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"workflow-hints.d.ts","sourceRoot":"","sources":["../src/workflow-hints.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AACpE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAO7C,MAAM,WAAW,iBAAiB;IAChC,qBAAqB,EAAE,OAAO,CAAC;IAC/B,sBAAsB,EAAE,OAAO,CAAC;IAChC;;;;OAIG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,kEAAkE;IAClE,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACzB,sEAAsE;IACtE,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAID;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,oBAAoB,EAAE,OAAO,EAC7B,YAAY,GAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAoB,GACvD,MAAM,CAER;AAED,+EAA+E;AAC/E,eAAO,MAAM,wBAAwB,QAA8B,CAAC;AAEpE,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,iBAAiB,GAAG,MAAM,GAAG,IAAI,CA+HzE;AAED,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,SAAS,EACjB,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,EACxB,iBAAiB,UAAQ,GACxB,MAAM,GAAG,IAAI,CAYf;AAMD,UAAU,gBAAgB;IACxB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED;;;;;;;;GAQG;AACH,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,YAAY,EAChB,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,gBAAgB,GACxB,IAAI,CAiCN"}
+{"version":3,"file":"workflow-hints.d.ts","sourceRoot":"","sources":["../src/workflow-hints.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AACpE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAO7C,MAAM,WAAW,iBAAiB;IAChC,qBAAqB,EAAE,OAAO,CAAC;IAC/B,sBAAsB,EAAE,OAAO,CAAC;IAChC;;;;OAIG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,kEAAkE;IAClE,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACzB,sEAAsE;IACtE,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAID;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,oBAAoB,EAAE,OAAO,EAC7B,YAAY,GAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAoB,GACvD,MAAM,CAER;AAED,+EAA+E;AAC/E,eAAO,MAAM,wBAAwB,QAA8B,CAAC;AAEpE,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,iBAAiB,GAAG,MAAM,GAAG,IAAI,CA+HzE;AAED,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,SAAS,EACjB,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,EACxB,iBAAiB,UAAQ,GACxB,MAAM,GAAG,IAAI,CAYf;AAMD,UAAU,gBAAgB;IACxB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED;;;;;GAKG;AACH,eAAO,MAAM,sBAAsB,uBAAuB,CAAC;AAE3D;;;;;;;;;;GAUG;AACH,MAAM,WAAW,uBAAuB;IACtC,YAAY,EAAE,MAAM,GAAG,SAAS,MAAM,EAAE,CAAC;IACzC,mBAAmB,CAAC,EAAE;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClC,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,CAAC;CACH;AAED,MAAM,MAAM,wBAAwB,GAAG;IAAE,YAAY,EAAE,MAAM,GAAG,MAAM,EAAE,CAAA;CAAE,GAAG,SAAS,CAAC;AAEvF;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,kBAAkB,CAChC,KAAK,EAAE,uBAAuB,EAC9B,UAAU,EAAE,MAAM,GACjB,wBAAwB,CAmB1B;AAED;;;;;;;;GAQG;AACH,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,YAAY,EAChB,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,gBAAgB,GACxB,IAAI,CA8BN"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cortexkit/aft-pi",
-  "version": "0.58.2",
+  "version": "0.59.0",
   "type": "module",
   "description": "Pi coding agent extension for Agent File Tools (AFT) — tree-sitter and LSP-powered code analysis",
   "main": "dist/index.js",
@@ -15,26 +15,27 @@
     "README.md"
   ],
   "scripts": {
-    "build": "bun build src/index.ts --outdir dist --target node --format esm --external @earendil-works/pi-coding-agent --external @earendil-works/pi-ai --external @earendil-works/pi-tui --external typebox --external diff && tsc --emitDeclarationOnly",
+    "build": "bun build src/index.ts --outdir dist --target node --format esm --external @earendil-works/pi-coding-agent --external @earendil-works/pi-ai --external @earendil-works/pi-tui --external @oh-my-pi/pi-coding-agent/internal-urls --external typebox --external diff && tsc --emitDeclarationOnly",
     "typecheck": "tsc --noEmit",
     "test": "bun test src/__tests__/",
     "lint": "biome check src",
     "prepublishOnly": "bun run build",
-    "test:unit": "bun test src/__tests__ --path-ignore-patterns 'src/__tests__/e2e/**'"
+    "test:unit": "bun test src/__tests__ --path-ignore-patterns 'src/__tests__/e2e/**'",
+    "test:e2e": "bun test src/__tests__/e2e/"
   },
   "dependencies": {
-    "@cortexkit/aft-bridge": "0.58.2",
+    "@cortexkit/aft-bridge": "0.59.0",
     "comment-json": "^5.0.0",
     "diff": "^8.0.4",
     "zod": "^4.4.3"
   },
   "optionalDependencies": {
-    "@cortexkit/aft-darwin-arm64": "0.58.2",
-    "@cortexkit/aft-darwin-x64": "0.58.2",
-    "@cortexkit/aft-linux-arm64": "0.58.2",
-    "@cortexkit/aft-linux-x64": "0.58.2",
-    "@cortexkit/aft-win32-arm64": "0.58.2",
-    "@cortexkit/aft-win32-x64": "0.58.2"
+    "@cortexkit/aft-darwin-arm64": "0.59.0",
+    "@cortexkit/aft-darwin-x64": "0.59.0",
+    "@cortexkit/aft-linux-arm64": "0.59.0",
+    "@cortexkit/aft-linux-x64": "0.59.0",
+    "@cortexkit/aft-win32-arm64": "0.59.0",
+    "@cortexkit/aft-win32-x64": "0.59.0"
   },
   "devDependencies": {
     "@earendil-works/pi-coding-agent": "^0.80.2",