@pi-oxide/pi-host-web 0.3.0 → 0.3.1

package/package.json

@@ -5,11 +5,11 @@
     "Irving Ou <irving@pi-oxide.dev>"
   ],
   "description": "WASM host for pi-core. Browser FileSystem Access API, fetch(), JS event loop.",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/pi-oxide/pi-oxide.git"
+    "url": "https://github.com/pi-oxide/pi-oxide"
   },
   "files": [
     "pi_host_web_bg.wasm",

package/pi_host_web.d.ts

@@ -389,8 +389,12 @@ export type ToolOutputStream = "stdout" | "stderr" | "status";
 export type WaitMode = "steering" | "follow_up" | "any";
 
 
+export function abort(handle: number): StepResult;
+
 export function appendSessionEntry(handle: number, entry: SessionEntry): EmptyResult;
 
+export function continueTurn(handle: number): StepResult;
+
 export function createAgent(options: AgentOptions): CreateAgentResult;
 
 export function destroyAgent(handle: number): EmptyResult;
@@ -439,7 +443,9 @@ export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembl
 
 export interface InitOutput {
     readonly memory: WebAssembly.Memory;
+    readonly abort: (a: number) => any;
     readonly appendSessionEntry: (a: number, b: any) => any;
+    readonly continueTurn: (a: number) => any;
     readonly createAgent: (a: any) => any;
     readonly destroyAgent: (a: number) => any;
     readonly estimateTokens: (a: any) => any;

package/pi_host_web.js

@@ -1,5 +1,14 @@
 /* @ts-self-types="./pi_host_web.d.ts" */
 
+/**
+ * @param {number} handle
+ * @returns {StepResult}
+ */
+export function abort(handle) {
+    const ret = wasm.abort(handle);
+    return ret;
+}
+
 /**
  * @param {number} handle
  * @param {SessionEntry} entry
@@ -10,6 +19,15 @@ export function appendSessionEntry(handle, entry) {
     return ret;
 }
 
+/**
+ * @param {number} handle
+ * @returns {StepResult}
+ */
+export function continueTurn(handle) {
+    const ret = wasm.continueTurn(handle);
+    return ret;
+}
+
 /**
  * @param {AgentOptions} options
  * @returns {CreateAgentResult}

package/sdk/index.d.ts

@@ -8,6 +8,8 @@ export * from "../pi_host_web.js";
 
 export declare function ensureInit(): Promise<void>;
 
+export declare function continueTurn(handle: number): StepResult;
+
 export declare function toolResult(
   text: string,
   opts?: { terminate?: boolean }
@@ -24,7 +26,7 @@ export interface LlmStream {
 }
 
 export interface LlmProvider {
-  call(context: LlmContext): Promise<LlmStream> | LlmStream;
+  call(context: LlmContext, signal?: AbortSignal): Promise<LlmStream> | LlmStream;
 }
 
 export type ToolMap = Record<
@@ -36,11 +38,13 @@ export interface AgentRunConfig {
   llm: LlmProvider;
   tools: ToolMap;
   onEvent?: (event: AgentEvent) => void;
+  signal?: AbortSignal;
 }
 
 export declare class Agent {
   static create(options: AgentOptions): Promise<Agent>;
   run(promptText: string, config: AgentRunConfig): Promise<AgentAction>;
+  stop(): void;
   reset(): void;
   state(): AgentState;
   getSessionState(): SessionState;

package/sdk/index.js

@@ -9,7 +9,9 @@
  */
 
 import {
+  abort,
   createAgent,
+  continueTurn,
   destroyAgent,
   feedLlmChunk,
   followUp,
@@ -109,28 +111,41 @@ export class Agent {
    * @param {LlmProvider} config.llm
    * @param {Record<string, (call: ToolCall) => Promise<ToolResult> | ToolResult>} config.tools
    * @param {(event: AgentEvent) => void} [config.onEvent]
+   * @param {AbortSignal} [config.signal] — abort to stop mid-stream or mid-tool
    * @returns {Promise<AgentAction>} terminal action (finished or wait_for_input)
    */
   async run(promptText, config) {
+    const signal = config.signal;
+    const checkAbort = () => {
+      if (signal?.aborted) {
+        this.stop();
+        throw new HostError("user_aborted", "Turn stopped by user");
+      }
+    };
+
     let step = unwrap(prompt(this.#handle, { text: promptText }));
     for (const event of step.events) {
       config.onEvent?.(event);
     }
 
     while (true) {
+      checkAbort();
       let actions = step.actions ?? [];
       if (actions.length === 0) {
         return { type: "finished", messages: [] };
       }
 
       for (const action of actions) {
+        checkAbort();
         switch (action.type) {
           case "stream_llm": {
-            const stream = await config.llm.call(action.context);
+            const stream = await config.llm.call(action.context, signal);
             for await (const chunk of stream.chunks) {
+              checkAbort();
               const ev = unwrap(feedLlmChunk(this.#handle, chunk));
               for (const e of ev.events) config.onEvent?.(e);
             }
+            checkAbort();
             const result = await stream.result;
             step = unwrap(onLlmDone(this.#handle, result));
             for (const e of step.events) config.onEvent?.(e);
@@ -139,6 +154,7 @@ export class Agent {
 
           case "execute_tools": {
             for (const call of action.calls) {
+              checkAbort();
               const started = unwrap(onToolStarted(this.#handle, call.id));
               for (const e of started.events) config.onEvent?.(e);
 
@@ -152,6 +168,10 @@ export class Agent {
               step = unwrap(onToolDone(this.#handle, call.id, result));
               for (const e of step.events) config.onEvent?.(e);
             }
+            if ((step.actions ?? []).length === 0) {
+              step = unwrap(continueTurn(this.#handle));
+              for (const e of step.events) config.onEvent?.(e);
+            }
             break;
           }
 
@@ -178,6 +198,15 @@ export class Agent {
     }
   }
 
+  /** Abort a running turn mid-stream or mid-tool. */
+  stop() {
+    try {
+      unwrap(abort(this.#handle));
+    } catch (e) {
+      if (e.code !== "wrong_phase") throw e;
+    }
+  }
+
   /** Reset agent state (clear messages, return to idle). */
   reset() {
     unwrap(reset(this.#handle));