@qoniai/qoni 0.4.0 → 0.4.2

package/dist/index.js

@@ -768,11 +768,11 @@ var RunHandle = class {
   }
   /** Drive the run to a terminal state and return the settled result. */
   async wait(opts = {}) {
-    const { signal, timedOut } = withTimeout(opts.signal, opts.timeoutMs);
+    const { signal, timedOut, dispose } = withTimeout(opts.signal, opts.timeoutMs);
     let lastEventId;
     let eventIdAtLastClose;
     let sawCleanClose = false;
-    let terminalPayload;
+    let terminalDetail;
     let terminalDoneReason;
     let screenshotIndex = 0;
     try {
@@ -780,24 +780,28 @@ var RunHandle = class {
         for await (const event of this.events({
           signal,
           lastEventId,
-          onWireEvent: opts.onWireEvent,
+          onWireEvent: (wire) => waitWithSignal(() => {
+            if (wire.id !== void 0) lastEventId = wire.id;
+            return opts.onWireEvent?.(wire);
+          }, signal),
           onReconnect: opts.onReconnect
         })) {
-          if (event.raw.id) lastEventId = event.raw.id;
-          await opts.onEvent?.(event);
+          await waitWithSignal(() => opts.onEvent?.(event), signal);
           if (event.type === "screenshot") {
-            await opts.onScreenshot?.(event.image, screenshotIndex++);
+            await waitWithSignal(() => opts.onScreenshot?.(event.image, screenshotIndex++), signal);
           } else if (event.type === "interaction") {
-            await opts.onInteraction?.(this.interactionHandle(event.data), event);
+            await waitWithSignal(() => opts.onInteraction?.(this.interactionHandle(event.data), event), signal);
           }
           if (event.isTerminal) {
-            terminalPayload = isJsonObject(event.raw.data) ? event.raw.data : void 0;
             if (event.type === "done") terminalDoneReason = event.data.terminalReason || void 0;
             break settled;
           }
         }
-        const detail2 = await this.ops.getDetail();
-        if (isTerminalRunStatus(detail2.status)) break;
+        const detail2 = await waitWithSignal(() => this.ops.getDetail(signal), signal);
+        if (isTerminalRunStatus(normalizeRunStatus(this.id, detail2).status)) {
+          terminalDetail = detail2;
+          break;
+        }
         if (sawCleanClose && lastEventId === eventIdAtLastClose) {
           throw new QoniError(
             `run event stream ended before a terminal event while the run is still "${String(detail2.status)}" \u2014 re-attach or poll status()`,
@@ -807,6 +811,11 @@ var RunHandle = class {
         sawCleanClose = true;
         eventIdAtLastClose = lastEventId;
       }
+      const detail = terminalDetail ?? await waitWithSignal(() => this.ops.getDetail(signal), signal);
+      this.#adoptSessionRef(detail);
+      const result = settleRunResult(this.id, detail, [], terminalDoneReason);
+      result.artifacts = this.ops.listArtifacts ? await waitWithSignal(() => this.ops.listArtifacts(signal), signal) : [];
+      return result;
     } catch (err) {
       if (timedOut() && !opts.signal?.aborted) {
         throw new QoniTimeoutError(
@@ -815,17 +824,9 @@ var RunHandle = class {
         );
       }
       throw err;
+    } finally {
+      dispose();
     }
-    let detail;
-    try {
-      detail = await this.ops.getDetail();
-    } catch (err) {
-      if (!terminalPayload) throw err;
-      detail = terminalPayload;
-    }
-    this.#adoptSessionRef(detail);
-    const artifacts = this.ops.listArtifacts ? await this.ops.listArtifacts() : [];
-    return settleRunResult(this.id, detail, artifacts, terminalDoneReason);
   }
   /**
    * Wrap an {@link Interaction} (from `event.data` of an `interaction` event)
@@ -878,15 +879,12 @@ function normalizeRunStatus(id, detail) {
 function settleRunResult(runId, detail, artifacts, fallbackTerminalReason) {
   const terminalReason = (typeof detail.terminal_reason === "string" ? detail.terminal_reason : void 0) ?? fallbackTerminalReason;
   const isTaskSuccessful = typeof detail.is_task_successful === "boolean" ? detail.is_task_successful : void 0;
-  let status;
-  if (isTerminalRunStatus(detail.status)) {
-    status = detail.status;
-  } else if (terminalReason === "canceled") {
-    status = "canceled";
-  } else if (terminalReason === "failed" || terminalReason === "expired" || isTaskSuccessful === false) {
-    status = "failed";
-  } else {
-    status = "succeeded";
+  const status = normalizeRunStatus(runId, detail).status;
+  if (!isTerminalRunStatus(status)) {
+    throw new QoniError(
+      "run detail did not confirm a terminal state \u2014 re-attach or poll status()",
+      { code: "upstream.failed", retryable: true, body: detail }
+    );
   }
   return {
     runId,
@@ -913,19 +911,40 @@ function isVideoFrame(wire) {
   return type === "browser_video_frame" || type === "run.browser_video_frame";
 }
 function withTimeout(signal, timeoutMs) {
-  if (!timeoutMs) return { signal, timedOut: () => false };
+  if (!timeoutMs) return { signal, timedOut: () => false, dispose: () => {
+  } };
   const controller = new AbortController();
   let fired = false;
   const timer = setTimeout(() => {
     fired = true;
     controller.abort();
   }, timeoutMs);
-  controller.signal.addEventListener("abort", () => clearTimeout(timer), { once: true });
+  const onAbort = () => controller.abort(signal?.reason);
+  const dispose = () => {
+    clearTimeout(timer);
+    signal?.removeEventListener("abort", onAbort);
+  };
+  controller.signal.addEventListener("abort", dispose, { once: true });
   if (signal) {
-    if (signal.aborted) controller.abort();
-    else signal.addEventListener("abort", () => controller.abort(), { once: true });
+    if (signal.aborted) onAbort();
+    else signal.addEventListener("abort", onAbort, { once: true });
   }
-  return { signal: controller.signal, timedOut: () => fired };
+  return { signal: controller.signal, timedOut: () => fired, dispose };
+}
+function waitWithSignal(operation, signal) {
+  if (!signal) return Promise.resolve().then(operation);
+  return new Promise((resolve, reject) => {
+    const onAbort = () => reject(signal.reason ?? new DOMException("Aborted", "AbortError"));
+    if (signal.aborted) {
+      onAbort();
+      return;
+    }
+    signal.addEventListener("abort", onAbort, { once: true });
+    Promise.resolve().then(() => {
+      signal.throwIfAborted();
+      return operation();
+    }).then(resolve, reject).finally(() => signal.removeEventListener("abort", onAbort));
+  });
 }
 
 // src/deep-research.ts
@@ -940,6 +959,7 @@ function createDeepResearchNamespace(transport) {
       `/deep_research/runs/${encodeURIComponent(input.runId)}`,
       input.token,
       {
+        signal: input.signal,
         requiredScopes: ["webagent.deep_research:read"]
       }
     ),
@@ -985,6 +1005,7 @@ function createDeepResearchNamespace(transport) {
       `/deep_research/runs/${encodeURIComponent(input.runId)}/artifacts`,
       input.token,
       {
+        signal: input.signal,
         requiredScopes: ["webagent.deep_research:read"]
       }
     ),
@@ -999,7 +1020,7 @@ function createDeepResearchNamespace(transport) {
   };
   function buildOps(token, runId) {
     return {
-      getDetail: async () => asRecord((await api.get({ token, runId })).data),
+      getDetail: async (signal) => asRecord((await api.get({ token, runId, signal })).data),
       streamEvents: (opts) => api.events({
         token,
         runId,
@@ -1019,8 +1040,8 @@ function createDeepResearchNamespace(transport) {
         );
       },
       cancel: async () => asRecord((await api.cancel({ token, runId })).data),
-      listArtifacts: async () => {
-        const res = await api.listArtifacts({ token, runId });
+      listArtifacts: async (signal) => {
+        const res = await api.listArtifacts({ token, runId, signal });
         const items = Array.isArray(res.data?.items) ? res.data.items : [];
         return items.filter(isJsonObject).map((item) => toArtifact(item, token, runId));
       }
@@ -1140,6 +1161,7 @@ function createDoAnythingNamespace(transport) {
       `/do_anything/runs/${encodeURIComponent(input.runId)}`,
       input.token,
       {
+        signal: input.signal,
         requiredScopes: ["webagent.do_anything:read"]
       }
     ),
@@ -1197,7 +1219,7 @@ function createDoAnythingNamespace(transport) {
   };
   function buildOps(token, runId, capture) {
     return {
-      getDetail: async () => asRecord2((await api.getRun({ token, runId })).data),
+      getDetail: async (signal) => asRecord2((await api.getRun({ token, runId, signal })).data),
       streamEvents: (opts) => api.events({
         token,
         runId,
@@ -1791,6 +1813,7 @@ function createWebSearchNamespace(transport) {
       `/web_search/runs/${encodeURIComponent(input.runId)}`,
       input.token,
       {
+        signal: input.signal,
         requiredScopes: ["webagent.web_search:read"]
       }
     ),
@@ -1816,7 +1839,7 @@ function createWebSearchNamespace(transport) {
   };
   function buildOps(token, runId) {
     return {
-      getDetail: async () => asRecord4((await api.get({ token, runId })).data),
+      getDetail: async (signal) => asRecord4((await api.get({ token, runId, signal })).data),
       streamEvents: (opts) => api.events({
         token,
         runId,
@@ -1909,8 +1932,8 @@ function renameKeys5(value, mapping) {
 }
 
 // src/client.ts
-var DEFAULT_QONI_HOST = "https://anima.eazo.ai/dashboard";
-var QoniAgentKit = class {
+var DEFAULT_QONI_HOST = "https://dashboard.qoni.ai";
+var Qoni = class {
   constructor(options) {
     this.options = options;
     const accessKey = options.accessKey ?? options.accessKeyId;
@@ -2089,7 +2112,9 @@ var QoniAgentKit = class {
     }
   }
   async webAgentJson(method, path, token, payload = {}) {
+    payload.signal?.throwIfAborted();
     const runtimeToken = await this.productTokenFor("webagent", token, payload.requiredScopes);
+    payload.signal?.throwIfAborted();
     const baseUrl = await this.baseUrlFor("webagent");
     return this.jsonRequest(
       baseUrl,
@@ -2100,9 +2125,11 @@ var QoniAgentKit = class {
   // Binary GET against the WebAgent API — artifact content downloads. JSON
   // error payloads still map to typed QoniErrors.
   async webAgentBytes(path, token, payload = {}) {
+    payload.signal?.throwIfAborted();
     const runtimeToken = await this.productTokenFor("webagent", token, payload.requiredScopes);
+    payload.signal?.throwIfAborted();
     const baseUrl = await this.baseUrlFor("webagent");
-    const response = await this.fetchWithTimeout(
+    return this.fetchWithTimeout(
       urlWithBasePath(baseUrl, this.webAgentPath(runtimeToken, path), payload.query),
       {
         method: "GET",
@@ -2111,20 +2138,22 @@ var QoniAgentKit = class {
           authorization: `Bearer ${runtimeToken}`,
           ...payload.headers || {}
         }
-      }
+      },
+      async (response, signal) => {
+        if (!response.ok) {
+          throw errorFromPayload(
+            response.status,
+            await readJson(response, signal),
+            `Qoni request failed with HTTP ${response.status}`
+          );
+        }
+        return {
+          bytes: await readBody(response, signal),
+          mime: response.headers.get("content-type") || "application/octet-stream"
+        };
+      },
+      payload.signal
     );
-    if (!response.ok) {
-      throw errorFromPayload(
-        response.status,
-        await readJson(response),
-        `Qoni request failed with HTTP ${response.status}`
-      );
-    }
-    const buffer = await response.arrayBuffer();
-    return {
-      bytes: new Uint8Array(buffer),
-      mime: response.headers.get("content-type") || "application/octet-stream"
-    };
   }
   async *webAgentSSE(path, token, payload = {}) {
     const runtimeToken = await this.productTokenFor("webagent", token, payload.requiredScopes);
@@ -2164,15 +2193,16 @@ var QoniAgentKit = class {
     });
   }
   async jsonRequest(baseUrl, pathname, options) {
-    const response = await this.fetchWithTimeout(
+    const { response, payload } = await this.fetchWithTimeout(
       urlWithBasePath(baseUrl, pathname, options.query),
       {
         method: options.method,
         headers: requestHeaders(options),
         body: requestBody(options.body)
-      }
+      },
+      async (response2, signal) => ({ response: response2, payload: await readJson(response2, signal) }),
+      options.signal
     );
-    const payload = await readJson(response);
     if (!response.ok) {
       throw errorFromPayload(
         response.status,
@@ -2222,45 +2252,55 @@ var QoniAgentKit = class {
     }
   }
   async *sseConnectOnce(baseUrl, pathname, options) {
-    const response = await this.fetchWithTimeout(
-      urlWithBasePath(baseUrl, pathname, options.query),
-      {
-        method: "GET",
-        headers: {
-          accept: "text/event-stream",
-          authorization: `Bearer ${options.token}`,
-          ...options.headers || {},
-          ...options.lastEventId ? { "last-event-id": options.lastEventId } : {}
-        }
-      },
-      options.signal,
-      { streamBody: true }
-    );
-    if (!response.ok) {
-      throw errorFromPayload(
-        response.status,
-        await readJson(response),
-        `Qoni stream failed with HTTP ${response.status}`
-      );
-    }
-    if (!response.body) {
-      throw errorFromPayload(502, void 0, "Qoni stream response has no body");
-    }
-    const reader = response.body.getReader();
+    let cleanup;
+    let reader;
+    const onAbort = () => {
+      void reader?.cancel(options.signal?.reason).catch(() => void 0);
+    };
     const decoder = new TextDecoder();
     let buffer = "";
     try {
+      const response = await this.fetchWithTimeout(
+        urlWithBasePath(baseUrl, pathname, options.query),
+        {
+          method: "GET",
+          headers: {
+            accept: "text/event-stream",
+            authorization: `Bearer ${options.token}`,
+            ...options.headers || {},
+            ...options.lastEventId ? { "last-event-id": options.lastEventId } : {}
+          }
+        },
+        async (response2, signal) => {
+          if (!response2.ok) {
+            throw errorFromPayload(
+              response2.status,
+              await readJson(response2, signal),
+              `Qoni stream failed with HTTP ${response2.status}`
+            );
+          }
+          return response2;
+        },
+        options.signal,
+        { streamCleanup: (dispose) => {
+          cleanup = dispose;
+        } }
+      );
+      if (!response.body) {
+        throw errorFromPayload(502, void 0, "Qoni stream response has no body");
+      }
+      reader = response.body.getReader();
+      options.signal?.addEventListener("abort", onAbort, { once: true });
       while (true) {
-        if (options.signal?.aborted) {
-          await reader.cancel().catch(() => void 0);
-          throw options.signal.reason || new DOMException("Aborted", "AbortError");
-        }
+        options.signal?.throwIfAborted();
         const { done, value } = await reader.read();
+        options.signal?.throwIfAborted();
         if (done) break;
         buffer += decoder.decode(value, { stream: true });
         const chunks = buffer.split(/\r?\n\r?\n/);
         buffer = chunks.pop() || "";
         for (const chunk of chunks) {
+          options.signal?.throwIfAborted();
           const event = parseSSEEvent(chunk);
           if (event) yield event;
         }
@@ -2268,10 +2308,16 @@ var QoniAgentKit = class {
       const finalEvent = parseSSEEvent(buffer);
       if (finalEvent) yield finalEvent;
     } finally {
-      reader.releaseLock();
+      cleanup?.();
+      options.signal?.removeEventListener("abort", onAbort);
+      void reader?.cancel().catch(() => void 0);
+      try {
+        reader?.releaseLock();
+      } catch {
+      }
     }
   }
-  async fetchWithTimeout(input, init, signal, opts) {
+  async fetchWithTimeout(input, init, consume, signal, opts) {
     if (signal?.aborted) throw signal.reason || new DOMException("Aborted", "AbortError");
     const controller = new AbortController();
     const resolved = resolveLocalGenauthRequest(input);
@@ -2286,24 +2332,46 @@ var QoniAgentKit = class {
       didTimeout = true;
       controller.abort();
     }, this.timeoutMs);
+    let response;
     try {
-      return await this.fetchImpl(resolved.url, { ...requestInit, signal: controller.signal });
+      response = await new Promise((resolve, reject) => {
+        const onAbort = () => reject(controller.signal.reason);
+        controller.signal.addEventListener("abort", onAbort, { once: true });
+        Promise.resolve().then(() => {
+          controller.signal.throwIfAborted();
+          return this.fetchImpl(resolved.url, { ...requestInit, signal: controller.signal });
+        }).then((response2) => {
+          if (controller.signal.aborted) {
+            void response2.body?.cancel(controller.signal.reason).catch(() => void 0);
+            reject(controller.signal.reason);
+          } else {
+            resolve(response2);
+          }
+        }, reject).finally(() => controller.signal.removeEventListener("abort", onAbort));
+      });
+      const result = await consume(response, controller.signal);
+      controller.signal.throwIfAborted();
+      return result;
     } catch (error) {
-      if (isAbortError(error) && didTimeout || isConnectTimeoutError(error)) {
+      void response?.body?.cancel().catch(() => void 0);
+      if (signal?.aborted) throw signal.reason;
+      if (didTimeout || isConnectTimeoutError(error)) {
         throw timeoutError(error);
       }
       if (isFetchNetworkError(error)) throw networkError(error);
       throw error;
     } finally {
       clearTimeout(timeout);
-      if (!opts?.streamBody) signal?.removeEventListener("abort", abortFromSignal);
+      const cleanup = () => signal?.removeEventListener("abort", abortFromSignal);
+      if (opts?.streamCleanup) opts.streamCleanup(cleanup);
+      else cleanup();
     }
   }
   async baseUrlFor(service) {
     const explicit = this.explicitBaseUrlFor(service);
     if (explicit) return explicit;
-    const runtime = await this.runtimeConfig();
     if (service === "qoni") return this.host;
+    const runtime = await this.runtimeConfig();
     const hostOrigin = originBaseUrl(this.host);
     if (service === "genauth") return runtime.genauthBaseUrl ?? hostOrigin;
     if (service === "gumem") return runtime.gumemBaseUrl ?? hostOrigin;
@@ -2352,7 +2420,7 @@ var QoniAgentKit = class {
       pathname,
       {}
     );
-    const response = await this.fetchWithTimeout(
+    const { response, payload } = await this.fetchWithTimeout(
       urlWithBasePath(this.host, pathname),
       {
         method: "GET",
@@ -2360,9 +2428,9 @@ var QoniAgentKit = class {
           accept: "application/json",
           ...headers
         }
-      }
+      },
+      async (response2, signal) => ({ response: response2, payload: await readJson(response2, signal) })
     );
-    const payload = await readJson(response);
     if (!response.ok) {
       throw errorFromPayload(
         response.status,
@@ -2456,18 +2524,52 @@ function requestBody(body) {
 function isBodyInit(value) {
   return typeof value === "string" || value instanceof ArrayBuffer || value instanceof Blob || value instanceof FormData || value instanceof URLSearchParams;
 }
-async function readJson(response) {
+async function readBody(response, signal) {
+  if (!response.body) {
+    signal.throwIfAborted();
+    return new Uint8Array();
+  }
+  const reader = response.body.getReader();
+  const onAbort = () => {
+    void reader.cancel(signal.reason).catch(() => void 0);
+  };
+  signal.addEventListener("abort", onAbort, { once: true });
+  try {
+    const chunks = [];
+    let size = 0;
+    for (; ; ) {
+      signal.throwIfAborted();
+      const { done, value } = await reader.read();
+      signal.throwIfAborted();
+      if (done) break;
+      chunks.push(value);
+      size += value.byteLength;
+    }
+    const bytes = new Uint8Array(size);
+    let offset = 0;
+    for (const chunk of chunks) {
+      bytes.set(chunk, offset);
+      offset += chunk.byteLength;
+    }
+    return bytes;
+  } finally {
+    signal.removeEventListener("abort", onAbort);
+    void reader.cancel().catch(() => void 0);
+    reader.releaseLock();
+  }
+}
+async function readJson(response, signal) {
   const contentType = response.headers.get("content-type")?.toLowerCase() || "";
-  if (contentType.includes("ndjson") || contentType.includes("jsonl")) {
-    const body = await response.text();
-    if (!body.trim()) return [];
-    try {
+  const body = new TextDecoder().decode(await readBody(response, signal));
+  try {
+    if (contentType.includes("ndjson") || contentType.includes("jsonl")) {
+      if (!body.trim()) return [];
       return body.split(/\r?\n/).filter((line) => line.trim().length > 0).map((line) => JSON.parse(line));
-    } catch {
-      return void 0;
     }
+    return JSON.parse(body);
+  } catch {
+    return void 0;
   }
-  return response.json().catch(() => void 0);
 }
 function normalizeResponse(payload, _service) {
   if (isJsonObject(payload) && ("data" in payload || "meta" in payload)) {
@@ -2710,6 +2812,7 @@ function isRetryableStreamError(err) {
   if (err && typeof err === "object") {
     const e = err;
     if (e.name === "AbortError") return false;
+    if (e.status === 408) return true;
     if (typeof e.retryable === "boolean") return e.retryable;
     if (typeof e.status === "number") {
       return e.status >= 500 || e.status === 408 || e.status === 429;
@@ -2816,9 +2919,6 @@ function normalizeDelegateAgentTokenResponse(response) {
     }
   };
 }
-function isAbortError(error) {
-  return error instanceof DOMException && error.name === "AbortError";
-}
 function isConnectTimeoutError(error) {
   const cause = errorCause(error);
   return errorCode(cause) === "UND_ERR_CONNECT_TIMEOUT" || errorName(cause) === "ConnectTimeoutError";
@@ -2843,8 +2943,7 @@ export {
   InteractionStatuses,
   InteractionTypes,
   MonitorHandle,
-  QoniAgentKit as Qoni,
-  QoniAgentKit,
+  Qoni,
   QoniAuthError,
   QoniDelegationRequiredError,
   QoniError,