@agent-harness-experimental/adapter-pi 0.0.0 → 0.0.1

package/LICENSE

@@ -0,0 +1 @@
+Copyright Vercel Inc. Pre-release. Not licensed as OSS yet.

package/README.md

@@ -1,3 +1,102 @@
-# @agent-harness-experimental/adapter-pi
+# `@agent-harness-experimental/adapter-pi`
 
-Placeholder package.
+Adapter for running the host-side [pi.dev](https://pi.dev) coding agent with
+`agent-harness-experimental`.
+
+## Install
+
+```bash
+npm install agent-harness-experimental @agent-harness-experimental/adapter-pi
+```
+
+## Usage
+
+```ts
+import { createAgentSession } from "agent-harness-experimental";
+import { piDev } from "@agent-harness-experimental/adapter-pi";
+
+const agent = createAgentSession({
+  adapter: piDev(),
+});
+
+const result = await agent.generate("Write unit tests for the auth module");
+console.log(result.text);
+await agent.close("stop");
+```
+
+## Configuration
+
+```ts
+piDev({
+  // Model ID
+  model: "claude-sonnet-4",
+
+  // Thinking/reasoning level
+  thinkingLevel: "extended",
+});
+```
+
+| Option | Type | Default | Description |
+| --- | --- | --- | --- |
+| `model` | `string` | — | Model ID |
+| `thinkingLevel` | `string` | — | Thinking/reasoning level |
+| `auth` | `object` | — | Optional per-adapter auth override (`gateway` or `customEnv`) |
+
+Sandbox runtime options such as `template`, `timeoutMs`, and `httpHandler`
+belong on `createAgentSession({ sandbox })`.
+
+```ts
+const agent = createAgentSession({
+  adapter: piDev({ model: "claude-sonnet-4", thinkingLevel: "extended" }),
+  sandbox: {
+    template: "snap_abc123",
+    timeoutMs: 60000,
+    httpHandler: async (request: Request) => fetch(request),
+  },
+});
+```
+
+## Auth
+
+`auth` is optional. If you omit it, pi.dev keeps the existing gateway fallback.
+
+```ts
+piDev({
+  model: "claude-sonnet-4",
+  auth: {
+    customEnv: {
+      env: {
+        OPENAI_API_KEY: { env: "OPENAI_TEAM_B_API_KEY" },
+        OPENAI_BASE_URL: "https://team-b.example.invalid/v1",
+      },
+    },
+  },
+});
+```
+
+Pi runs on the host and keeps model credentials outside the sandbox, so both
+`auth.gateway` and `auth.customEnv` are supported.
+
+## Built-in Tools
+
+Pi exposes standardized builtin names through `agent-harness-experimental`.
+
+| Tool | Native Tool | Description |
+| --- | --- | --- |
+| `read` | `read` | Read file contents |
+| `write` | `write` | Write content to file |
+| `edit` | `edit` | Edit file (find and replace) |
+| `bash` | `bash` | Execute shell command |
+| `grep` | `grep` | Search file contents |
+| `find` | `find` | Find files matching a glob pattern |
+| `ls` | `ls` | List directory contents |
+
+## Features
+
+- **Tool interception**: Built-in tools can be intercepted for approval via the
+  `permissions` option on `createAgentSession`
+- **Host-side runtime**: Pi runs on the host with sandbox-backed file and shell execution
+- **Durable resume**: Pi persists session state through workflow export/import
+- **Skills**: Written to `.pi/skills/{name}/SKILL.md` with YAML frontmatter
+- **Subagents**: Written to `.pi/agents/{name}.md` with model and tool metadata
+- **Token-level streaming**: Real-time text deltas

package/dist/adapter.d.ts

@@ -0,0 +1,18 @@
+import type { AgentAdapter, CustomEnvAuthOptions, GatewayAuthOptions } from 'agent-harness-experimental';
+import { type SplitPiAdapterState } from './split-runtime-support.js';
+export declare const PI_BUILTIN_TOOLS: {
+    name: string;
+    inputSchema: import("agent-harness-experimental").JsonSchema;
+    nativeName?: string | undefined;
+    description?: string | undefined;
+}[], PI_PERMISSION_TOOL_CATEGORIES: import("agent-harness-experimental").PermissionToolCategories;
+export interface PiAdapterOptions {
+    model?: string;
+    thinkingLevel?: string;
+    auth?: {
+        gateway?: GatewayAuthOptions;
+        customEnv?: CustomEnvAuthOptions;
+    };
+}
+export declare function piDev(options?: PiAdapterOptions): AgentAdapter<SplitPiAdapterState | undefined>;
+//# sourceMappingURL=adapter.d.ts.map

package/dist/adapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter.d.ts","sourceRoot":"","sources":["../src/adapter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAkB,oBAAoB,EAAE,kBAAkB,EAAc,MAAM,4BAA4B,CAAC;AAIrI,OAAO,EAA6B,KAAK,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAGjG,eAAO,MAAe,gBAAgB;;;;;KAA4B,6BAA6B,+DAiD7F,CAAC;AAEH,MAAM,WAAW,gBAAgB;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,CAAC,EAAE;QACL,OAAO,CAAC,EAAE,kBAAkB,CAAC;QAC7B,SAAS,CAAC,EAAE,oBAAoB,CAAC;KAClC,CAAC;CACH;AAsGD,wBAAgB,KAAK,CAAC,OAAO,CAAC,EAAE,gBAAgB,GAAG,YAAY,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAE/F"}

package/dist/adapter.js

@@ -0,0 +1,143 @@
+import { defineAdapterBuiltinTools, normalizeAdapterError, selectConfiguredAuthBag } from 'agent-harness-experimental';
+import { getSandboxImageName, withVercelSandboxBackend } from '@agent-harness-experimental/sandbox-vercel';
+import { getPiSandboxSetup, PI_DEFAULT_CONTEXT_WINDOW } from './setup.js';
+import { splitPiAdapterStateSchema } from './split-runtime-support.js';
+export const { tools: PI_BUILTIN_TOOLS, permissionToolCategories: PI_PERMISSION_TOOL_CATEGORIES } = defineAdapterBuiltinTools([
+    'read',
+    'write',
+    {
+        name: 'edit',
+        description: 'Edit a file by replacing a string',
+    },
+    'bash',
+    {
+        name: 'grep',
+        description: 'Search file contents',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                pattern: { type: 'string' },
+                path: { type: 'string' },
+                glob: { type: 'string' },
+                ignoreCase: { type: 'boolean' },
+                literal: { type: 'boolean' },
+                context: { type: 'number' },
+                limit: { type: 'number' },
+            },
+            required: ['pattern'],
+        },
+    },
+    {
+        name: 'find',
+        description: 'Find files matching a glob pattern',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                pattern: { type: 'string' },
+                path: { type: 'string' },
+                limit: { type: 'number' },
+            },
+            required: ['pattern'],
+        },
+    },
+    {
+        name: 'ls',
+        description: 'List directory contents',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                path: { type: 'string' },
+                limit: { type: 'number' },
+            },
+        },
+    },
+]);
+function resolvePiErrorProvider(options) {
+    if (typeof options?.model === 'string' && options.model.includes('/')) {
+        const [provider] = options.model.split('/');
+        return provider;
+    }
+    if (selectConfiguredAuthBag(options?.auth, ['customEnv', 'gateway']) === 'gateway') {
+        return 'ai-gateway';
+    }
+    return undefined;
+}
+function createPiAdapter(options) {
+    let runtime = null;
+    let runtimeLoad = null;
+    let pendingState;
+    const sandboxSetup = getPiSandboxSetup();
+    const sandboxImageName = getSandboxImageName('pi', sandboxSetup);
+    const getRuntime = async () => {
+        if (runtime) {
+            return runtime;
+        }
+        runtimeLoad ??= import('./split-runtime.js').then(({ SplitPiRuntime }) => {
+            const loadedRuntime = new SplitPiRuntime(options, sandboxImageName);
+            if (pendingState !== undefined) {
+                loadedRuntime.importState(pendingState);
+                pendingState = undefined;
+            }
+            runtime = loadedRuntime;
+            return loadedRuntime;
+        });
+        runtime = await runtimeLoad;
+        return runtime;
+    };
+    return {
+        name: 'pi',
+        supportsToolInterception: true,
+        supportsHostAccessRequests: true,
+        prepareStepSupport: 'continuation',
+        adapterStateSchema: splitPiAdapterStateSchema.optional(),
+        normalizeError(error) {
+            return normalizeAdapterError(error, {
+                adapter: 'pi',
+                provider: resolvePiErrorProvider(options),
+                model: options?.model,
+            });
+        },
+        builtinTools: PI_BUILTIN_TOOLS,
+        defaultContextWindow: PI_DEFAULT_CONTEXT_WINDOW,
+        permissionToolCategories: PI_PERMISSION_TOOL_CATEGORIES,
+        getBridgeFiles: () => sandboxSetup.getBridgeFiles(),
+        getSandboxFiles: (ctx) => sandboxSetup.getSandboxFiles(ctx),
+        resolveSandboxConfig: (sandboxConfig) => withVercelSandboxBackend(sandboxConfig),
+        exportAdapterState() {
+            return runtime?.exportState() ?? pendingState;
+        },
+        importAdapterState(state) {
+            pendingState = state;
+            runtime?.importState(state);
+        },
+        async start(ctx) {
+            await (await getRuntime()).start(ctx);
+        },
+        async prompt(message, ctx) {
+            await (await getRuntime()).prompt(message, ctx);
+        },
+        async registerTools(tools, ctx) {
+            ctx.registeredTools = tools;
+        },
+        async stop() {
+            if (!runtime) {
+                return;
+            }
+            await runtime.stop();
+            runtime = null;
+            runtimeLoad = null;
+        },
+        async detach() {
+            if (!runtime) {
+                return;
+            }
+            await runtime.detach();
+            runtime = null;
+            runtimeLoad = null;
+        },
+    };
+}
+export function piDev(options) {
+    return createPiAdapter(options);
+}
+//# sourceMappingURL=adapter.js.map

package/dist/adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapter.js","sourceRoot":"","sources":["../src/adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,yBAAyB,EAAE,qBAAqB,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACvH,OAAO,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,MAAM,4CAA4C,CAAC;AAC3G,OAAO,EAAE,iBAAiB,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AAC1E,OAAO,EAAE,yBAAyB,EAA4B,MAAM,4BAA4B,CAAC;AAGjG,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,GAAG,yBAAyB,CAAC;IAC5H,MAAM;IACN,OAAO;IACP;QACE,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,mCAAmC;KACjD;IACD,MAAM;IACN;QACE,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,sBAAsB;QACnC,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;gBAC/B,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;gBAC5B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC1B;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,oCAAoC;QACjD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC1B;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,IAAI;QACV,WAAW,EAAE,yBAAyB;QACtC,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC1B;SACF;KACF;CACF,CAAC,CAAC;AAWH,SAAS,sBAAsB,CAAC,OAA0B;IACxD,IAAI,OAAO,OAAO,EAAE,KAAK,KAAK,QAAQ,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACtE,MAAM,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC5C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,IAAI,uBAAuB,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,SAAS,CAAU,CAAC,KAAK,SAAS,EAAE,CAAC;QAC5F,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,OAA0B;IACjD,IAAI,OAAO,GAA0B,IAAI,CAAC;IAC1C,IAAI,WAAW,GAAmC,IAAI,CAAC;IACvD,IAAI,YAA6C,CAAC;IAClD,MAAM,YAAY,GAAG,iBAAiB,EAAE,CAAC;IACzC,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IAEjE,MAAM,UAAU,GAAG,KAAK,IAAI,EAAE;QAC5B,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,WAAW,KAAK,MAAM,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE;YACvE,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;YACpE,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC/B,aAAa,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;gBACxC,YAAY,GAAG,SAAS,CAAC;YAC3B,CAAC;YACD,OAAO,GAAG,aAAa,CAAC;YACxB,OAAO,aAAa,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,OAAO,GAAG,MAAM,WAAW,CAAC;QAC5B,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,IAAI;QACV,wBAAwB,EAAE,IAAI;QAC9B,0BAA0B,EAAE,IAAI;QAChC,kBAAkB,EAAE,cAAc;QAClC,kBAAkB,EAAE,yBAAyB,CAAC,QAAQ,EAAE;QACxD,cAAc,CAAC,KAAc;YAC3B,OAAO,qBAAqB,CAAC,KAAK,EAAE;gBAClC,OAAO,EAAE,IAAI;gBACb,QAAQ,EAAE,sBAAsB,CAAC,OAAO,CAAC;gBACzC,KAAK,EAAE,OAAO,EAAE,KAAK;aACtB,CAAC,CAAC;QACL,CAAC;QACD,YAAY,EAAE,gBAAgB;QAC9B,oBAAoB,EAAE,yBAAyB;QAC/C,wBAAwB,EAAE,6BAA6B;QACvD,cAAc,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE;QACnD,eAAe,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,eAAe,CAAC,GAAG,CAAC;QAC3D,oBAAoB,EAAE,CAAC,aAAa,EAAE,EAAE,CAAC,wBAAwB,CAAC,aAAa,CAAoC;QACnH,kBAAkB;YAChB,OAAO,OAAO,EAAE,WAAW,EAAE,IAAI,YAAY,CAAC;QAChD,CAAC;QACD,kBAAkB,CAAC,KAAK;YACtB,YAAY,GAAG,KAAK,CAAC;YACrB,OAAO,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QAED,KAAK,CAAC,KAAK,CAAC,GAAmB;YAC7B,MAAM,CAAC,MAAM,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,GAAmB;YAC/C,MAAM,CAAC,MAAM,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAClD,CAAC;QAED,KAAK,CAAC,aAAa,CAAC,KAAmB,EAAE,GAAmB;YAC1D,GAAG,CAAC,eAAe,GAAG,KAAK,CAAC;QAC9B,CAAC;QAED,KAAK,CAAC,IAAI;YACR,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO;YACT,CAAC;YAED,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,OAAO,GAAG,IAAI,CAAC;YACf,WAAW,GAAG,IAAI,CAAC;QACrB,CAAC;QAED,KAAK,CAAC,MAAM;YACV,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO;YACT,CAAC;YAED,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO,GAAG,IAAI,CAAC;YACf,WAAW,GAAG,IAAI,CAAC;QACrB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,KAAK,CAAC,OAA0B;IAC9C,OAAO,eAAe,CAAC,OAAO,CAAC,CAAC;AAClC,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,4 @@
+export { PI_BUILTIN_TOOLS, PI_PERMISSION_TOOL_CATEGORIES, piDev } from './adapter.js';
+export type { PiAdapterOptions } from './adapter.js';
+export { PI_DEFAULT_CONTEXT_WINDOW, PI_SANDBOX_SETUP, PI_SUPPORT_MATRIX_ROW, getPiSandboxSetup, } from './setup.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,6BAA6B,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AACtF,YAAY,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AACrD,OAAO,EACL,yBAAyB,EACzB,gBAAgB,EAChB,qBAAqB,EACrB,iBAAiB,GAClB,MAAM,YAAY,CAAC"}

package/dist/index.js

@@ -0,0 +1,3 @@
+export { PI_BUILTIN_TOOLS, PI_PERMISSION_TOOL_CATEGORIES, piDev } from './adapter.js';
+export { PI_DEFAULT_CONTEXT_WINDOW, PI_SANDBOX_SETUP, PI_SUPPORT_MATRIX_ROW, getPiSandboxSetup, } from './setup.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,6BAA6B,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AAEtF,OAAO,EACL,yBAAyB,EACzB,gBAAgB,EAChB,qBAAqB,EACrB,iBAAiB,GAClB,MAAM,YAAY,CAAC"}

package/dist/model-resolver.d.mts

@@ -0,0 +1,31 @@
+type PiBridgeConfig = {
+    workDir?: string;
+    model?: string;
+    thinkingLevel?: string;
+    agents?: Record<string, unknown>;
+};
+type PiModel = {
+    id?: string;
+    name?: string;
+    api?: string;
+    provider?: string;
+    baseUrl?: string;
+    reasoning?: boolean;
+    input?: string[];
+    cost?: {
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+    };
+    contextWindow?: number;
+    maxTokens?: number;
+};
+type PiModelRegistry = {
+    getModels?(): PiModel[];
+    getAll?(): PiModel[];
+};
+export declare const DEFAULT_PI_GATEWAY_MODEL: PiModel;
+export declare function createPiModelResolver(modelRegistry: PiModelRegistry, env?: NodeJS.ProcessEnv): (config: PiBridgeConfig, override?: string) => PiModel | undefined;
+export {};
+//# sourceMappingURL=model-resolver.d.mts.map

package/dist/model-resolver.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-resolver.d.mts","sourceRoot":"","sources":["../src/model-resolver.mts"],"names":[],"mappings":"AAAA,KAAK,cAAc,GAAG;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC,CAAC;AAEF,KAAK,OAAO,GAAG;IACb,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,IAAI,CAAC,EAAE;QACL,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,KAAK,eAAe,GAAG;IACrB,SAAS,CAAC,IAAI,OAAO,EAAE,CAAC;IACxB,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;CACtB,CAAC;AAEF,eAAO,MAAM,wBAAwB,EAAE,OAWtC,CAAC;AAEF,wBAAgB,qBAAqB,CAAC,aAAa,EAAE,eAAe,EAAE,GAAG,GAAE,MAAM,CAAC,UAAwB,IAgBhG,QAAQ,cAAc,EAAE,WAAW,MAAM,KAAG,OAAO,GAAG,SAAS,CAgBxE"}

package/dist/model-resolver.mjs

@@ -0,0 +1,42 @@
+export const DEFAULT_PI_GATEWAY_MODEL = {
+    id: 'anthropic/claude-sonnet-4',
+    name: 'Claude Sonnet 4 (Gateway)',
+    api: 'anthropic-messages',
+    provider: 'vercel-ai-gateway',
+    baseUrl: 'https://ai-gateway.vercel.sh',
+    reasoning: true,
+    input: ['text', 'image'],
+    cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+    contextWindow: 200000,
+    maxTokens: 16384,
+};
+export function createPiModelResolver(modelRegistry, env = process.env) {
+    let cachedModels;
+    const loadModels = () => {
+        if (cachedModels) {
+            return cachedModels;
+        }
+        try {
+            cachedModels = modelRegistry.getModels?.() ?? modelRegistry.getAll?.() ?? [];
+        }
+        catch {
+            cachedModels = [];
+        }
+        return cachedModels;
+    };
+    return (config, override) => {
+        const modelId = override ?? config.model;
+        let model;
+        if (modelId) {
+            model = loadModels().find((candidate) => candidate.id === modelId || candidate.name === modelId);
+        }
+        if (!model && !modelId && env.AI_GATEWAY_API_KEY) {
+            return {
+                ...DEFAULT_PI_GATEWAY_MODEL,
+                ...(env.AI_GATEWAY_BASE_URL ? { baseUrl: env.AI_GATEWAY_BASE_URL } : {}),
+            };
+        }
+        return model;
+    };
+}
+//# sourceMappingURL=model-resolver.mjs.map

package/dist/model-resolver.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-resolver.mjs","sourceRoot":"","sources":["../src/model-resolver.mts"],"names":[],"mappings":"AA8BA,MAAM,CAAC,MAAM,wBAAwB,GAAY;IAC/C,EAAE,EAAE,2BAA2B;IAC/B,IAAI,EAAE,2BAA2B;IACjC,GAAG,EAAE,oBAAoB;IACzB,QAAQ,EAAE,mBAAmB;IAC7B,OAAO,EAAE,8BAA8B;IACvC,SAAS,EAAE,IAAI;IACf,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;IACxB,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE;IAChE,aAAa,EAAE,MAAM;IACrB,SAAS,EAAE,KAAK;CACjB,CAAC;AAEF,MAAM,UAAU,qBAAqB,CAAC,aAA8B,EAAE,MAAyB,OAAO,CAAC,GAAG;IACxG,IAAI,YAAmC,CAAC;IAExC,MAAM,UAAU,GAAG,GAAc,EAAE;QACjC,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,IAAI,CAAC;YACH,YAAY,GAAG,aAAa,CAAC,SAAS,EAAE,EAAE,IAAI,aAAa,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC;QAC/E,CAAC;QAAC,MAAM,CAAC;YACP,YAAY,GAAG,EAAE,CAAC;QACpB,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC;IAEF,OAAO,CAAC,MAAsB,EAAE,QAAiB,EAAuB,EAAE;QACxE,MAAM,OAAO,GAAG,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC;QACzC,IAAI,KAA0B,CAAC;QAC/B,IAAI,OAAO,EAAE,CAAC;YACZ,KAAK,GAAG,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,KAAK,OAAO,IAAI,SAAS,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QACnG,CAAC;QAED,IAAI,CAAC,KAAK,IAAI,CAAC,OAAO,IAAI,GAAG,CAAC,kBAAkB,EAAE,CAAC;YACjD,OAAO;gBACL,GAAG,wBAAwB;gBAC3B,GAAG,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACzE,CAAC;QACJ,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;AACJ,CAAC"}

package/dist/model-resolver.mts

@@ -0,0 +1,76 @@
+type PiBridgeConfig = {
+  workDir?: string;
+  model?: string;
+  thinkingLevel?: string;
+  agents?: Record<string, unknown>;
+};
+
+type PiModel = {
+  id?: string;
+  name?: string;
+  api?: string;
+  provider?: string;
+  baseUrl?: string;
+  reasoning?: boolean;
+  input?: string[];
+  cost?: {
+    input: number;
+    output: number;
+    cacheRead: number;
+    cacheWrite: number;
+  };
+  contextWindow?: number;
+  maxTokens?: number;
+};
+
+type PiModelRegistry = {
+  getModels?(): PiModel[];
+  getAll?(): PiModel[];
+};
+
+export const DEFAULT_PI_GATEWAY_MODEL: PiModel = {
+  id: 'anthropic/claude-sonnet-4',
+  name: 'Claude Sonnet 4 (Gateway)',
+  api: 'anthropic-messages',
+  provider: 'vercel-ai-gateway',
+  baseUrl: 'https://ai-gateway.vercel.sh',
+  reasoning: true,
+  input: ['text', 'image'],
+  cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  contextWindow: 200000,
+  maxTokens: 16384,
+};
+
+export function createPiModelResolver(modelRegistry: PiModelRegistry, env: NodeJS.ProcessEnv = process.env) {
+  let cachedModels: PiModel[] | undefined;
+
+  const loadModels = (): PiModel[] => {
+    if (cachedModels) {
+      return cachedModels;
+    }
+
+    try {
+      cachedModels = modelRegistry.getModels?.() ?? modelRegistry.getAll?.() ?? [];
+    } catch {
+      cachedModels = [];
+    }
+    return cachedModels;
+  };
+
+  return (config: PiBridgeConfig, override?: string): PiModel | undefined => {
+    const modelId = override ?? config.model;
+    let model: PiModel | undefined;
+    if (modelId) {
+      model = loadModels().find((candidate) => candidate.id === modelId || candidate.name === modelId);
+    }
+
+    if (!model && !modelId && env.AI_GATEWAY_API_KEY) {
+      return {
+        ...DEFAULT_PI_GATEWAY_MODEL,
+        ...(env.AI_GATEWAY_BASE_URL ? { baseUrl: env.AI_GATEWAY_BASE_URL } : {}),
+      };
+    }
+
+    return model;
+  };
+}

package/dist/prepare-step-extension.d.ts

@@ -0,0 +1,31 @@
+import type { AdapterContext, BridgePrepareStepConfig } from 'agent-harness-experimental';
+import type { ExtensionFactory } from '@mariozechner/pi-coding-agent';
+/**
+ * Host-side bridge for per-step prepareStep on Pi.
+ *
+ * `SplitPiRuntime` owns one of these. The extension reads the current
+ * `AdapterContext` via `getCtx()` and stores the latest resolved step config.
+ *
+ * We separate the bridge object from the factory so the extension can be
+ * loaded once per session while the adapter context rotates per prompt.
+ */
+interface PrepareStepBridge {
+    getCtx: () => AdapterContext | null;
+    getInstructionsAppend: () => string | undefined;
+    setResolved: (config: BridgePrepareStepConfig) => void;
+}
+export declare function createPrepareStepBridge(): PrepareStepBridge & {
+    setCtx: (ctx: AdapterContext | null) => void;
+    setInitialConfig: (config: BridgePrepareStepConfig) => void;
+    getResolved: () => BridgePrepareStepConfig | undefined;
+    reset: () => void;
+};
+/**
+ * Build a Pi extension factory that wires `ctx.resolvePrepareStep` into Pi's
+ * per-turn lifecycle. `before_agent_start` handles the initial step resolved
+ * by the host; `turn_start` re-resolves inside the prompt loop; the resolved
+ * config shapes the outgoing system prompt and the active tool set.
+ */
+export declare function createPrepareStepExtension(bridge: PrepareStepBridge): ExtensionFactory;
+export {};
+//# sourceMappingURL=prepare-step-extension.d.ts.map

package/dist/prepare-step-extension.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prepare-step-extension.d.ts","sourceRoot":"","sources":["../src/prepare-step-extension.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAC1F,OAAO,KAAK,EAAgB,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAEpF;;;;;;;;GAQG;AACH,UAAU,iBAAiB;IACzB,MAAM,EAAE,MAAM,cAAc,GAAG,IAAI,CAAC;IACpC,qBAAqB,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;IAChD,WAAW,EAAE,CAAC,MAAM,EAAE,uBAAuB,KAAK,IAAI,CAAC;CACxD;AAED,wBAAgB,uBAAuB,IAAI,iBAAiB,GAAG;IAC7D,MAAM,EAAE,CAAC,GAAG,EAAE,cAAc,GAAG,IAAI,KAAK,IAAI,CAAC;IAC7C,gBAAgB,EAAE,CAAC,MAAM,EAAE,uBAAuB,KAAK,IAAI,CAAC;IAC5D,WAAW,EAAE,MAAM,uBAAuB,GAAG,SAAS,CAAC;IACvD,KAAK,EAAE,MAAM,IAAI,CAAC;CACnB,CAwBA;AAED;;;;;GAKG;AACH,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,iBAAiB,GAAG,gBAAgB,CAetF"}

package/dist/prepare-step-extension.js

@@ -0,0 +1,48 @@
+export function createPrepareStepBridge() {
+    let ctx = null;
+    let initial;
+    let resolved;
+    return {
+        getCtx: () => ctx,
+        getInstructionsAppend: () => resolved?.instructions ?? initial?.instructions,
+        setResolved: (config) => {
+            resolved = config;
+        },
+        setCtx: (next) => {
+            ctx = next;
+        },
+        setInitialConfig: (config) => {
+            initial = config;
+            resolved = undefined;
+        },
+        getResolved: () => resolved,
+        reset: () => {
+            initial = undefined;
+            resolved = undefined;
+        },
+    };
+}
+/**
+ * Build a Pi extension factory that wires `ctx.resolvePrepareStep` into Pi's
+ * per-turn lifecycle. `before_agent_start` handles the initial step resolved
+ * by the host; `turn_start` re-resolves inside the prompt loop; the resolved
+ * config shapes the outgoing system prompt and the active tool set.
+ */
+export function createPrepareStepExtension(bridge) {
+    return (pi) => {
+        pi.on('before_agent_start', async (event) => {
+            const append = bridge.getInstructionsAppend();
+            if (!append)
+                return undefined;
+            return { systemPrompt: `${event.systemPrompt}\n\n${append}` };
+        });
+        pi.on('turn_start', async (event) => {
+            const ctx = bridge.getCtx();
+            if (!ctx?.resolvePrepareStep)
+                return;
+            const resolved = await ctx.resolvePrepareStep(event.turnIndex);
+            bridge.setResolved(resolved);
+        });
+    };
+}
+//# sourceMappingURL=prepare-step-extension.js.map

package/dist/prepare-step-extension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prepare-step-extension.js","sourceRoot":"","sources":["../src/prepare-step-extension.ts"],"names":[],"mappings":"AAkBA,MAAM,UAAU,uBAAuB;IAMrC,IAAI,GAAG,GAA0B,IAAI,CAAC;IACtC,IAAI,OAA4C,CAAC;IACjD,IAAI,QAA6C,CAAC;IAElD,OAAO;QACL,MAAM,EAAE,GAAG,EAAE,CAAC,GAAG;QACjB,qBAAqB,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,YAAY,IAAI,OAAO,EAAE,YAAY;QAC5E,WAAW,EAAE,CAAC,MAAM,EAAE,EAAE;YACtB,QAAQ,GAAG,MAAM,CAAC;QACpB,CAAC;QACD,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACf,GAAG,GAAG,IAAI,CAAC;QACb,CAAC;QACD,gBAAgB,EAAE,CAAC,MAAM,EAAE,EAAE;YAC3B,OAAO,GAAG,MAAM,CAAC;YACjB,QAAQ,GAAG,SAAS,CAAC;QACvB,CAAC;QACD,WAAW,EAAE,GAAG,EAAE,CAAC,QAAQ;QAC3B,KAAK,EAAE,GAAG,EAAE;YACV,OAAO,GAAG,SAAS,CAAC;YACpB,QAAQ,GAAG,SAAS,CAAC;QACvB,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,0BAA0B,CAAC,MAAyB;IAClE,OAAO,CAAC,EAAgB,EAAE,EAAE;QAC1B,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC1C,MAAM,MAAM,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC;YAC9C,IAAI,CAAC,MAAM;gBAAE,OAAO,SAAS,CAAC;YAC9B,OAAO,EAAE,YAAY,EAAE,GAAG,KAAK,CAAC,YAAY,OAAO,MAAM,EAAE,EAAE,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAClC,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;YAC5B,IAAI,CAAC,GAAG,EAAE,kBAAkB;gBAAE,OAAO;YACrC,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,kBAAkB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC/D,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC"}

package/dist/remote-ops.d.ts

@@ -0,0 +1,40 @@
+import type { SandboxWorkspace } from 'agent-harness-experimental';
+interface SplitPathMapper {
+    localWorkDir: string;
+    remoteWorkDir: string;
+    toRemotePath(inputPath: string): string;
+    toRelativePath(inputPath: string): string;
+}
+interface RemoteOpsOptions {
+    sandbox: SandboxWorkspace;
+    paths: SplitPathMapper;
+    env?: Record<string, string>;
+    onFileChange?: (event: 'create' | 'modify', relativePath: string, content: Buffer) => void;
+}
+export declare function createSplitPathMapper(localWorkDir: string, remoteWorkDir: string): SplitPathMapper;
+export declare function createRemoteOperations(options: RemoteOpsOptions): {
+    paths: SplitPathMapper;
+    readBuffer: (inputPath: string) => Promise<Buffer>;
+    writeFile: (inputPath: string, content: string) => Promise<void>;
+    editFile: (inputPath: string, oldText: string, newText: string) => Promise<string>;
+    listDirectory: (inputPath?: string, limit?: number) => Promise<string[]>;
+    findFiles: (pattern: string, inputPath?: string, limit?: number) => Promise<string[]>;
+    grepFiles: (pattern: string, input: {
+        path?: string;
+        glob?: string;
+        ignoreCase?: boolean;
+        literal?: boolean;
+        context?: number;
+        limit?: number;
+    }) => Promise<string>;
+    access(inputPath: string): Promise<void>;
+    exec(command: string, cwd: string, input: {
+        onData: (data: Buffer) => void;
+        signal?: AbortSignal;
+        timeout?: number;
+    }): Promise<{
+        exitCode: number | null;
+    }>;
+};
+export {};
+//# sourceMappingURL=remote-ops.d.ts.map

package/dist/remote-ops.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"remote-ops.d.ts","sourceRoot":"","sources":["../src/remote-ops.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAInE,UAAU,eAAe;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC;IACxC,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC;CAC3C;AAED,UAAU,gBAAgB;IACxB,OAAO,EAAE,gBAAgB,CAAC;IAC1B,KAAK,EAAE,eAAe,CAAC;IACvB,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7B,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,QAAQ,GAAG,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;CAC5F;AA4BD,wBAAgB,qBAAqB,CAAC,YAAY,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,eAAe,CAmClG;AAED,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,gBAAgB;;4BA4BzB,MAAM,KAAG,OAAO,CAAC,MAAM,CAAC;2BAUzB,MAAM,WAAW,MAAM,KAAG,OAAO,CAAC,IAAI,CAAC;0BAaxC,MAAM,WAAW,MAAM,WAAW,MAAM,KAAG,OAAO,CAAC,MAAM,CAAC;2DAWjC,OAAO,CAAC,MAAM,EAAE,CAAC;yBA2B3C,MAAM,yCAAmC,OAAO,CAAC,MAAM,EAAE,CAAC;yBA+BjF,MAAM,SACR;QACL,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,OAAO,CAAC;QACrB,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,KACA,OAAO,CAAC,MAAM,CAAC;sBAqCQ,MAAM;kBAGV,MAAM,OAAO,MAAM,SAAS;QAAE,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;QAAC,MAAM,CAAC,EAAE,WAAW,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE;;;EA6B7H"}