@onkernel/cua-agent 0.1.0 → 0.2.0

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## 0.2.0 - 2026-05-13
+
+- Adds `CuaAgentHarness`, a provider-aware harness API with session-backed turns, resource and prompt helpers, active tool selection, and model switching.
+- Keeps CUA runtime defaults in sync when changing models so provider-specific tools, prompts, and payload middleware update together.
+- Improves browser keyboard shortcut translation for Kernel computer actions.
+
 ## 0.1.0
 
 - Class-first CUA runtime: `CuaAgent` and `CuaHarness` on top of pi-agent-core.

package/README.md

@@ -1,9 +1,9 @@
 # `@onkernel/cua-agent`
 
-Kernel browser computer-use classes built on
-[`@earendil-works/pi-agent-core`](https://github.com/earendil-works/pi/tree/main/packages/agent).
+Kernel browser computer-use classes built on vendored pi `Agent` and
+`AgentHarness` source.
 
-This package keeps pi-agent-core semantics intact and adds browser execution
+This package keeps pi agent semantics intact and adds browser execution
 plumbing for canonical CUA tools.
 
 ## Installation
@@ -33,45 +33,59 @@ const agent = new CuaAgent({
 await agent.prompt("Open news.ycombinator.com and summarize the top story.");
 ```
 
-## Quick Start (`CuaHarness`)
+## Quick Start (`CuaAgentHarness`)
 
 ```ts
-import { CuaHarness } from "@onkernel/cua-agent";
+import { CuaAgentHarness, InMemorySessionRepo, NodeExecutionEnv } from "@onkernel/cua-agent";
 
-const harness = new CuaHarness({
+const sessionRepo = new InMemorySessionRepo();
+const session = await sessionRepo.create({ id: "example" });
+
+const harness = new CuaAgentHarness({
   browser,
   client,
+  env: new NodeExecutionEnv({ cwd: process.cwd() }),
   model: "openai:gpt-5.5",
+  session,
 });
 
-await harness.prompt("Open example.com and tell me the current URL.");
-const transcript = harness.getTranscript();
-console.log("messages in transcript:", transcript.length);
+const response = await harness.prompt("Open example.com and tell me the current URL.");
+const branch = await session.getBranch();
+const lastAssistant = [...branch]
+  .reverse()
+  .flatMap((entry) =>
+    entry.type === "message" && entry.message.role === "assistant" ? [entry.message] : [],
+  )[0];
+const assistant = lastAssistant ?? response;
+const assistantText = assistant.content
+  .flatMap((block) => (block.type === "text" ? [block.text] : []))
+  .join("")
+  .trim();
+console.log("assistant stopReason:", assistant.stopReason);
+console.log("assistant text:", assistantText || "(no text)");
 ```
 
-Use `CuaAgent` when you want direct pi `Agent` control: raw transcript state,
+Use `CuaAgent` when you want direct pi `Agent` control: raw message state,
 lifecycle events, custom streaming, and explicit prompt/continue/queue control.
 Reach for the harness shape when you want an app layer around the loop:
-session/transcript helpers, resource and prompt entry points, provider/auth
-hooks, active tool selection, compaction/tree workflows, and higher-level queue
-events. `CuaHarness` is the thin CUA version of that shape today: it installs
-CUA defaults, delegates runtime methods to the wrapped `Agent`, and adds
-`getTranscript()`.
+session-backed turns, resource and prompt entry points, provider/auth hooks,
+active tool selection, compaction/tree workflows, and higher-level queue events.
+`CuaAgentHarness` extends pi `AgentHarness`, installs CUA defaults, and refreshes
+provider-specific runtime state when `setModel()` changes models.
 
 ## Core Concepts
 
 ### Class-First API
 
 - `CuaAgent extends Agent`
-- `CuaHarness` wraps a pi `Agent` with a harness-style constructor and
-  delegated runtime methods.
+- `CuaAgentHarness extends AgentHarness`
 
 Both classes mirror pi constructor shapes and behavior, with minimal additions:
 - `browser` (Kernel browser response)
 - `client` (Kernel SDK client)
 - CUA model refs (`"provider:model"`) accepted where pi expects a concrete model
 
-If `getApiKey` is omitted, both classes default to CUA env var conventions:
+If auth callbacks are omitted, both classes default to CUA env var conventions:
 - OpenAI: `OPENAI_API_KEY`
 - Anthropic: `ANTHROPIC_OAUTH_TOKEN` or `ANTHROPIC_API_KEY`
 - Gemini: `GOOGLE_API_KEY` or `GEMINI_API_KEY`
@@ -84,6 +98,17 @@ If tools are omitted, the classes install canonical CUA computer tool executors
 using runtime specs from `@onkernel/cua-ai`. If tools are provided, they are
 used exactly.
 
+### Model Switching
+
+`CuaAgent` follows pi `Agent` semantics: assign `agent.state.model` to a
+concrete model or CUA model ref. CUA-owned tools and the default system prompt
+refresh with the new provider runtime.
+
+`CuaAgentHarness` follows pi `AgentHarness` semantics: call
+`await harness.setModel(model)`. The harness updates its model through pi's
+snapshot machinery and refreshes CUA-owned tools and default prompt state for
+the next provider request.
+
 ### Tool Composition
 
 Use `createCuaComputerTools()` to compose your own tool list from canonical
@@ -105,4 +130,4 @@ const tools = [
 ```
 
 For full event semantics, steering, follow-up queues, and tool execution
-details, see the pi-agent-core README.
+details, see the pi agent core source vendored in this package.

package/dist/agent.d.ts

@@ -1,46 +1,102 @@
-import { Agent, type AgentOptions, type AgentTool, type AgentEvent, type AgentMessage } from "@earendil-works/pi-agent-core";
-import { type ImageContent, type Api, type Model, type CuaModelRef } from "@onkernel/cua-ai";
+import { Agent, AgentHarness, type AgentHarnessOptions, type AgentOptions, type AgentState, type AgentTool, type PromptTemplate, type Skill } from "./vendor/pi-agent-core/index";
+import { type Api, type CuaModelRef, type Model, type SimpleStreamOptions } from "@onkernel/cua-ai";
 import type Kernel from "@onkernel/sdk";
 import type { KernelBrowser } from "./translator/translator";
+/** A CUA model reference string or a concrete pi model object. */
+type CuaRuntimeInput = CuaModelRef | Model<Api>;
+/**
+ * Agent state exposed by {@link CuaAgent}.
+ *
+ * It is the regular pi `AgentState`, except assigning `state.model` may use a
+ * CUA model ref such as `"openai:gpt-5.5"`. CUA-owned tools and the default
+ * system prompt are refreshed to match the new provider runtime.
+ */
+export interface CuaAgentState extends Omit<AgentState, "model"> {
+    /** The concrete pi model currently used by the underlying agent loop. */
+    get model(): Model<Api>;
+    /** Assign a concrete pi model or CUA model ref and refresh CUA runtime defaults. */
+    set model(model: CuaRuntimeInput);
+}
+/** Initial state for {@link CuaAgent}. */
 type CuaAgentInitialState = Omit<NonNullable<AgentOptions["initialState"]>, "model" | "tools"> & {
-    model: CuaModelRef | Model<Api>;
+    /** Model to use for the first turn. CUA refs are resolved before pi sees the state. */
+    model: CuaRuntimeInput;
+    /** Optional caller-owned tools. Omit this to install the provider's default CUA tools. */
     tools?: AgentTool[];
 };
+/**
+ * Constructor options for {@link CuaAgent}.
+ *
+ * `browser` and `client` are used to build the default computer-use tools.
+ * Everything else follows pi `AgentOptions`, with `initialState.model`
+ * widened to accept CUA model refs.
+ */
 export type CuaAgentOptions = Omit<AgentOptions, "initialState"> & {
+    /** Kernel browser session used by default CUA tools. */
     browser: KernelBrowser;
+    /** Kernel SDK client used by default CUA tools. */
     client: Kernel;
+    /** Initial pi state plus a CUA-aware model value. */
     initialState: CuaAgentInitialState;
 };
-export type CuaHarnessOptions = Omit<AgentOptions, "initialState"> & {
+/**
+ * Constructor options for {@link CuaAgentHarness}.
+ *
+ * The harness keeps pi `AgentHarnessOptions` intact except that `model`
+ * accepts CUA refs and `browser`/`client` are required to build default
+ * computer-use tools. Callers provide pi's `env` and `session` directly.
+ */
+export type CuaAgentHarnessOptions<TSkill extends Skill = Skill, TPromptTemplate extends PromptTemplate = PromptTemplate> = Omit<AgentHarnessOptions<TSkill, TPromptTemplate, AgentTool>, "model"> & {
+    /** Kernel browser session used by default CUA tools. */
     browser: KernelBrowser;
+    /** Kernel SDK client used by default CUA tools. */
     client: Kernel;
-    model: CuaModelRef | Model<Api>;
-    tools?: AgentTool[];
-    systemPrompt?: string;
+    /** Model used by the harness. CUA refs are resolved before pi sees the model. */
+    model: CuaRuntimeInput;
+    /** Optional payload hook composed after the provider-specific CUA payload hook. */
+    onPayload?: SimpleStreamOptions["onPayload"];
 };
+/**
+ * Pi `Agent` configured for Kernel browser computer use.
+ *
+ * Use this class when you want direct access to the lower-level pi agent state,
+ * queues, event stream, and `state.model` mutation model. It resolves CUA model
+ * refs, installs provider-appropriate CUA tools by default, and keeps those
+ * defaults in sync when `agent.state.model` changes.
+ */
 export declare class CuaAgent extends Agent {
+    private readonly runtime;
+    private stateProxy?;
     constructor(options: CuaAgentOptions);
+    /**
+     * Return a state proxy so `agent.state.model = "provider:model"` can behave
+     * like pi's normal mutable state while also re-resolving CUA tools, prompt,
+     * and payload hooks for the selected provider.
+     */
+    get state(): CuaAgentState;
+    private applyRuntime;
 }
-export declare class CuaHarness {
-    readonly agent: Agent;
-    constructor(options: CuaHarnessOptions);
-    get state(): import("@earendil-works/pi-agent-core").AgentState;
-    get steeringMode(): "all" | "one-at-a-time";
-    set steeringMode(mode: "all" | "one-at-a-time");
-    get followUpMode(): "all" | "one-at-a-time";
-    set followUpMode(mode: "all" | "one-at-a-time");
-    subscribe(listener: (event: AgentEvent, signal: AbortSignal) => Promise<void> | void): () => void;
-    prompt(message: AgentMessage | AgentMessage[]): Promise<void>;
-    prompt(input: string, images?: ImageContent[]): Promise<void>;
-    steer(message: AgentMessage): void;
-    followUp(message: AgentMessage): void;
-    continue(): Promise<void>;
-    clearSteeringQueue(): void;
-    clearFollowUpQueue(): void;
-    clearAllQueues(): void;
-    abort(): void;
-    waitForIdle(): Promise<void>;
-    getTranscript(): AgentMessage[];
+/**
+ * Pi `AgentHarness` configured for Kernel browser computer use.
+ *
+ * Use this class when you want pi's higher-level harness APIs for sessions,
+ * resources, prompt templates, queue events, compaction, and model selection.
+ * It installs provider CUA tools by default and keeps CUA-owned runtime
+ * defaults in sync through `setModel()`.
+ */
+export declare class CuaAgentHarness<TSkill extends Skill = Skill, TPromptTemplate extends PromptTemplate = PromptTemplate> extends AgentHarness<TSkill, TPromptTemplate, AgentTool> {
+    private readonly runtime;
+    private requestedActiveToolNames?;
+    constructor(options: CuaAgentHarnessOptions<TSkill, TPromptTemplate>);
+    /**
+     * Mirror pi `AgentHarness.setModel()` while accepting CUA model refs.
+     *
+     * The override refreshes CUA-owned tools before delegating to pi so the
+     * harness snapshot and session model-change entry are written with the
+     * concrete model selected by `@onkernel/cua-ai`.
+     */
+    setModel(model: CuaRuntimeInput): Promise<void>;
+    setActiveTools(toolNames: string[]): Promise<void>;
 }
 export {};
 //# sourceMappingURL=agent.d.ts.map

package/dist/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,EACL,KAAK,YAAY,EACjB,KAAK,SAAS,EACd,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACN,KAAK,YAAY,EACjB,KAAK,GAAG,EACR,KAAK,KAAK,EACV,KAAK,WAAW,EAIhB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AAExC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE7D,KAAK,oBAAoB,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,GAAG;IAChG,KAAK,EAAE,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IAChC,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,EAAE,cAAc,CAAC,GAAG;IAClE,OAAO,EAAE,aAAa,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,oBAAoB,CAAC;CACnC,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,IAAI,CAAC,YAAY,EAAE,cAAc,CAAC,GAAG;IACpE,OAAO,EAAE,aAAa,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IAChC,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,qBAAa,QAAS,SAAQ,KAAK;gBACtB,OAAO,EAAE,eAAe;CAmBpC;AAED,qBAAa,UAAU;IACtB,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;gBAEV,OAAO,EAAE,iBAAiB;IAiBtC,IAAI,KAAK,uDAER;IAED,IAAI,YAAY,IAIO,KAAK,GAAG,eAAe,CAF7C;IAED,IAAI,YAAY,CAAC,IAAI,EAAE,KAAK,GAAG,eAAe,EAE7C;IAED,IAAI,YAAY,IAIO,KAAK,GAAG,eAAe,CAF7C;IAED,IAAI,YAAY,CAAC,IAAI,EAAE,KAAK,GAAG,eAAe,EAE7C;IAED,SAAS,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,MAAM,IAAI;IAI3F,MAAM,CAAC,OAAO,EAAE,YAAY,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAC7D,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IASnE,KAAK,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAIlC,QAAQ,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAI/B,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B,kBAAkB,IAAI,IAAI;IAI1B,kBAAkB,IAAI,IAAI;IAI1B,cAAc,IAAI,IAAI;IAItB,KAAK,IAAI,IAAI;IAIP,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAIlC,aAAa,IAAI,YAAY,EAAE;CAG/B"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,EACL,YAAY,EACZ,KAAK,mBAAmB,EACxB,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,KAAK,SAAS,EACd,KAAK,cAAc,EACnB,KAAK,KAAK,EAEV,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACN,KAAK,GAAG,EACR,KAAK,WAAW,EAEhB,KAAK,KAAK,EAEV,KAAK,mBAAmB,EAExB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AAExC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE7D,kEAAkE;AAClE,KAAK,eAAe,GAAG,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;AAIhD;;;;;;GAMG;AACH,MAAM,WAAW,aAAc,SAAQ,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC;IAC/D,yEAAyE;IACzE,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC;IACxB,oFAAoF;IACpF,IAAI,KAAK,CAAC,KAAK,EAAE,eAAe,EAAE;CAClC;AAED,0CAA0C;AAC1C,KAAK,oBAAoB,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,GAAG;IAChG,uFAAuF;IACvF,KAAK,EAAE,eAAe,CAAC;IACvB,0FAA0F;IAC1F,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;CACpB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,EAAE,cAAc,CAAC,GAAG;IAClE,wDAAwD;IACxD,OAAO,EAAE,aAAa,CAAC;IACvB,mDAAmD;IACnD,MAAM,EAAE,MAAM,CAAC;IACf,qDAAqD;IACrD,YAAY,EAAE,oBAAoB,CAAC;CACnC,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,sBAAsB,CACjC,MAAM,SAAS,KAAK,GAAG,KAAK,EAC5B,eAAe,SAAS,cAAc,GAAG,cAAc,IACpD,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,eAAe,EAAE,SAAS,CAAC,EAAE,OAAO,CAAC,GAAG;IAC5E,wDAAwD;IACxD,OAAO,EAAE,aAAa,CAAC;IACvB,mDAAmD;IACnD,MAAM,EAAE,MAAM,CAAC;IACf,iFAAiF;IACjF,KAAK,EAAE,eAAe,CAAC;IACvB,mFAAmF;IACnF,SAAS,CAAC,EAAE,mBAAmB,CAAC,WAAW,CAAC,CAAC;CAC7C,CAAC;AA8DF;;;;;;;GAOG;AACH,qBAAa,QAAS,SAAQ,KAAK;IAClC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAuB;IAC/C,OAAO,CAAC,UAAU,CAAC,CAAgB;gBAEvB,OAAO,EAAE,eAAe;IA2DpC;;;;OAIG;IACH,IAAa,KAAK,IAAI,aAAa,CAalC;IAED,OAAO,CAAC,YAAY;CAWpB;AAED;;;;;;;GAOG;AACH,qBAAa,eAAe,CAC3B,MAAM,SAAS,KAAK,GAAG,KAAK,EAC5B,eAAe,SAAS,cAAc,GAAG,cAAc,CACtD,SAAQ,YAAY,CAAC,MAAM,EAAE,eAAe,EAAE,SAAS,CAAC;IACzD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAuB;IAC/C,OAAO,CAAC,wBAAwB,CAAC,CAAW;gBAEhC,OAAO,EAAE,sBAAsB,CAAC,MAAM,EAAE,eAAe,CAAC;IAsCpE;;;;;;OAMG;IACY,QAAQ,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IAS/C,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAIjE"}

package/dist/agent.js

@@ -1,95 +1,197 @@
-import { Agent, } from "@earendil-works/pi-agent-core";
+import { Agent, AgentHarness, } from "./vendor/pi-agent-core/index";
 import { getCuaEnvApiKey, resolveCuaRuntimeSpec, streamSimple, } from "@onkernel/cua-ai";
 import { createCuaComputerTools } from "./tools";
+/**
+ * Holds the CUA-specific pieces that have to change when a model changes.
+ *
+ * If callers omit `tools` or `systemPrompt`, CUA owns those values and refreshes
+ * them from `@onkernel/cua-ai` whenever the model changes. If callers pass
+ * their own tools or prompt, the controller preserves those caller-owned values.
+ */
+class CuaRuntimeController {
+    options;
+    runtimeSpec;
+    constructor(options) {
+        this.options = options;
+        this.runtimeSpec = resolveCuaRuntimeSpec(options.model);
+    }
+    get model() {
+        return this.runtimeSpec.model;
+    }
+    get ownsTools() {
+        return this.options.tools === undefined;
+    }
+    get ownsSystemPrompt() {
+        return this.options.systemPrompt === undefined;
+    }
+    get systemPrompt() {
+        return this.runtimeSpec.defaultSystemPrompt;
+    }
+    setModel(model) {
+        this.runtimeSpec = resolveCuaRuntimeSpec(model);
+    }
+    tools() {
+        return (this.options.tools ??
+            createCuaComputerTools({
+                browser: this.options.browser,
+                client: this.options.client,
+                toolDefinitions: this.runtimeSpec.toolDefinitions,
+            }));
+    }
+    onPayloadFor(model) {
+        const runtimeSpec = resolveCuaRuntimeSpec(model);
+        return composeOnPayload(runtimeSpec.onPayload, this.options.onPayload);
+    }
+}
+/**
+ * Pi `Agent` configured for Kernel browser computer use.
+ *
+ * Use this class when you want direct access to the lower-level pi agent state,
+ * queues, event stream, and `state.model` mutation model. It resolves CUA model
+ * refs, installs provider-appropriate CUA tools by default, and keeps those
+ * defaults in sync when `agent.state.model` changes.
+ */
 export class CuaAgent extends Agent {
+    runtime;
+    stateProxy;
     constructor(options) {
-        const { browser, client, initialState, onPayload, streamFn, ...agentOptions } = options;
-        const runtimeSpec = resolveCuaRuntimeSpec(initialState.model);
-        const tools = initialState.tools ?? createCuaComputerTools({ browser, client, toolDefinitions: runtimeSpec.toolDefinitions });
-        const systemPrompt = initialState.systemPrompt ?? runtimeSpec.defaultSystemPrompt;
+        const { browser, client, initialState, onPayload, streamFn, prepareNextTurn, ...agentOptions } = options;
+        const runtime = new CuaRuntimeController({
+            browser,
+            client,
+            model: initialState.model,
+            tools: initialState.tools,
+            systemPrompt: initialState.systemPrompt,
+            onPayload,
+        });
+        const wrappedStreamFn = (model, context, streamOptions) => (streamFn ?? streamSimple)(model, context, {
+            ...streamOptions,
+            onPayload: runtime.onPayloadFor(model),
+        });
         super({
             ...agentOptions,
             getApiKey: agentOptions.getApiKey ?? getCuaEnvApiKey,
-            streamFn: streamFn ?? streamSimple,
-            onPayload: composeOnPayload(runtimeSpec.onPayload, onPayload),
+            streamFn: wrappedStreamFn,
             initialState: {
                 ...initialState,
-                model: runtimeSpec.model,
-                tools,
-                systemPrompt,
+                model: runtime.model,
+                tools: runtime.tools(),
+                systemPrompt: initialState.systemPrompt ?? runtime.systemPrompt,
             },
         });
+        this.runtime = runtime;
+        /**
+         * pi calls `prepareNextTurn` between provider requests. Wrapping it lets CUA
+         * honor any user-provided turn update while also refreshing provider-specific
+         * defaults if that update changes the model.
+         */
+        this.prepareNextTurn = async (signal) => {
+            const update = await prepareNextTurn?.(signal);
+            if (update?.model) {
+                this.applyRuntime(update.model);
+            }
+            const state = super.state;
+            const context = update?.context ?? {
+                systemPrompt: state.systemPrompt,
+                messages: state.messages.slice(),
+                tools: state.tools.slice(),
+            };
+            return {
+                ...update,
+                model: state.model,
+                context: {
+                    ...context,
+                    systemPrompt: this.runtime.ownsSystemPrompt ? state.systemPrompt : context.systemPrompt,
+                    tools: this.runtime.ownsTools ? state.tools.slice() : context.tools,
+                },
+            };
+        };
+    }
+    /**
+     * Return a state proxy so `agent.state.model = "provider:model"` can behave
+     * like pi's normal mutable state while also re-resolving CUA tools, prompt,
+     * and payload hooks for the selected provider.
+     */
+    get state() {
+        if (!this.stateProxy) {
+            this.stateProxy = new Proxy(super.state, {
+                set: (target, prop, value, receiver) => {
+                    if (prop === "model") {
+                        this.applyRuntime(value);
+                        return true;
+                    }
+                    return Reflect.set(target, prop, value, receiver);
+                },
+            });
+        }
+        return this.stateProxy;
+    }
+    applyRuntime(model) {
+        this.runtime.setModel(model);
+        const state = super.state;
+        state.model = this.runtime.model;
+        if (this.runtime.ownsTools) {
+            state.tools = this.runtime.tools();
+        }
+        if (this.runtime.ownsSystemPrompt) {
+            state.systemPrompt = this.runtime.systemPrompt;
+        }
     }
 }
-export class CuaHarness {
-    agent;
+/**
+ * Pi `AgentHarness` configured for Kernel browser computer use.
+ *
+ * Use this class when you want pi's higher-level harness APIs for sessions,
+ * resources, prompt templates, queue events, compaction, and model selection.
+ * It installs provider CUA tools by default and keeps CUA-owned runtime
+ * defaults in sync through `setModel()`.
+ */
+export class CuaAgentHarness extends AgentHarness {
+    runtime;
+    requestedActiveToolNames;
     constructor(options) {
-        const { browser, client, model, tools, systemPrompt, onPayload, streamFn, ...agentOptions } = options;
-        const runtimeSpec = resolveCuaRuntimeSpec(model);
-        const resolvedTools = tools ?? createCuaComputerTools({ browser, client, toolDefinitions: runtimeSpec.toolDefinitions });
-        this.agent = new Agent({
-            ...agentOptions,
-            getApiKey: agentOptions.getApiKey ?? getCuaEnvApiKey,
-            streamFn: streamFn ?? streamSimple,
-            onPayload: composeOnPayload(runtimeSpec.onPayload, onPayload),
-            initialState: {
-                model: runtimeSpec.model,
-                tools: resolvedTools,
-                systemPrompt: systemPrompt ?? runtimeSpec.defaultSystemPrompt,
-            },
+        const { browser, client, model, tools, systemPrompt, getApiKeyAndHeaders, onPayload, activeToolNames, ...harnessOptions } = options;
+        const runtime = new CuaRuntimeController({ browser, client, model, tools, systemPrompt, onPayload });
+        const resolvedTools = runtime.tools();
+        super({
+            ...harnessOptions,
+            model: runtime.model,
+            tools: resolvedTools,
+            systemPrompt: systemPrompt ?? (() => runtime.systemPrompt),
+            getApiKeyAndHeaders: getApiKeyAndHeaders ??
+                (async (requestModel) => {
+                    const apiKey = getCuaEnvApiKey(requestModel.provider);
+                    return apiKey ? { apiKey } : undefined;
+                }),
+            activeToolNames: activeToolNames ?? resolvedTools.map((tool) => tool.name),
+        });
+        this.runtime = runtime;
+        this.requestedActiveToolNames = activeToolNames;
+        this.on("before_provider_payload", async ({ model, payload }) => {
+            const onPayload = this.runtime.onPayloadFor(model);
+            if (!onPayload)
+                return { payload };
+            return { payload: (await onPayload(payload, model)) ?? payload };
         });
     }
-    get state() {
-        return this.agent.state;
-    }
-    get steeringMode() {
-        return this.agent.steeringMode;
-    }
-    set steeringMode(mode) {
-        this.agent.steeringMode = mode;
-    }
-    get followUpMode() {
-        return this.agent.followUpMode;
-    }
-    set followUpMode(mode) {
-        this.agent.followUpMode = mode;
-    }
-    subscribe(listener) {
-        return this.agent.subscribe(listener);
-    }
-    async prompt(input, images) {
-        if (typeof input === "string") {
-            await this.agent.prompt(input, images);
-            return;
+    /**
+     * Mirror pi `AgentHarness.setModel()` while accepting CUA model refs.
+     *
+     * The override refreshes CUA-owned tools before delegating to pi so the
+     * harness snapshot and session model-change entry are written with the
+     * concrete model selected by `@onkernel/cua-ai`.
+     */
+    async setModel(model) {
+        this.runtime.setModel(model);
+        if (this.runtime.ownsTools) {
+            const tools = this.runtime.tools();
+            await super.setTools(tools, this.requestedActiveToolNames ?? tools.map((tool) => tool.name));
         }
-        await this.agent.prompt(input);
-    }
-    steer(message) {
-        this.agent.steer(message);
-    }
-    followUp(message) {
-        this.agent.followUp(message);
-    }
-    async continue() {
-        await this.agent.continue();
-    }
-    clearSteeringQueue() {
-        this.agent.clearSteeringQueue();
-    }
-    clearFollowUpQueue() {
-        this.agent.clearFollowUpQueue();
-    }
-    clearAllQueues() {
-        this.agent.clearAllQueues();
-    }
-    abort() {
-        this.agent.abort();
-    }
-    async waitForIdle() {
-        await this.agent.waitForIdle();
+        await super.setModel(this.runtime.model);
     }
-    getTranscript() {
-        return [...this.agent.state.messages];
+    async setActiveTools(toolNames) {
+        await super.setActiveTools(toolNames);
+        this.requestedActiveToolNames = [...toolNames];
     }
 }
 function composeOnPayload(first, second) {

package/dist/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,GAKL,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAKN,eAAe,EACf,qBAAqB,EACrB,YAAY,GACZ,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAsBjD,MAAM,OAAO,QAAS,SAAQ,KAAK;IAClC,YAAY,OAAwB;QACnC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC;QACxF,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,sBAAsB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,WAAW,CAAC,eAAe,EAAE,CAAC,CAAC;QAC9H,MAAM,YAAY,GAAG,YAAY,CAAC,YAAY,IAAI,WAAW,CAAC,mBAAmB,CAAC;QAElF,KAAK,CAAC;YACL,GAAG,YAAY;YACf,SAAS,EAAE,YAAY,CAAC,SAAS,IAAI,eAAe;YACpD,QAAQ,EAAE,QAAQ,IAAI,YAAY;YAClC,SAAS,EAAE,gBAAgB,CAAC,WAAW,CAAC,SAAS,EAAE,SAAS,CAAC;YAC7D,YAAY,EAAE;gBACb,GAAG,YAAY;gBACf,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,KAAK;gBACL,YAAY;aACZ;SACD,CAAC,CAAC;IACJ,CAAC;CACD;AAED,MAAM,OAAO,UAAU;IACb,KAAK,CAAQ;IAEtB,YAAY,OAA0B;QACrC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC;QACtG,MAAM,WAAW,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,aAAa,GAAG,KAAK,IAAI,sBAAsB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,WAAW,CAAC,eAAe,EAAE,CAAC,CAAC;QACzH,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,GAAG,YAAY;YACf,SAAS,EAAE,YAAY,CAAC,SAAS,IAAI,eAAe;YACpD,QAAQ,EAAE,QAAQ,IAAI,YAAY;YAClC,SAAS,EAAE,gBAAgB,CAAC,WAAW,CAAC,SAAS,EAAE,SAAS,CAAC;YAC7D,YAAY,EAAE;gBACb,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,KAAK,EAAE,aAAa;gBACpB,YAAY,EAAE,YAAY,IAAI,WAAW,CAAC,mBAAmB;aAC7D;SACD,CAAC,CAAC;IACJ,CAAC;IAED,IAAI,KAAK;QACR,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;IACzB,CAAC;IAED,IAAI,YAAY;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IAChC,CAAC;IAED,IAAI,YAAY,CAAC,IAA6B;QAC7C,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,IAAI,YAAY;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IAChC,CAAC;IAED,IAAI,YAAY,CAAC,IAA6B;QAC7C,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,QAA0E;QACnF,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAID,KAAK,CAAC,MAAM,CAAC,KAA6C,EAAE,MAAuB;QAClF,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC/B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACvC,OAAO;QACR,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,OAAqB;QAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED,QAAQ,CAAC,OAAqB;QAC7B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,QAAQ;QACb,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC;IAED,kBAAkB;QACjB,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACjC,CAAC;IAED,kBAAkB;QACjB,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;IACjC,CAAC;IAED,cAAc;QACb,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;IAC7B,CAAC;IAED,KAAK;QACJ,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;IAChC,CAAC;IAED,aAAa;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;CACD;AAED,SAAS,gBAAgB,CAAC,KAAgC,EAAE,MAAiC;IAC5F,IAAI,CAAC,KAAK;QAAE,OAAO,MAAM,CAAC;IAC1B,IAAI,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IAC1B,OAAO,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;QAClC,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC,CAAC;AACH,CAAC"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,EACL,YAAY,GAQZ,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAGN,eAAe,EAEf,qBAAqB,EAErB,YAAY,GACZ,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAmEjD;;;;;;GAMG;AACH,MAAM,oBAAoB;IAIP;IAHV,WAAW,CAAiB;IAEpC,YACkB,OAOhB;QAPgB,YAAO,GAAP,OAAO,CAOvB;QAED,IAAI,CAAC,WAAW,GAAG,qBAAqB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACzD,CAAC;IAED,IAAI,KAAK;QACR,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC;IAC/B,CAAC;IAED,IAAI,SAAS;QACZ,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC;IACzC,CAAC;IAED,IAAI,gBAAgB;QACnB,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC;IAChD,CAAC;IAED,IAAI,YAAY;QACf,OAAO,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC;IAC7C,CAAC;IAED,QAAQ,CAAC,KAAsB;QAC9B,IAAI,CAAC,WAAW,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;IAED,KAAK;QACJ,OAAO,CACN,IAAI,CAAC,OAAO,CAAC,KAAK;YAClB,sBAAsB,CAAC;gBACtB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;gBAC7B,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC3B,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,eAAe;aACjD,CAAC,CACF,CAAC;IACH,CAAC;IAED,YAAY,CAAC,KAAsB;QAClC,MAAM,WAAW,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACjD,OAAO,gBAAgB,CAAC,WAAW,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACxE,CAAC;CACD;AAED;;;;;;;GAOG;AACH,MAAM,OAAO,QAAS,SAAQ,KAAK;IACjB,OAAO,CAAuB;IACvC,UAAU,CAAiB;IAEnC,YAAY,OAAwB;QACnC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,YAAY,EAAE,GAAG,OAAO,CAAC;QACzG,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;YACxC,OAAO;YACP,MAAM;YACN,KAAK,EAAE,YAAY,CAAC,KAAK;YACzB,KAAK,EAAE,YAAY,CAAC,KAAK;YACzB,YAAY,EAAE,YAAY,CAAC,YAAY;YACvC,SAAS;SACT,CAAC,CAAC;QACH,MAAM,eAAe,GAAa,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CACnE,CAAC,QAAQ,IAAI,YAAY,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE;YAC1C,GAAG,aAAa;YAChB,SAAS,EAAE,OAAO,CAAC,YAAY,CAAC,KAAmB,CAAC;SACpD,CAAC,CAAC;QAEJ,KAAK,CAAC;YACL,GAAG,YAAY;YACf,SAAS,EAAE,YAAY,CAAC,SAAS,IAAI,eAAe;YACpD,QAAQ,EAAE,eAAe;YACzB,YAAY,EAAE;gBACb,GAAG,YAAY;gBACf,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE;gBACtB,YAAY,EAAE,YAAY,CAAC,YAAY,IAAI,OAAO,CAAC,YAAY;aAC/D;SACD,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB;;;;WAIG;QACH,IAAI,CAAC,eAAe,GAAG,KAAK,EAAE,MAA+B,EAAE,EAAE;YAChE,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC;YAC/C,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;gBACnB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAwB,CAAC,CAAC;YACpD,CAAC;YAED,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YAC1B,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,IAAI;gBAClC,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE;gBAChC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE;aAC1B,CAAC;YAEF,OAAO;gBACN,GAAG,MAAM;gBACT,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,OAAO,EAAE;oBACR,GAAG,OAAO;oBACV,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY;oBACvF,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK;iBACnE;aACD,CAAC;QACH,CAAC,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,IAAa,KAAK;QACjB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACtB,IAAI,CAAC,UAAU,GAAG,IAAI,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE;gBACxC,GAAG,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;oBACtC,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;wBACtB,IAAI,CAAC,YAAY,CAAC,KAAwB,CAAC,CAAC;wBAC5C,OAAO,IAAI,CAAC;oBACb,CAAC;oBACD,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBACnD,CAAC;aACD,CAAkB,CAAC;QACrB,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC;IACxB,CAAC;IAEO,YAAY,CAAC,KAAsB;QAC1C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC7B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;QACjC,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC5B,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YACnC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;QAChD,CAAC;IACF,CAAC;CACD;AAED;;;;;;;GAOG;AACH,MAAM,OAAO,eAGX,SAAQ,YAAgD;IACxC,OAAO,CAAuB;IACvC,wBAAwB,CAAY;IAE5C,YAAY,OAAwD;QACnE,MAAM,EACL,OAAO,EACP,MAAM,EACN,KAAK,EACL,KAAK,EACL,YAAY,EACZ,mBAAmB,EACnB,SAAS,EACT,eAAe,EACf,GAAG,cAAc,EACjB,GAAG,OAAO,CAAC;QACZ,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC;QACrG,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;QAEtC,KAAK,CAAC;YACL,GAAG,cAAc;YACjB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,KAAK,EAAE,aAAa;YACpB,YAAY,EAAE,YAAY,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,YAAY,CAAC;YAC1D,mBAAmB,EAClB,mBAAmB;gBACnB,CAAC,KAAK,EAAE,YAAwB,EAAE,EAAE;oBACnC,MAAM,MAAM,GAAG,eAAe,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;oBACtD,OAAO,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBACxC,CAAC,CAAC;YACH,eAAe,EAAE,eAAe,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SAC1E,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,wBAAwB,GAAG,eAAe,CAAC;QAChD,IAAI,CAAC,EAAE,CAAC,yBAAyB,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,EAA2C,EAAE,EAAE;YACxG,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,KAAmB,CAAC,CAAC;YACjE,IAAI,CAAC,SAAS;gBAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YACnC,OAAO,EAAE,OAAO,EAAE,CAAC,MAAM,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,IAAI,OAAO,EAAE,CAAC;QAClE,CAAC,CAAC,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACM,KAAK,CAAC,QAAQ,CAAC,KAAsB;QAC7C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,MAAM,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,wBAAwB,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9F,CAAC;QACD,MAAM,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC;IAEQ,KAAK,CAAC,cAAc,CAAC,SAAmB;QAChD,MAAM,KAAK,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QACtC,IAAI,CAAC,wBAAwB,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC;IAChD,CAAC;CACD;AAED,SAAS,gBAAgB,CAAC,KAAgC,EAAE,MAAiC;IAC5F,IAAI,CAAC,KAAK;QAAE,OAAO,MAAM,CAAC;IAC1B,IAAI,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IAC1B,OAAO,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;QAClC,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -1,8 +1,8 @@
-export * from "@earendil-works/pi-agent-core";
+export * from "./vendor/pi-agent-core/index";
 export type { KernelBrowser } from "./translator/translator";
 export { createCuaComputerTools } from "./tools";
 export type { BatchDetails, ComputerToolOptions, CuaExecutorTool, NavigationDetails, SupportedCuaExecutorToolName, } from "./tools";
 export { SUPPORTED_CUA_EXECUTOR_TOOL_NAMES } from "./tools";
-export { CuaAgent, CuaHarness } from "./agent";
-export type { CuaAgentOptions, CuaHarnessOptions } from "./agent";
+export { CuaAgent, CuaAgentHarness } from "./agent";
+export type { CuaAgentHarnessOptions, CuaAgentOptions, CuaAgentState } from "./agent";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,+BAA+B,CAAC;AAE9C,YAAY,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AACjD,YAAY,EACX,YAAY,EACZ,mBAAmB,EACnB,eAAe,EACf,iBAAiB,EACjB,4BAA4B,GAC5B,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,iCAAiC,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAC/C,YAAY,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,8BAA8B,CAAC;AAE7C,YAAY,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AACjD,YAAY,EACX,YAAY,EACZ,mBAAmB,EACnB,eAAe,EACf,iBAAiB,EACjB,4BAA4B,GAC5B,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,iCAAiC,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AACpD,YAAY,EAAE,sBAAsB,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC"}

package/dist/index.js

@@ -1,5 +1,5 @@
-export * from "@earendil-works/pi-agent-core";
+export * from "./vendor/pi-agent-core/index";
 export { createCuaComputerTools } from "./tools";
 export { SUPPORTED_CUA_EXECUTOR_TOOL_NAMES } from "./tools";
-export { CuaAgent, CuaHarness } from "./agent";
+export { CuaAgent, CuaAgentHarness } from "./agent";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,+BAA+B,CAAC;AAG9C,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAQjD,OAAO,EAAE,iCAAiC,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,8BAA8B,CAAC;AAG7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAQjD,OAAO,EAAE,iCAAiC,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC"}

package/dist/tools.d.ts

@@ -1,8 +1,8 @@
 import type Kernel from "@onkernel/sdk";
-import type { AgentTool } from "@earendil-works/pi-agent-core";
 import type { Tool } from "@earendil-works/pi-ai";
 import { CuaBatchSchema, CuaNavigationSchema } from "@onkernel/cua-ai";
 import { type KernelBrowser } from "./translator/translator";
+import type { AgentTool } from "./vendor/pi-agent-core/index";
 export interface ComputerToolOptions {
     browser: KernelBrowser;
     client: Kernel;

package/dist/tools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,EAAE,SAAS,EAAmB,MAAM,+BAA+B,CAAC;AAChF,OAAO,KAAK,EAA6B,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAC7E,OAAO,EAGN,cAAc,EACd,mBAAmB,EAGnB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAA8B,KAAK,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAEzF,MAAM,WAAW,mBAAmB;IACnC,OAAO,EAAE,aAAa,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,EAAE,IAAI,EAAE,CAAC;CACxB;AAED,eAAO,MAAM,iCAAiC,2DAA2D,CAAC;AAC1G,MAAM,MAAM,4BAA4B,GAAG,CAAC,OAAO,iCAAiC,CAAC,CAAC,MAAM,CAAC,CAAC;AAI9F,MAAM,WAAW,YAAY;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,KAAK,CAAC;QAAC,GAAG,EAAE,MAAM,CAAA;KAAE,GAAG;QAAE,IAAI,EAAE,YAAY,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG;QAAE,IAAI,EAAE,iBAAiB,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC7I,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,iBAAiB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,KAAK,SAAS,GAAG,SAAS,CAAC,OAAO,cAAc,EAAE,YAAY,CAAC,CAAC;AAChE,KAAK,cAAc,GAAG,SAAS,CAAC,OAAO,mBAAmB,EAAE,iBAAiB,CAAC,CAAC;AAC/E,MAAM,MAAM,eAAe,GAAG,SAAS,GAAG,cAAc,CAAC;AAEzD,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,mBAAmB,GAAG,eAAe,EAAE,CAGnF"}
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,EAA6B,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAC7E,OAAO,EAGN,cAAc,EACd,mBAAmB,EAGnB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAA8B,KAAK,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACzF,OAAO,KAAK,EAAE,SAAS,EAAmB,MAAM,8BAA8B,CAAC;AAE/E,MAAM,WAAW,mBAAmB;IACnC,OAAO,EAAE,aAAa,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,EAAE,IAAI,EAAE,CAAC;CACxB;AAED,eAAO,MAAM,iCAAiC,2DAA2D,CAAC;AAC1G,MAAM,MAAM,4BAA4B,GAAG,CAAC,OAAO,iCAAiC,CAAC,CAAC,MAAM,CAAC,CAAC;AAI9F,MAAM,WAAW,YAAY;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,KAAK,CAAC;QAAC,GAAG,EAAE,MAAM,CAAA;KAAE,GAAG;QAAE,IAAI,EAAE,YAAY,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG;QAAE,IAAI,EAAE,iBAAiB,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC7I,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,iBAAiB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,KAAK,SAAS,GAAG,SAAS,CAAC,OAAO,cAAc,EAAE,YAAY,CAAC,CAAC;AAChE,KAAK,cAAc,GAAG,SAAS,CAAC,OAAO,mBAAmB,EAAE,iBAAiB,CAAC,CAAC;AAC/E,MAAM,MAAM,eAAe,GAAG,SAAS,GAAG,cAAc,CAAC;AAEzD,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,mBAAmB,GAAG,eAAe,EAAE,CAGnF"}