@harnessio/ai-chat-core 0.0.23 → 0.0.24

package/dist/index.js

@@ -1,7 +1,7 @@
 var H = Object.defineProperty;
 var J = (n, s, t) => s in n ? H(n, s, { enumerable: !0, configurable: !0, writable: !0, value: t }) : n[s] = t;
 var u = (n, s, t) => J(n, typeof s != "symbol" ? s + "" : s, t);
-import q, { createContext as U, useContext as B, useState as y, useEffect as m, useReducer as V, useCallback as C, useMemo as z } from "react";
+import q, { createContext as B, useContext as U, useState as y, useEffect as b, useReducer as V, useCallback as C, useMemo as z } from "react";
 class G {
   constructor() {
     u(this, "plugins", /* @__PURE__ */ new Map());
@@ -414,10 +414,10 @@ let tt = 0, et = 0;
 function v() {
   return `msg-${Date.now()}-${++tt}`;
 }
-function k() {
+function F() {
   return `thread-${Date.now()}-${++et}`;
 }
-class R extends _ {
+class T extends _ {
   constructor(t) {
     super();
     u(this, "_state");
@@ -473,7 +473,7 @@ class st extends _ {
     this._isSubmitting = t, this.notifySubscribers();
   }
 }
-class T extends _ {
+class R extends _ {
   constructor(t) {
     super();
     u(this, "composer");
@@ -553,6 +553,23 @@ class T extends _ {
   async sendSystemEvent(t) {
     await this._core.waitForIdle(), await this._core.startSystemEventRun(t);
   }
+  /**
+   * Registers a generic background task that keeps the thread reported as
+   * running for as long as it is active, even when no stream run is in
+   * flight. Use this for UI-driven async work (e.g. a card polling a status
+   * API) so the composer shows a running/stop affordance. The work is
+   * cancelled when the user presses stop (cancelRun), which invokes the
+   * provided onCancel callback. Returns an opaque id for stopBackgroundTask.
+   */
+  startBackgroundTask(t) {
+    return this._core.startBackgroundTask(t);
+  }
+  /**
+   * Removes a previously registered background task by its id.
+   */
+  stopBackgroundTask(t) {
+    this._core.stopBackgroundTask(t);
+  }
   cancelRun() {
     this._core.cancelRun();
   }
@@ -591,6 +608,13 @@ class M extends _ {
     // dispatch against an in-progress run instead of having their error
     // swallowed by the caller's try/catch.
     u(this, "_idleWaiters", []);
+    // Generic background tasks that hold the thread in a "running" state
+    // without an active stream run. Any UI-driven async work (e.g. a card
+    // that polls a status API) can register one to keep the composer
+    // showing a running/stop affordance, and is cancelled alongside the
+    // run when the user presses stop. Keyed by an opaque task id; the
+    // optional onCancel is invoked when the task is cancelled via cancelRun.
+    u(this, "_backgroundTasks", /* @__PURE__ */ new Map());
     // Track current part being accumulated
     u(this, "_currentPart", null);
     this.config = t, t.initialMessages && (this._messages = [...t.initialMessages]);
@@ -599,7 +623,7 @@ class M extends _ {
     return this._messages;
   }
   get isRunning() {
-    return this._isRunning;
+    return this._isRunning || this._backgroundTasks.size > 0;
   }
   get isDisabled() {
     return this._isDisabled;
@@ -665,6 +689,40 @@ class M extends _ {
       } catch {
       }
   }
+  /**
+   * Registers a generic background task that keeps the thread reported as
+   * running (isRunning === true) for as long as it is active, even when no
+   * stream run is in flight. Use this for UI-driven async work — e.g. a card
+   * that polls a status API — so the composer shows a running/stop
+   * affordance and the work is cancelled when the user presses stop.
+   *
+   * Returns an opaque id to pass to stopBackgroundTask. The optional onCancel
+   * callback is invoked if the task is cancelled via cancelRun (stop button).
+   */
+  startBackgroundTask(t) {
+    const e = `bg-${Date.now()}-${Math.random().toString(36).slice(2)}`;
+    return this._backgroundTasks.set(e, { onCancel: t == null ? void 0 : t.onCancel }), this.notifySubscribers(), e;
+  }
+  /**
+   * Removes a previously registered background task. Once the last task is
+   * removed (and no run is in flight) any idle waiters are drained so queued
+   * dispatches can proceed.
+   */
+  stopBackgroundTask(t) {
+    this._backgroundTasks.delete(t) && (this.isRunning || this.drainIdleWaiters(), this.notifySubscribers());
+  }
+  cancelBackgroundTasks() {
+    var e;
+    if (this._backgroundTasks.size === 0) return;
+    const t = [...this._backgroundTasks.values()];
+    this._backgroundTasks.clear();
+    for (const i of t)
+      try {
+        (e = i.onCancel) == null || e.call(i);
+      } catch {
+      }
+    this.isRunning || this.drainIdleWaiters(), this.notifySubscribers();
+  }
   async startSystemEventRun(t) {
     if (this._isRunning)
       throw new Error("A run is already in progress");
@@ -912,7 +970,7 @@ class M extends _ {
     this.updateMessages([]);
   }
   cancelRun() {
-    !this._isRunning || !this._abortController || this._abortController.abort();
+    this._isRunning && this._abortController && this._abortController.abort(), this.cancelBackgroundTasks();
   }
   reset(t = []) {
     this.updateMessages([...t]);
@@ -945,12 +1003,12 @@ class it extends _ {
     u(this, "_searchQuery");
     u(this, "_abortController", null);
     u(this, "main");
-    this.config = t, this._mainThreadId = k();
+    this.config = t, this._mainThreadId = F();
     const e = new M({
       streamAdapter: t.streamAdapter,
       capabilityExecutionManager: t.capabilityExecutionManager
     });
-    this.main = new T(e), this._threads.set(this._mainThreadId, this.main);
+    this.main = new R(e), this._threads.set(this._mainThreadId, this.main);
     const i = {
       id: this._mainThreadId,
       title: "New Chat",
@@ -960,7 +1018,7 @@ class it extends _ {
       updatedAt: Date.now()
     };
     this._threadStates.set(this._mainThreadId, i);
-    const r = new R({
+    const r = new T({
       state: i,
       onSwitchTo: this.switchToThread.bind(this),
       onRename: this.renameThread.bind(this),
@@ -1012,10 +1070,10 @@ class it extends _ {
     this.notifySubscribers();
   }
   async switchToNewThread() {
-    const t = k(), e = new M({
+    const t = F(), e = new M({
       streamAdapter: this.config.streamAdapter,
       capabilityExecutionManager: this.config.capabilityExecutionManager
-    }), i = new T(e);
+    }), i = new R(e);
     this._threads.set(t, i);
     const r = {
       id: t,
@@ -1026,7 +1084,7 @@ class it extends _ {
       updatedAt: Date.now()
     };
     this._threadStates.set(t, r);
-    const a = new R({
+    const a = new T({
       state: r,
       onSwitchTo: this.switchToThread.bind(this),
       onRename: this.renameThread.bind(this),
@@ -1062,9 +1120,9 @@ class it extends _ {
         const r = new M({
           streamAdapter: this.config.streamAdapter,
           capabilityExecutionManager: this.config.capabilityExecutionManager
-        }), a = new T(r);
+        }), a = new R(r);
         this._threads.set(i.id, a), this._threadStates.set(i.id, i);
-        const o = new R({
+        const o = new T({
           state: i,
           onSwitchTo: this.switchToThread.bind(this),
           onRename: this.renameThread.bind(this),
@@ -1238,8 +1296,8 @@ function rt() {
         s.call(o, d) && (c[d] = o[d]);
       if (n) {
         h = n(o);
-        for (var p = 0; p < h.length; p++)
-          t.call(o, h[p]) && (c[h[p]] = o[h[p]]);
+        for (var g = 0; g < h.length; g++)
+          t.call(o, h[g]) && (c[h[g]] = o[h[g]]);
       }
     }
     return c;
@@ -1264,11 +1322,11 @@ function nt() {
   }
   var e = n.__SECRET_INTERNALS_DO_NOT_USE_OR_YOU_WILL_BE_FIRED.ReactCurrentOwner, i = Object.prototype.hasOwnProperty, r = { key: !0, ref: !0, __self: !0, __source: !0 };
   function a(o, c, h) {
-    var l, d = {}, p = null, g = null;
-    h !== void 0 && (p = "" + h), c.key !== void 0 && (p = "" + c.key), c.ref !== void 0 && (g = c.ref);
+    var l, d = {}, g = null, p = null;
+    h !== void 0 && (g = "" + h), c.key !== void 0 && (g = "" + c.key), c.ref !== void 0 && (p = c.ref);
     for (l in c) i.call(c, l) && !r.hasOwnProperty(l) && (d[l] = c[l]);
     if (o && o.defaultProps) for (l in c = o.defaultProps, c) d[l] === void 0 && (d[l] = c[l]);
-    return { $$typeof: s, type: o, key: p, ref: g, props: d, _owner: e.current };
+    return { $$typeof: s, type: o, key: g, ref: p, props: d, _owner: e.current };
   }
   return x.jsx = a, x.jsxs = a, x;
 }
@@ -1277,12 +1335,12 @@ function at() {
   return O || (O = 1, E.exports = nt()), E.exports;
 }
 var w = at();
-const W = U(null);
+const W = B(null);
 function _t({ runtime: n, children: s }) {
   return /* @__PURE__ */ w.jsx(W.Provider, { value: n, children: s });
 }
 function ot() {
-  const n = B(W);
+  const n = U(W);
   if (!n)
     throw new Error("useAssistantRuntimeContext must be used within AssistantRuntimeProvider");
   return n;
@@ -1308,17 +1366,17 @@ function St() {
 }
 function xt() {
   const n = f(), [s, t] = y(n.threads.getState());
-  return m(() => n.threads.subscribe(() => {
+  return b(() => n.threads.subscribe(() => {
     t(n.threads.getState());
   }), [n]), s;
 }
 function ct() {
   const s = f().threads.getMainThread(), [, t] = y({});
-  return m(() => s.subscribe(() => {
+  return b(() => s.subscribe(() => {
     t({});
   }), [s]), s;
 }
-var P = { exports: {} }, F = {};
+var k = { exports: {} }, P = {};
 /**
  * @license React
  * use-sync-external-store-shim.production.js
@@ -1330,48 +1388,48 @@ var P = { exports: {} }, F = {};
  */
 var j;
 function ut() {
-  if (j) return F;
+  if (j) return P;
   j = 1;
   var n = q;
-  function s(d, p) {
-    return d === p && (d !== 0 || 1 / d === 1 / p) || d !== d && p !== p;
+  function s(d, g) {
+    return d === g && (d !== 0 || 1 / d === 1 / g) || d !== d && g !== g;
   }
   var t = typeof Object.is == "function" ? Object.is : s, e = n.useState, i = n.useEffect, r = n.useLayoutEffect, a = n.useDebugValue;
-  function o(d, p) {
-    var g = p(), S = e({ inst: { value: g, getSnapshot: p } }), b = S[0].inst, I = S[1];
+  function o(d, g) {
+    var p = g(), S = e({ inst: { value: p, getSnapshot: g } }), m = S[0].inst, I = S[1];
     return r(
       function() {
-        b.value = g, b.getSnapshot = p, c(b) && I({ inst: b });
+        m.value = p, m.getSnapshot = g, c(m) && I({ inst: m });
       },
-      [d, g, p]
+      [d, p, g]
     ), i(
       function() {
-        return c(b) && I({ inst: b }), d(function() {
-          c(b) && I({ inst: b });
+        return c(m) && I({ inst: m }), d(function() {
+          c(m) && I({ inst: m });
         });
       },
       [d]
-    ), a(g), g;
+    ), a(p), p;
   }
   function c(d) {
-    var p = d.getSnapshot;
+    var g = d.getSnapshot;
     d = d.value;
     try {
-      var g = p();
-      return !t(d, g);
+      var p = g();
+      return !t(d, p);
     } catch {
       return !0;
     }
   }
-  function h(d, p) {
-    return p();
+  function h(d, g) {
+    return g();
   }
   var l = typeof window > "u" || typeof window.document > "u" || typeof window.document.createElement > "u" ? h : o;
-  return F.useSyncExternalStore = n.useSyncExternalStore !== void 0 ? n.useSyncExternalStore : l, F;
+  return P.useSyncExternalStore = n.useSyncExternalStore !== void 0 ? n.useSyncExternalStore : l, P;
 }
 var L;
 function ht() {
-  return L || (L = 1, P.exports = ut()), P.exports;
+  return L || (L = 1, k.exports = ut()), k.exports;
 }
 var dt = ht();
 function wt() {
@@ -1406,7 +1464,7 @@ function Ct() {
     append: (a) => t((o) => o + a)
   };
 }
-function Q(n, s) {
+function $(n, s) {
   var r, a, o, c;
   const e = f().pluginRegistry.getBestRendererForType(n);
   if (!e)
@@ -1428,7 +1486,7 @@ function Q(n, s) {
 }
 function It() {
   const n = f(), [s, t] = y(n.contentFocus.state);
-  return m(() => n.contentFocus.subscribe(() => {
+  return b(() => n.contentFocus.subscribe(() => {
     t(n.contentFocus.state);
   }), [n]), {
     ...s,
@@ -1442,7 +1500,7 @@ function It() {
 }
 function lt(n) {
   const s = f();
-  m(() => {
+  b(() => {
     const t = n.priority ?? 0;
     return n.execute && s.capabilityRegistry.registerHandler(n.name, { execute: n.execute }, t), n.render && s.capabilityRegistry.registerRenderer(
       n.name,
@@ -1455,11 +1513,11 @@ function lt(n) {
     };
   }, [n.name, s]);
 }
-function pt(n) {
+function gt(n) {
   const s = f(), [t, e] = y(
     () => s.capabilityExecutionManager.getExecution(n)
   ), [, i] = V((r) => r + 1, 0);
-  return m(() => {
+  return b(() => {
     const r = s.capabilityExecutionManager.getExecution(n);
     return e(r), s.capabilityExecutionManager.subscribe((o) => {
       if (o === n) {
@@ -1469,13 +1527,13 @@ function pt(n) {
     });
   }, [n, s]), t;
 }
-const $ = U(void 0);
+const Q = B(void 0);
 function vt({ children: n }) {
   const [s, t] = y({}), e = C((c, h) => {
     t((l) => {
       if (h === null) {
-        const { [c]: d, ...p } = l;
-        return p;
+        const { [c]: d, ...g } = l;
+        return g;
       }
       return {
         ...l,
@@ -1490,9 +1548,9 @@ function vt({ children: n }) {
   }, []), r = C(() => Object.values(s).reduce(
     (c, h) => {
       const l = Object.entries(h.data).reduce(
-        (d, [p, g]) => ({
+        (d, [g, p]) => ({
           ...d,
-          [p]: typeof g == "string" ? g : JSON.stringify(g)
+          [g]: typeof p == "string" ? p : JSON.stringify(p)
         }),
         {}
       );
@@ -1511,29 +1569,29 @@ function vt({ children: n }) {
     }),
     [s, e, i, r, a]
   );
-  return /* @__PURE__ */ w.jsx($.Provider, { value: o, children: n });
+  return /* @__PURE__ */ w.jsx(Q.Provider, { value: o, children: n });
 }
-function gt() {
-  const n = B($);
+function pt() {
+  const n = U(Q);
   if (!n)
     throw new Error("useChatContext must be used within a ChatContextProvider");
   return n;
 }
-function Rt(n) {
-  const { setContext: s, removeContext: t } = gt(), { id: e = "currentPage", ...i } = n;
-  m(() => (s(e, i), () => {
+function Tt(n) {
+  const { setContext: s, removeContext: t } = pt(), { id: e = "currentPage", ...i } = n;
+  b(() => (s(e, i), () => {
     t(e);
   }), [e, i.displayName, JSON.stringify(i.data), i.priority, i.icon]);
 }
-function Tt(n, s = "merge") {
+function Rt(n, s = "merge") {
   const t = f();
-  m(() => (t.quickActionRegistry.activateScope(n, s), () => {
+  b(() => (t.quickActionRegistry.activateScope(n, s), () => {
     t.quickActionRegistry.deactivateScope(n);
   }), [n, s, t]);
 }
 function Mt(n, s) {
   const t = f();
-  m(() => {
+  b(() => {
     const e = s || t.quickActionRegistry.getActiveScope();
     return t.quickActionRegistry.register(n, e), () => {
       t.quickActionRegistry.unregister(n.id, e);
@@ -1552,13 +1610,13 @@ function Mt(n, s) {
 }
 function Et(n) {
   const s = f(), [t, e] = y([]);
-  return m(() => (s.quickActionRegistry.getVisible(n).then(e), s.subscribe(() => {
+  return b(() => (s.quickActionRegistry.getVisible(n).then(e), s.subscribe(() => {
     s.quickActionRegistry.getVisible(n).then(e);
   })), [s, n == null ? void 0 : n.scope]), t;
 }
 function At(n, s) {
   const t = f();
-  m(() => {
+  b(() => {
     const e = s || t.quickActionRegistry.getActiveScope();
     return n.forEach((i) => {
       t.quickActionRegistry.register(i, e);
@@ -1570,7 +1628,7 @@ function At(n, s) {
   }, [n, s, t]);
 }
 function ft({ message: n, content: s }) {
-  const t = f(), e = s, i = pt(e.capabilityId), r = t.capabilityRegistry.getRenderer(e.capabilityName);
+  const t = f(), e = s, i = gt(e.capabilityId), r = t.capabilityRegistry.getRenderer(e.capabilityName);
   if (!r || !i)
     return null;
   const a = r.component;
@@ -1586,8 +1644,8 @@ function ft({ message: n, content: s }) {
     }
   );
 }
-function Pt({ message: n, content: s }) {
-  const { component: t } = Q(s.type);
+function kt({ message: n, content: s }) {
+  const { component: t } = $(s.type);
   if (s.type === "capability")
     return /* @__PURE__ */ w.jsx(ft, { message: n, content: s });
   if (!t)
@@ -1595,7 +1653,7 @@ function Pt({ message: n, content: s }) {
   const e = t;
   return /* @__PURE__ */ w.jsx(e, { content: s, message: n });
 }
-function Ft({
+function Pt({
   message: n,
   content: s,
   context: t,
@@ -1603,7 +1661,7 @@ function Ft({
   onNavigate: i,
   onSwitchContext: r
 }) {
-  const { auxiliaryComponent: a } = Q(s.type, t);
+  const { auxiliaryComponent: a } = $(s.type, t);
   if (!a)
     return null;
   const o = a;
@@ -1619,7 +1677,7 @@ function Ft({
     }
   );
 }
-function kt(n) {
+function Ft(n) {
   const s = () => (lt(n), null);
   return s.displayName = `Capability(${n.name})`, s;
 }
@@ -1662,15 +1720,15 @@ class Dt {
               continue;
             }
             if (d.startsWith("data:")) {
-              const p = d.substring(5).trim();
-              if (p === "eof" || d === "eof")
+              const g = d.substring(5).trim();
+              if (g === "eof" || d === "eof")
                 break;
               try {
-                const g = JSON.parse(p);
+                const p = JSON.parse(g);
                 if (a && this.shouldProcessEvent(a)) {
                   const S = this.convertEvent({
                     event: a,
-                    data: g
+                    data: p
                   });
                   S && (yield S);
                 }
@@ -1689,34 +1747,34 @@ class Dt {
 export {
   yt as AssistantRuntime,
   _t as AssistantRuntimeProvider,
-  Ft as AuxiliaryRenderer,
+  Pt as AuxiliaryRenderer,
   Dt as BaseSSEStreamAdapter,
   K as CapabilityExecutionManager,
   Y as CapabilityRegistry,
   ft as CapabilityRendererComp,
   vt as ChatContextProvider,
   st as ComposerRuntime,
-  Pt as ContentRenderer,
+  kt as ContentRenderer,
   G as PluginRegistry,
   X as QuickActionRegistry,
-  R as ThreadListItemRuntime,
+  T as ThreadListItemRuntime,
   it as ThreadListRuntime,
-  T as ThreadRuntime,
+  R as ThreadRuntime,
   M as ThreadRuntimeCore,
   v as generateMessageId,
-  kt as makeCapability,
+  Ft as makeCapability,
   f as useAssistantRuntime,
   lt as useCapability,
-  pt as useCapabilityExecution,
-  gt as useChatContext,
+  gt as useCapabilityExecution,
+  pt as useChatContext,
   Ct as useComposer,
   It as useContentFocus,
-  Q as useContentRenderer,
+  $ as useContentRenderer,
   ct as useCurrentThread,
   wt as useMessages,
-  Rt as usePageContext,
+  Tt as usePageContext,
   Mt as useQuickAction,
-  Tt as useQuickActionScope,
+  Rt as useQuickActionScope,
   Et as useQuickActions,
   At as useRegisterQuickActions,
   St as useThreadList,