@harnessio/ai-chat-core 0.0.23 → 0.0.25

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 P() {
   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 = P();
     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 = P(), 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),
@@ -1118,7 +1176,7 @@ class it extends _ {
     this.notifySubscribers();
   }
 }
-class yt extends _ {
+class rt extends _ {
   constructor(t) {
     super();
     u(this, "threads");
@@ -1193,15 +1251,15 @@ class yt extends _ {
       }
   }
 }
-var E = { exports: {} }, x = {};
+var A = { exports: {} }, x = {};
 /*
 object-assign
 (c) Sindre Sorhus
 @license MIT
 */
-var A, D;
-function rt() {
-  if (D) return A;
+var E, D;
+function nt() {
+  if (D) return E;
   D = 1;
   var n = Object.getOwnPropertySymbols, s = Object.prototype.hasOwnProperty, t = Object.prototype.propertyIsEnumerable;
   function e(r) {
@@ -1231,19 +1289,19 @@ function rt() {
       return !1;
     }
   }
-  return A = i() ? Object.assign : function(r, a) {
+  return E = i() ? Object.assign : function(r, a) {
     for (var o, c = e(r), h, l = 1; l < arguments.length; l++) {
       o = Object(arguments[l]);
       for (var d in o)
         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;
-  }, A;
+  }, E;
 }
 /** @license React v17.0.2
  * react-jsx-runtime.production.min.js
@@ -1254,9 +1312,9 @@ function rt() {
  * LICENSE file in the root directory of this source tree.
  */
 var N;
-function nt() {
+function at() {
   if (N) return x;
-  N = 1, rt();
+  N = 1, nt();
   var n = q, s = 60103;
   if (x.Fragment = 60107, typeof Symbol == "function" && Symbol.for) {
     var t = Symbol.for;
@@ -1264,33 +1322,31 @@ 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;
 }
 var O;
-function at() {
-  return O || (O = 1, E.exports = nt()), E.exports;
+function ot() {
+  return O || (O = 1, A.exports = at()), A.exports;
 }
-var w = at();
-const W = U(null);
-function _t({ runtime: n, children: s }) {
+var w = ot();
+const ct = { stream: async function* () {
+} }, ut = new rt({ streamAdapter: ct }), W = B(ut);
+function wt({ runtime: n, children: s }) {
   return /* @__PURE__ */ w.jsx(W.Provider, { value: n, children: s });
 }
-function ot() {
-  const n = B(W);
-  if (!n)
-    throw new Error("useAssistantRuntimeContext must be used within AssistantRuntimeProvider");
-  return n;
+function ht() {
+  return U(W);
 }
 function f() {
-  return ot();
+  return ht();
 }
-function St() {
+function Ct() {
   const n = f();
   return {
     switchToThread: (s) => n.threads.switchToThread(s),
@@ -1306,19 +1362,19 @@ function St() {
     }
   };
 }
-function xt() {
+function It() {
   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() {
+function dt() {
   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: {} }, F = {};
 /**
  * @license React
  * use-sync-external-store-shim.production.js
@@ -1329,54 +1385,54 @@ var P = { exports: {} }, F = {};
  * LICENSE file in the root directory of this source tree.
  */
 var j;
-function ut() {
+function lt() {
   if (j) return F;
   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;
 }
 var L;
-function ht() {
-  return L || (L = 1, P.exports = ut()), P.exports;
+function gt() {
+  return L || (L = 1, k.exports = lt()), k.exports;
 }
-var dt = ht();
-function wt() {
+var pt = gt();
+function vt() {
   const n = f();
-  return dt.useSyncExternalStore(
+  return pt.useSyncExternalStore(
     (s) => n.subscribe(() => {
       s();
     }),
@@ -1384,8 +1440,8 @@ function wt() {
     () => n.thread.messages
   );
 }
-function Ct() {
-  const n = ct(), [s, t] = y(""), [e, i] = y(!1);
+function Tt() {
+  const n = dt(), [s, t] = y(""), [e, i] = y(!1);
   return {
     text: s,
     setText: t,
@@ -1406,7 +1462,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)
@@ -1426,9 +1482,9 @@ function Q(n, s) {
     supportsFullscreen: ((c = e.capabilities) == null ? void 0 : c.supportsFullscreen) ?? !1
   };
 }
-function It() {
+function Rt() {
   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,
@@ -1440,9 +1496,9 @@ function It() {
     focusPrevious: () => n.contentFocus.focusPrevious(n.thread.messages)
   };
 }
-function lt(n) {
+function ft(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 +1511,11 @@ function lt(n) {
     };
   }, [n.name, s]);
 }
-function pt(n) {
+function bt(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 +1525,22 @@ function pt(n) {
     });
   }, [n, s]), t;
 }
-const $ = U(void 0);
-function vt({ children: n }) {
+const mt = {
+  contexts: {},
+  setContext: () => {
+  },
+  removeContext: () => {
+  },
+  getContextData: () => ({}),
+  clearContexts: () => {
+  }
+}, Q = B(mt);
+function Mt({ 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 +1555,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 +1576,26 @@ 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($);
-  if (!n)
-    throw new Error("useChatContext must be used within a ChatContextProvider");
-  return n;
+function yt() {
+  return U(Q);
 }
-function Rt(n) {
-  const { setContext: s, removeContext: t } = gt(), { id: e = "currentPage", ...i } = n;
-  m(() => (s(e, i), () => {
+function At(n) {
+  const { setContext: s, removeContext: t } = yt(), { 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 Et(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) {
+function kt(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);
@@ -1550,15 +1612,15 @@ function Mt(n, s) {
     t
   ]);
 }
-function Et(n) {
+function Ft(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) {
+function Pt(n, s) {
   const t = f();
-  m(() => {
+  b(() => {
     const e = s || t.quickActionRegistry.getActiveScope();
     return n.forEach((i) => {
       t.quickActionRegistry.register(i, e);
@@ -1569,8 +1631,8 @@ 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);
+function _t({ message: n, content: s }) {
+  const t = f(), e = s, i = bt(e.capabilityId), r = t.capabilityRegistry.getRenderer(e.capabilityName);
   if (!r || !i)
     return null;
   const a = r.component;
@@ -1586,16 +1648,16 @@ function ft({ message: n, content: s }) {
     }
   );
 }
-function Pt({ message: n, content: s }) {
-  const { component: t } = Q(s.type);
+function Dt({ message: n, content: s }) {
+  const { component: t } = $(s.type);
   if (s.type === "capability")
-    return /* @__PURE__ */ w.jsx(ft, { message: n, content: s });
+    return /* @__PURE__ */ w.jsx(_t, { message: n, content: s });
   if (!t)
     return null;
   const e = t;
   return /* @__PURE__ */ w.jsx(e, { content: s, message: n });
 }
-function Ft({
+function Nt({
   message: n,
   content: s,
   context: t,
@@ -1603,7 +1665,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,11 +1681,11 @@ function Ft({
     }
   );
 }
-function kt(n) {
-  const s = () => (lt(n), null);
+function Ot(n) {
+  const s = () => (ft(n), null);
   return s.displayName = `Capability(${n.name})`, s;
 }
-class Dt {
+class jt {
   getAllowedEvents() {
     return null;
   }
@@ -1662,15 +1724,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);
                 }
@@ -1687,39 +1749,39 @@ class Dt {
   }
 }
 export {
-  yt as AssistantRuntime,
-  _t as AssistantRuntimeProvider,
-  Ft as AuxiliaryRenderer,
-  Dt as BaseSSEStreamAdapter,
+  rt as AssistantRuntime,
+  wt as AssistantRuntimeProvider,
+  Nt as AuxiliaryRenderer,
+  jt as BaseSSEStreamAdapter,
   K as CapabilityExecutionManager,
   Y as CapabilityRegistry,
-  ft as CapabilityRendererComp,
-  vt as ChatContextProvider,
+  _t as CapabilityRendererComp,
+  Mt as ChatContextProvider,
   st as ComposerRuntime,
-  Pt as ContentRenderer,
+  Dt 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,
+  Ot as makeCapability,
   f as useAssistantRuntime,
-  lt as useCapability,
-  pt as useCapabilityExecution,
-  gt as useChatContext,
-  Ct as useComposer,
-  It as useContentFocus,
-  Q as useContentRenderer,
-  ct as useCurrentThread,
-  wt as useMessages,
-  Rt as usePageContext,
-  Mt as useQuickAction,
-  Tt as useQuickActionScope,
-  Et as useQuickActions,
-  At as useRegisterQuickActions,
-  St as useThreadList,
-  xt as useThreadListState
+  ft as useCapability,
+  bt as useCapabilityExecution,
+  yt as useChatContext,
+  Tt as useComposer,
+  Rt as useContentFocus,
+  $ as useContentRenderer,
+  dt as useCurrentThread,
+  vt as useMessages,
+  At as usePageContext,
+  kt as useQuickAction,
+  Et as useQuickActionScope,
+  Ft as useQuickActions,
+  Pt as useRegisterQuickActions,
+  Ct as useThreadList,
+  It as useThreadListState
 };
 //# sourceMappingURL=index.js.map