@copilotkit/react-core 1.51.3-next.6 → 1.51.3-next.7

package/dist/components/copilot-provider/index.js

@@ -86,8 +86,8 @@ __export(copilot_provider_exports, {
 module.exports = __toCommonJS(copilot_provider_exports);
 
 // src/components/copilot-provider/copilotkit.tsx
-var import_react14 = require("react");
-var import_react15 = require("@copilotkitnext/react");
+var import_react15 = require("react");
+var import_react16 = require("@copilotkitnext/react");
 
 // src/context/copilot-context.tsx
 var import_react = __toESM(require("react"));
@@ -326,7 +326,7 @@ function setsHaveIntersection(setA, setB) {
 
 // src/components/copilot-provider/copilotkit.tsx
 var import_react_dom = require("react-dom");
-var import_shared12 = require("@copilotkit/shared");
+var import_shared13 = require("@copilotkit/shared");
 
 // src/hooks/use-flat-category-store.ts
 var import_react3 = require("react");
@@ -1228,8 +1228,8 @@ function useCoAgentStateRenders() {
 }
 
 // src/hooks/use-coagent-state-render-bridge.tsx
-var import_react10 = require("@copilotkitnext/react");
-var import_react11 = require("react");
+var import_react11 = require("@copilotkitnext/react");
+var import_react12 = require("react");
 
 // src/context/threads-context.tsx
 var import_react9 = require("react");
@@ -1259,6 +1259,9 @@ function useThreads() {
 }
 
 // src/hooks/use-coagent-state-render-bridge.tsx
+var import_shared10 = require("@copilotkit/shared");
+
+// src/hooks/use-coagent-state-render-bridge.helpers.ts
 var import_shared9 = require("@copilotkit/shared");
 function getStateWithoutConstantKeys(state) {
   if (!state)
@@ -1281,16 +1284,294 @@ function areStatesEquals(a, b) {
   ]);
   return JSON.stringify(aWithoutConstantKeys) === JSON.stringify(bWithoutConstantKeys);
 }
+function isPlaceholderMessageId(messageId) {
+  return !!messageId && messageId.startsWith("coagent-state-render-");
+}
+function isPlaceholderMessageName(messageName) {
+  return messageName === "coagent-state-render";
+}
+function readCachedMessageEntry(entry) {
+  if (!entry || typeof entry !== "object") {
+    return { snapshot: entry, runId: void 0 };
+  }
+  const snapshot = "snapshot" in entry ? entry.snapshot : entry;
+  const runId = "runId" in entry ? entry.runId : void 0;
+  return { snapshot, runId };
+}
+function getEffectiveRunId({
+  existingClaimRunId,
+  cachedMessageRunId,
+  runId
+}) {
+  return existingClaimRunId || cachedMessageRunId || runId || "pending";
+}
+function resolveClaim({
+  claims,
+  context,
+  stateSnapshot
+}) {
+  const { messageId, stateRenderId, runId, messageIndex } = context;
+  const existing = claims[messageId];
+  if (existing) {
+    const canRender = existing.stateRenderId === stateRenderId;
+    const shouldUpdateRunId = canRender && runId && (!existing.runId || existing.runId === "pending");
+    return {
+      canRender,
+      action: canRender ? "existing" /* Existing */ : "block" /* Block */,
+      updateRunId: shouldUpdateRunId ? runId : void 0
+    };
+  }
+  const normalizedRunId = runId != null ? runId : "pending";
+  const renderClaimedByOtherMessageEntry = Object.entries(claims).find(
+    ([, claim]) => {
+      var _a;
+      return claim.stateRenderId === stateRenderId && ((_a = claim.runId) != null ? _a : "pending") === normalizedRunId && (0, import_shared9.dataToUUID)(getStateWithoutConstantKeys(claim.stateSnapshot)) === (0, import_shared9.dataToUUID)(getStateWithoutConstantKeys(stateSnapshot));
+    }
+  );
+  const renderClaimedByOtherMessage = renderClaimedByOtherMessageEntry == null ? void 0 : renderClaimedByOtherMessageEntry[1];
+  const claimedMessageId = renderClaimedByOtherMessageEntry == null ? void 0 : renderClaimedByOtherMessageEntry[0];
+  if (renderClaimedByOtherMessage) {
+    if (messageIndex !== void 0 && renderClaimedByOtherMessage.messageIndex !== void 0 && messageIndex > renderClaimedByOtherMessage.messageIndex) {
+      return {
+        canRender: true,
+        action: "override" /* Override */,
+        nextClaim: { stateRenderId, runId, messageIndex },
+        lockOthers: runId === renderClaimedByOtherMessage.runId || isPlaceholderMessageId(claimedMessageId)
+      };
+    }
+    if (runId && renderClaimedByOtherMessage.runId && runId !== renderClaimedByOtherMessage.runId) {
+      return {
+        canRender: true,
+        action: "override" /* Override */,
+        nextClaim: { stateRenderId, runId, messageIndex },
+        lockOthers: isPlaceholderMessageId(claimedMessageId)
+      };
+    }
+    if (isPlaceholderMessageId(claimedMessageId)) {
+      return {
+        canRender: true,
+        action: "override" /* Override */,
+        nextClaim: { stateRenderId, runId, messageIndex },
+        lockOthers: true
+      };
+    }
+    if (stateSnapshot && renderClaimedByOtherMessage.stateSnapshot && !areStatesEquals(renderClaimedByOtherMessage.stateSnapshot, stateSnapshot)) {
+      return {
+        canRender: true,
+        action: "override" /* Override */,
+        nextClaim: { stateRenderId, runId }
+      };
+    }
+    return { canRender: false, action: "block" /* Block */ };
+  }
+  if (!runId) {
+    return { canRender: false, action: "block" /* Block */ };
+  }
+  return {
+    canRender: true,
+    action: "create" /* Create */,
+    nextClaim: { stateRenderId, runId, messageIndex }
+  };
+}
+function selectSnapshot({
+  messageId,
+  messageName,
+  allowLiveState,
+  skipLatestCache,
+  stateRenderId,
+  effectiveRunId,
+  stateSnapshotProp,
+  agentState,
+  agentMessages,
+  existingClaim,
+  caches
+}) {
+  var _a, _b, _c, _d, _e, _f;
+  const lastAssistantId = agentMessages ? (_a = [...agentMessages].reverse().find((msg) => msg.role === "assistant")) == null ? void 0 : _a.id : void 0;
+  const latestSnapshot = stateRenderId !== void 0 ? caches.byStateRenderAndRun[`${stateRenderId}::latest`] : void 0;
+  const messageIndex = agentMessages ? agentMessages.findIndex((msg) => msg.id === messageId) : -1;
+  const messageRole = messageIndex >= 0 && agentMessages ? (_b = agentMessages[messageIndex]) == null ? void 0 : _b.role : void 0;
+  let previousUserMessageId;
+  if (messageIndex > 0 && agentMessages) {
+    for (let i = messageIndex - 1; i >= 0; i -= 1) {
+      if (((_c = agentMessages[i]) == null ? void 0 : _c.role) === "user") {
+        previousUserMessageId = (_d = agentMessages[i]) == null ? void 0 : _d.id;
+        break;
+      }
+    }
+  }
+  const liveStateIsStale = stateSnapshotProp === void 0 && latestSnapshot !== void 0 && agentState !== void 0 && areStatesEquals(latestSnapshot, agentState);
+  const shouldUseLiveState = (Boolean(allowLiveState) || !lastAssistantId || messageId === lastAssistantId) && !liveStateIsStale;
+  const snapshot = stateSnapshotProp ? (0, import_shared9.parseJson)(stateSnapshotProp, stateSnapshotProp) : shouldUseLiveState ? agentState : void 0;
+  const hasSnapshotKeys = !!(snapshot && Object.keys(snapshot).length > 0);
+  const allowEmptySnapshot = snapshot !== void 0 && !hasSnapshotKeys && (stateSnapshotProp !== void 0 || shouldUseLiveState);
+  const messageCacheEntry = caches.byMessageId[messageId];
+  const cachedMessageSnapshot = readCachedMessageEntry(messageCacheEntry).snapshot;
+  const cacheKey = stateRenderId !== void 0 ? `${stateRenderId}::${effectiveRunId}` : void 0;
+  let cachedSnapshot = cachedMessageSnapshot != null ? cachedMessageSnapshot : caches.byMessageId[messageId];
+  if (cachedSnapshot === void 0 && cacheKey && caches.byStateRenderAndRun[cacheKey] !== void 0) {
+    cachedSnapshot = caches.byStateRenderAndRun[cacheKey];
+  }
+  if (cachedSnapshot === void 0 && stateRenderId && previousUserMessageId && caches.byStateRenderAndRun[`${stateRenderId}::pending:${previousUserMessageId}`] !== void 0) {
+    cachedSnapshot = caches.byStateRenderAndRun[`${stateRenderId}::pending:${previousUserMessageId}`];
+  }
+  if (cachedSnapshot === void 0 && !skipLatestCache && stateRenderId && messageRole !== "assistant" && (stateSnapshotProp !== void 0 || agentState && Object.keys(agentState).length > 0)) {
+    cachedSnapshot = caches.byStateRenderAndRun[`${stateRenderId}::latest`];
+  }
+  const snapshotForClaim = (existingClaim == null ? void 0 : existingClaim.locked) ? (_e = existingClaim.stateSnapshot) != null ? _e : cachedSnapshot : hasSnapshotKeys ? snapshot : (_f = existingClaim == null ? void 0 : existingClaim.stateSnapshot) != null ? _f : cachedSnapshot;
+  return { snapshot, hasSnapshotKeys, cachedSnapshot, allowEmptySnapshot, snapshotForClaim };
+}
+
+// src/hooks/use-coagent-state-render-registry.ts
+var import_react10 = require("react");
+var LAST_SNAPSHOTS_BY_RENDER_AND_RUN = "__lastSnapshotsByStateRenderIdAndRun";
+var LAST_SNAPSHOTS_BY_MESSAGE = "__lastSnapshotsByMessageId";
+function getClaimsStore(claimsRef) {
+  return claimsRef.current;
+}
+function getSnapshotCaches(claimsRef) {
+  var _a, _b;
+  const store = getClaimsStore(claimsRef);
+  return {
+    byStateRenderAndRun: (_a = store[LAST_SNAPSHOTS_BY_RENDER_AND_RUN]) != null ? _a : {},
+    byMessageId: (_b = store[LAST_SNAPSHOTS_BY_MESSAGE]) != null ? _b : {}
+  };
+}
+function useStateRenderRegistry({
+  agentId,
+  stateRenderId,
+  message,
+  messageIndex,
+  stateSnapshot,
+  agentState,
+  agentMessages,
+  claimsRef
+}) {
+  var _a, _b, _c, _d, _e, _f;
+  const store = getClaimsStore(claimsRef);
+  const runId = message.runId;
+  const cachedMessageEntry = (_a = store[LAST_SNAPSHOTS_BY_MESSAGE]) == null ? void 0 : _a[message.id];
+  const { runId: cachedMessageRunId } = readCachedMessageEntry(cachedMessageEntry);
+  const existingClaimRunId = (_b = claimsRef.current[message.id]) == null ? void 0 : _b.runId;
+  const effectiveRunId = getEffectiveRunId({
+    existingClaimRunId,
+    cachedMessageRunId,
+    runId
+  });
+  (0, import_react10.useEffect)(() => {
+    return () => {
+      var _a2, _b2, _c2, _d2;
+      const existingClaim2 = claimsRef.current[message.id];
+      if ((existingClaim2 == null ? void 0 : existingClaim2.stateSnapshot) && Object.keys(existingClaim2.stateSnapshot).length > 0) {
+        const snapshotCache = __spreadValues({}, (_a2 = store[LAST_SNAPSHOTS_BY_RENDER_AND_RUN]) != null ? _a2 : {});
+        const cacheKey = `${existingClaim2.stateRenderId}::${(_b2 = existingClaim2.runId) != null ? _b2 : "pending"}`;
+        snapshotCache[cacheKey] = existingClaim2.stateSnapshot;
+        snapshotCache[`${existingClaim2.stateRenderId}::latest`] = existingClaim2.stateSnapshot;
+        store[LAST_SNAPSHOTS_BY_RENDER_AND_RUN] = snapshotCache;
+        const messageCache = __spreadValues({}, (_c2 = store[LAST_SNAPSHOTS_BY_MESSAGE]) != null ? _c2 : {});
+        messageCache[message.id] = {
+          snapshot: existingClaim2.stateSnapshot,
+          runId: (_d2 = existingClaim2.runId) != null ? _d2 : effectiveRunId
+        };
+        store[LAST_SNAPSHOTS_BY_MESSAGE] = messageCache;
+      }
+      delete claimsRef.current[message.id];
+    };
+  }, [claimsRef, effectiveRunId, message.id]);
+  if (!stateRenderId) {
+    return { canRender: false };
+  }
+  const caches = getSnapshotCaches(claimsRef);
+  const existingClaim = claimsRef.current[message.id];
+  const { snapshot, hasSnapshotKeys, allowEmptySnapshot, snapshotForClaim } = selectSnapshot({
+    messageId: message.id,
+    messageName: message.name,
+    allowLiveState: isPlaceholderMessageName(message.name) || isPlaceholderMessageId(message.id),
+    skipLatestCache: isPlaceholderMessageName(message.name) || isPlaceholderMessageId(message.id),
+    stateRenderId,
+    effectiveRunId,
+    stateSnapshotProp: stateSnapshot,
+    agentState,
+    agentMessages,
+    existingClaim,
+    caches
+  });
+  const resolution = resolveClaim({
+    claims: claimsRef.current,
+    context: {
+      agentId,
+      messageId: message.id,
+      stateRenderId,
+      runId: effectiveRunId,
+      messageIndex
+    },
+    stateSnapshot: snapshotForClaim
+  });
+  if (resolution.action === "block" /* Block */) {
+    return { canRender: false };
+  }
+  if (resolution.updateRunId && claimsRef.current[message.id]) {
+    claimsRef.current[message.id].runId = resolution.updateRunId;
+  }
+  if (resolution.nextClaim) {
+    claimsRef.current[message.id] = resolution.nextClaim;
+  }
+  if (resolution.lockOthers) {
+    Object.entries(claimsRef.current).forEach(([id, claim]) => {
+      if (id !== message.id && claim.stateRenderId === stateRenderId) {
+        claim.locked = true;
+      }
+    });
+  }
+  if (existingClaim && !existingClaim.locked && (agentMessages == null ? void 0 : agentMessages.length)) {
+    const indexInAgentMessages = agentMessages.findIndex((msg) => msg.id === message.id);
+    if (indexInAgentMessages >= 0 && indexInAgentMessages < agentMessages.length - 1) {
+      existingClaim.locked = true;
+    }
+  }
+  const existingSnapshot = claimsRef.current[message.id].stateSnapshot;
+  const snapshotChanged = stateSnapshot && existingSnapshot !== void 0 && !areStatesEquals(existingSnapshot, snapshot);
+  if (snapshot && (stateSnapshot || hasSnapshotKeys || allowEmptySnapshot) && (!claimsRef.current[message.id].locked || snapshotChanged)) {
+    if (!claimsRef.current[message.id].locked || snapshotChanged) {
+      claimsRef.current[message.id].stateSnapshot = snapshot;
+      const snapshotCache = __spreadValues({}, (_c = store[LAST_SNAPSHOTS_BY_RENDER_AND_RUN]) != null ? _c : {});
+      const cacheKey = `${stateRenderId}::${effectiveRunId}`;
+      snapshotCache[cacheKey] = snapshot;
+      snapshotCache[`${stateRenderId}::latest`] = snapshot;
+      store[LAST_SNAPSHOTS_BY_RENDER_AND_RUN] = snapshotCache;
+      const messageCache = __spreadValues({}, (_d = store[LAST_SNAPSHOTS_BY_MESSAGE]) != null ? _d : {});
+      messageCache[message.id] = { snapshot, runId: effectiveRunId };
+      store[LAST_SNAPSHOTS_BY_MESSAGE] = messageCache;
+      if (stateSnapshot) {
+        claimsRef.current[message.id].locked = true;
+      }
+    }
+  } else if (snapshotForClaim) {
+    const existingSnapshot2 = claimsRef.current[message.id].stateSnapshot;
+    if (!existingSnapshot2) {
+      claimsRef.current[message.id].stateSnapshot = snapshotForClaim;
+      const snapshotCache = __spreadValues({}, (_e = store[LAST_SNAPSHOTS_BY_RENDER_AND_RUN]) != null ? _e : {});
+      const cacheKey = `${stateRenderId}::${effectiveRunId}`;
+      snapshotCache[cacheKey] = snapshotForClaim;
+      snapshotCache[`${stateRenderId}::latest`] = snapshotForClaim;
+      store[LAST_SNAPSHOTS_BY_RENDER_AND_RUN] = snapshotCache;
+      const messageCache = __spreadValues({}, (_f = store[LAST_SNAPSHOTS_BY_MESSAGE]) != null ? _f : {});
+      messageCache[message.id] = { snapshot: snapshotForClaim, runId: effectiveRunId };
+      store[LAST_SNAPSHOTS_BY_MESSAGE] = messageCache;
+    }
+  }
+  return { canRender: true };
+}
+
+// src/hooks/use-coagent-state-render-bridge.tsx
 function useCoagentStateRenderBridge(agentId, props) {
   var _a;
-  const { stateSnapshot, messageIndexInRun, message } = props;
+  const { stateSnapshot, message } = props;
   const { coAgentStateRenders, claimsRef } = useCoAgentStateRenders();
-  const { agent } = (0, import_react10.useAgent)({ agentId });
-  const [nodeName, setNodeName] = (0, import_react11.useState)(void 0);
-  const [, forceUpdate] = (0, import_react11.useState)(0);
-  const runId = (_a = props.runId) != null ? _a : message.runId;
-  const effectiveRunId = runId || "pending";
-  (0, import_react11.useEffect)(() => {
+  const { agent } = (0, import_react11.useAgent)({ agentId });
+  const [nodeName, setNodeName] = (0, import_react12.useState)(void 0);
+  const [, forceUpdate] = (0, import_react12.useState)(0);
+  (0, import_react12.useEffect)(() => {
     if (!agent)
       return;
     const subscriber = {
@@ -1313,103 +1594,69 @@ function useCoagentStateRenderBridge(agentId, props) {
       unsubscribe();
     };
   }, [agentId, nodeName]);
-  const getStateRender = (0, import_react11.useCallback)(
+  const getStateRender = (0, import_react12.useCallback)(
     (messageId) => {
-      return Object.entries(coAgentStateRenders).find(([stateRenderId, stateRender]) => {
+      return Object.entries(coAgentStateRenders).find(([stateRenderId2, stateRender2]) => {
         if (claimsRef.current[messageId]) {
-          return stateRenderId === claimsRef.current[messageId].stateRenderId;
+          return stateRenderId2 === claimsRef.current[messageId].stateRenderId;
         }
-        const matchingAgentName = stateRender.name === agentId;
-        const matchesNodeContext = stateRender.nodeName ? stateRender.nodeName === nodeName : true;
+        const matchingAgentName = stateRender2.name === agentId;
+        const matchesNodeContext = stateRender2.nodeName ? stateRender2.nodeName === nodeName : true;
         return matchingAgentName && matchesNodeContext;
       });
     },
     [coAgentStateRenders, nodeName, agentId]
   );
-  const handleRenderRequest = ({
+  const stateRenderEntry = (0, import_react12.useMemo)(() => getStateRender(message.id), [getStateRender, message.id]);
+  const stateRenderId = stateRenderEntry == null ? void 0 : stateRenderEntry[0];
+  const stateRender = stateRenderEntry == null ? void 0 : stateRenderEntry[1];
+  const registryMessage = __spreadProps(__spreadValues({}, message), {
+    runId: (_a = props.runId) != null ? _a : message.runId
+  });
+  const { canRender } = useStateRenderRegistry({
+    agentId,
     stateRenderId,
-    messageId,
-    runId: runId2,
-    stateSnapshot: renderSnapshot
-  }) => {
-    if (claimsRef.current[messageId]) {
-      const canRender = claimsRef.current[messageId].stateRenderId === stateRenderId;
-      if (canRender && runId2 && (!claimsRef.current[messageId].runId || claimsRef.current[messageId].runId === "pending")) {
-        claimsRef.current[messageId].runId = runId2;
-      }
-      return canRender;
-    }
-    const renderClaimedByOtherMessage = Object.values(claimsRef.current).find(
-      (c) => c.stateRenderId === stateRenderId && (0, import_shared9.dataToUUID)(getStateWithoutConstantKeys(c.stateSnapshot)) === (0, import_shared9.dataToUUID)(getStateWithoutConstantKeys(renderSnapshot))
-    );
-    if (renderClaimedByOtherMessage) {
-      if (renderSnapshot && renderClaimedByOtherMessage.stateSnapshot && !areStatesEquals(renderClaimedByOtherMessage.stateSnapshot, renderSnapshot)) {
-        claimsRef.current[messageId] = { stateRenderId, runId: runId2 };
-        return true;
-      }
-      return false;
-    }
-    if (!runId2) {
-      return false;
-    }
-    claimsRef.current[messageId] = { stateRenderId, runId: runId2 };
-    return true;
-  };
-  return (0, import_react11.useMemo)(() => {
-    var _a2, _b, _c;
-    if (messageIndexInRun !== 0) {
-      return null;
-    }
-    const [stateRenderId, stateRender] = (_a2 = getStateRender(message.id)) != null ? _a2 : [];
+    message: registryMessage,
+    messageIndex: props.messageIndex,
+    stateSnapshot,
+    agentState: agent == null ? void 0 : agent.state,
+    agentMessages: agent == null ? void 0 : agent.messages,
+    claimsRef
+  });
+  return (0, import_react12.useMemo)(() => {
+    var _a2, _b;
     if (!stateRender || !stateRenderId) {
       return null;
     }
-    const snapshot = stateSnapshot ? (0, import_shared9.parseJson)(stateSnapshot, stateSnapshot) : agent == null ? void 0 : agent.state;
-    const canRender = handleRenderRequest({
-      stateRenderId,
-      messageId: message.id,
-      runId: effectiveRunId,
-      stateSnapshot: snapshot
-    });
     if (!canRender) {
       return null;
     }
-    if (snapshot) {
-      const existingSnapshot = claimsRef.current[message.id].stateSnapshot;
-      const snapshotChanged = stateSnapshot && existingSnapshot !== void 0 && !areStatesEquals(existingSnapshot, snapshot);
-      if (!claimsRef.current[message.id].locked || snapshotChanged) {
-        claimsRef.current[message.id].stateSnapshot = snapshot;
-        if (stateSnapshot) {
-          claimsRef.current[message.id].locked = true;
-        }
-      }
-    }
     if (stateRender.handler) {
       stateRender.handler({
-        state: stateSnapshot ? (0, import_shared9.parseJson)(stateSnapshot, stateSnapshot) : (_b = agent == null ? void 0 : agent.state) != null ? _b : {},
+        state: stateSnapshot ? (0, import_shared10.parseJson)(stateSnapshot, stateSnapshot) : (_a2 = agent == null ? void 0 : agent.state) != null ? _a2 : {},
         nodeName: nodeName != null ? nodeName : ""
       });
     }
     if (stateRender.render) {
-      const status = (agent == null ? void 0 : agent.isRunning) ? "inProgress" : "complete";
+      const status = (agent == null ? void 0 : agent.isRunning) ? "inProgress" /* InProgress */ : "complete" /* Complete */;
       if (typeof stateRender.render === "string")
         return stateRender.render;
       return stateRender.render({
         status,
         // Always use state from claim, to make sure the state does not seem "wiped" for a fraction of a second
-        state: (_c = claimsRef.current[message.id].stateSnapshot) != null ? _c : {},
+        state: (_b = claimsRef.current[message.id].stateSnapshot) != null ? _b : {},
         nodeName: nodeName != null ? nodeName : ""
       });
     }
   }, [
-    getStateRender,
-    stateSnapshot,
+    stateRender,
+    stateRenderId,
     agent == null ? void 0 : agent.state,
     agent == null ? void 0 : agent.isRunning,
     nodeName,
-    effectiveRunId,
     message.id,
-    messageIndexInRun
+    stateSnapshot,
+    canRender
   ]);
 }
 function CoAgentStateRenderBridge(props) {
@@ -1417,13 +1664,13 @@ function CoAgentStateRenderBridge(props) {
 }
 
 // src/components/CopilotListeners.tsx
-var import_react12 = require("react");
-var import_react13 = require("@copilotkitnext/react");
-var import_shared10 = require("@copilotkit/shared");
+var import_react13 = require("react");
+var import_react14 = require("@copilotkitnext/react");
 var import_shared11 = require("@copilotkit/shared");
+var import_shared12 = require("@copilotkit/shared");
 var usePredictStateSubscription = (agent) => {
-  const predictStateToolsRef = (0, import_react12.useRef)([]);
-  const getSubscriber = (0, import_react12.useCallback)(
+  const predictStateToolsRef = (0, import_react13.useRef)([]);
+  const getSubscriber = (0, import_react13.useCallback)(
     (agent2) => ({
       onCustomEvent: ({ event }) => {
         if (event.name === "PredictState") {
@@ -1434,7 +1681,7 @@ var usePredictStateSubscription = (agent) => {
         predictStateToolsRef.current.forEach((t) => {
           if ((t == null ? void 0 : t.tool) !== toolCallName)
             return;
-          const emittedState = typeof partialToolCallArgs === "string" ? (0, import_shared10.parseJson)(partialToolCallArgs, partialToolCallArgs) : partialToolCallArgs;
+          const emittedState = typeof partialToolCallArgs === "string" ? (0, import_shared11.parseJson)(partialToolCallArgs, partialToolCallArgs) : partialToolCallArgs;
           agent2.setState({
             [t.state_key]: emittedState[t.state_key]
           });
@@ -1443,7 +1690,7 @@ var usePredictStateSubscription = (agent) => {
     }),
     []
   );
-  (0, import_react12.useEffect)(() => {
+  (0, import_react13.useEffect)(() => {
     if (!agent)
       return;
     const subscriber = getSubscriber(agent);
@@ -1454,16 +1701,16 @@ var usePredictStateSubscription = (agent) => {
   }, [agent, getSubscriber]);
 };
 function CopilotListeners() {
-  const { copilotkit } = (0, import_react13.useCopilotKit)();
-  const existingConfig = (0, import_react13.useCopilotChatConfiguration)();
+  const { copilotkit } = (0, import_react14.useCopilotKit)();
+  const existingConfig = (0, import_react14.useCopilotChatConfiguration)();
   const resolvedAgentId = existingConfig == null ? void 0 : existingConfig.agentId;
   const { setBannerError } = useToast();
-  const { agent } = (0, import_react13.useAgent)({ agentId: resolvedAgentId });
+  const { agent } = (0, import_react14.useAgent)({ agentId: resolvedAgentId });
   usePredictStateSubscription(agent);
-  (0, import_react12.useEffect)(() => {
+  (0, import_react13.useEffect)(() => {
     const subscriber = {
       onError: ({ error }) => {
-        setBannerError(new import_shared11.CopilotKitLowLevelError({ error, message: error.message }));
+        setBannerError(new import_shared12.CopilotKitLowLevelError({ error, message: error.message }));
       }
     };
     const subscription = copilotkit.subscribe(subscriber);
@@ -1481,9 +1728,9 @@ function CopilotKit(_a) {
   const enabled = shouldShowDevConsole(props.showDevConsole);
   const showInspector = shouldShowDevConsole(props.enableInspector);
   const publicApiKey = props.publicApiKey || props.publicLicenseKey;
-  const renderArr = (0, import_react14.useMemo)(() => [{ render: CoAgentStateRenderBridge }], []);
+  const renderArr = (0, import_react15.useMemo)(() => [{ render: CoAgentStateRenderBridge }], []);
   return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(ToastProvider, { enabled, children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(CopilotErrorBoundary, { publicApiKey, showUsageBanner: enabled, children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(ThreadsProvider, { threadId: props.threadId, children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
-    import_react15.CopilotKitProvider,
+    import_react16.CopilotKitProvider,
     __spreadProps(__spreadValues({}, props), {
       showDevConsole: showInspector,
       renderCustomMessages: renderArr,
@@ -1492,44 +1739,83 @@ function CopilotKit(_a) {
     })
   ) }) }) });
 }
+function CopilotKitErrorBridge() {
+  const { copilotkit } = (0, import_react16.useCopilotKit)();
+  const { onError, copilotApiConfig } = useCopilotContext();
+  (0, import_react15.useEffect)(() => {
+    if (!copilotkit)
+      return;
+    const subscription = copilotkit.subscribe({
+      onError: (event) => __async(this, null, function* () {
+        const errorEvent = {
+          type: "error",
+          timestamp: Date.now(),
+          context: __spreadValues({
+            source: "agent",
+            request: {
+              operation: event.code || "unknown",
+              url: copilotApiConfig == null ? void 0 : copilotApiConfig.chatApiEndpoint,
+              startTime: Date.now()
+            },
+            technical: {
+              environment: "browser",
+              userAgent: typeof navigator !== "undefined" ? navigator.userAgent : void 0,
+              stackTrace: event.error.stack
+            }
+          }, event.context),
+          error: event.error
+        };
+        try {
+          yield onError(errorEvent);
+        } catch (handlerError) {
+          console.error("Error in onError handler:", handlerError);
+        }
+      })
+    });
+    return () => {
+      subscription.unsubscribe();
+    };
+  }, [copilotkit, onError, copilotApiConfig]);
+  return null;
+}
 function CopilotKitInternal(cpkProps) {
   var _b;
   const _a = cpkProps, { children } = _a, props = __objRest(_a, ["children"]);
   validateProps(cpkProps);
   const publicApiKey = props.publicLicenseKey || props.publicApiKey;
-  const chatApiEndpoint = props.runtimeUrl || import_shared12.COPILOT_CLOUD_CHAT_URL;
-  const [actions, setActions] = (0, import_react14.useState)({});
-  const [registeredActionConfigs, setRegisteredActionConfigs] = (0, import_react14.useState)(/* @__PURE__ */ new Map());
-  const chatComponentsCache = (0, import_react14.useRef)({
+  const chatApiEndpoint = props.runtimeUrl || import_shared13.COPILOT_CLOUD_CHAT_URL;
+  const [actions, setActions] = (0, import_react15.useState)({});
+  const [registeredActionConfigs, setRegisteredActionConfigs] = (0, import_react15.useState)(/* @__PURE__ */ new Map());
+  const chatComponentsCache = (0, import_react15.useRef)({
     actions: {},
     coAgentStateRenders: {}
   });
   const { addElement, removeElement, printTree, getAllElements } = use_tree_default();
-  const [isLoading, setIsLoading] = (0, import_react14.useState)(false);
-  const [chatInstructions, setChatInstructions] = (0, import_react14.useState)("");
-  const [authStates, setAuthStates] = (0, import_react14.useState)({});
-  const [extensions, setExtensions] = (0, import_react14.useState)({});
-  const [additionalInstructions, setAdditionalInstructions] = (0, import_react14.useState)([]);
+  const [isLoading, setIsLoading] = (0, import_react15.useState)(false);
+  const [chatInstructions, setChatInstructions] = (0, import_react15.useState)("");
+  const [authStates, setAuthStates] = (0, import_react15.useState)({});
+  const [extensions, setExtensions] = (0, import_react15.useState)({});
+  const [additionalInstructions, setAdditionalInstructions] = (0, import_react15.useState)([]);
   const {
     addElement: addDocument,
     removeElement: removeDocument,
     allElements: allDocuments
   } = use_flat_category_store_default();
-  const setAction = (0, import_react14.useCallback)((id, action) => {
+  const setAction = (0, import_react15.useCallback)((id, action) => {
     setActions((prevPoints) => {
       return __spreadProps(__spreadValues({}, prevPoints), {
         [id]: action
       });
     });
   }, []);
-  const removeAction = (0, import_react14.useCallback)((id) => {
+  const removeAction = (0, import_react15.useCallback)((id) => {
     setActions((prevPoints) => {
       const newPoints = __spreadValues({}, prevPoints);
       delete newPoints[id];
       return newPoints;
     });
   }, []);
-  const getContextString = (0, import_react14.useCallback)(
+  const getContextString = (0, import_react15.useCallback)(
     (documents, categories) => {
       const documentsString = documents.map((document) => {
         return `${document.name} (${document.sourceApplication}):
@@ -1542,46 +1828,46 @@ ${nonDocumentStrings}`;
     },
     [printTree]
   );
-  const addContext = (0, import_react14.useCallback)(
+  const addContext = (0, import_react15.useCallback)(
     (context, parentId, categories = defaultCopilotContextCategories) => {
       return addElement(context, categories, parentId);
     },
     [addElement]
   );
-  const removeContext = (0, import_react14.useCallback)(
+  const removeContext = (0, import_react15.useCallback)(
     (id) => {
       removeElement(id);
     },
     [removeElement]
   );
-  const getAllContext = (0, import_react14.useCallback)(() => {
+  const getAllContext = (0, import_react15.useCallback)(() => {
     return getAllElements();
   }, [getAllElements]);
-  const getFunctionCallHandler = (0, import_react14.useCallback)(
+  const getFunctionCallHandler = (0, import_react15.useCallback)(
     (customEntryPoints) => {
       return entryPointsToFunctionCallHandler(Object.values(customEntryPoints || actions));
     },
     [actions]
   );
-  const getDocumentsContext = (0, import_react14.useCallback)(
+  const getDocumentsContext = (0, import_react15.useCallback)(
     (categories) => {
       return allDocuments(categories);
     },
     [allDocuments]
   );
-  const addDocumentContext = (0, import_react14.useCallback)(
+  const addDocumentContext = (0, import_react15.useCallback)(
     (documentPointer, categories = defaultCopilotContextCategories) => {
       return addDocument(documentPointer, categories);
     },
     [addDocument]
   );
-  const removeDocumentContext = (0, import_react14.useCallback)(
+  const removeDocumentContext = (0, import_react15.useCallback)(
     (documentId) => {
       removeDocument(documentId);
     },
     [removeDocument]
   );
-  const copilotApiConfig = (0, import_react14.useMemo)(() => {
+  const copilotApiConfig = (0, import_react15.useMemo)(() => {
     var _a2, _b2;
     let cloud = void 0;
     if (publicApiKey) {
@@ -1617,7 +1903,7 @@ ${nonDocumentStrings}`;
     props.cloudRestrictToTopic,
     props.guardrails_c
   ]);
-  const headers = (0, import_react14.useMemo)(() => {
+  const headers = (0, import_react15.useMemo)(() => {
     const authHeaders = Object.values(authStates || {}).reduce((acc, state) => {
       if (state.status === "authenticated" && state.authHeaders) {
         return __spreadValues(__spreadValues({}, acc), Object.entries(state.authHeaders).reduce(
@@ -1629,27 +1915,27 @@ ${nonDocumentStrings}`;
       }
       return acc;
     }, {});
-    return __spreadValues(__spreadValues(__spreadValues({}, copilotApiConfig.headers || {}), copilotApiConfig.publicApiKey ? { [import_shared12.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: copilotApiConfig.publicApiKey } : {}), authHeaders);
+    return __spreadValues(__spreadValues(__spreadValues({}, copilotApiConfig.headers || {}), copilotApiConfig.publicApiKey ? { [import_shared13.COPILOT_CLOUD_PUBLIC_API_KEY_HEADER]: copilotApiConfig.publicApiKey } : {}), authHeaders);
   }, [copilotApiConfig.headers, copilotApiConfig.publicApiKey, authStates]);
-  const [internalErrorHandlers, _setInternalErrorHandler] = (0, import_react14.useState)({});
-  const setInternalErrorHandler = (0, import_react14.useCallback)((handler) => {
+  const [internalErrorHandlers, _setInternalErrorHandler] = (0, import_react15.useState)({});
+  const setInternalErrorHandler = (0, import_react15.useCallback)((handler) => {
     _setInternalErrorHandler((prev) => __spreadValues(__spreadValues({}, prev), handler));
   }, []);
-  const removeInternalErrorHandler = (0, import_react14.useCallback)((key) => {
+  const removeInternalErrorHandler = (0, import_react15.useCallback)((key) => {
     _setInternalErrorHandler((prev) => {
       const _a2 = prev, { [key]: _removed } = _a2, rest = __objRest(_a2, [__restKey(key)]);
       return rest;
     });
   }, []);
-  const onErrorRef = (0, import_react14.useRef)(props.onError);
-  (0, import_react14.useEffect)(() => {
+  const onErrorRef = (0, import_react15.useRef)(props.onError);
+  (0, import_react15.useEffect)(() => {
     onErrorRef.current = props.onError;
   }, [props.onError]);
-  const internalHandlersRef = (0, import_react14.useRef)({});
-  (0, import_react14.useEffect)(() => {
+  const internalHandlersRef = (0, import_react15.useRef)({});
+  (0, import_react15.useEffect)(() => {
     internalHandlersRef.current = internalErrorHandlers;
   }, [internalErrorHandlers]);
-  const handleErrors = (0, import_react14.useCallback)(
+  const handleErrors = (0, import_react15.useCallback)(
     (error) => __async(this, null, function* () {
       if (copilotApiConfig.publicApiKey && onErrorRef.current) {
         try {
@@ -1669,14 +1955,14 @@ ${nonDocumentStrings}`;
     }),
     [copilotApiConfig.publicApiKey]
   );
-  const [chatSuggestionConfiguration, setChatSuggestionConfiguration] = (0, import_react14.useState)({});
-  const addChatSuggestionConfiguration = (0, import_react14.useCallback)(
+  const [chatSuggestionConfiguration, setChatSuggestionConfiguration] = (0, import_react15.useState)({});
+  const addChatSuggestionConfiguration = (0, import_react15.useCallback)(
     (id, suggestion) => {
       setChatSuggestionConfiguration((prev) => __spreadProps(__spreadValues({}, prev), { [id]: suggestion }));
     },
     [setChatSuggestionConfiguration]
   );
-  const removeChatSuggestionConfiguration = (0, import_react14.useCallback)(
+  const removeChatSuggestionConfiguration = (0, import_react15.useCallback)(
     (id) => {
       setChatSuggestionConfiguration((prev) => {
         const _a2 = prev, { [id]: _ } = _a2, rest = __objRest(_a2, [__restKey(id)]);
@@ -1685,10 +1971,10 @@ ${nonDocumentStrings}`;
     },
     [setChatSuggestionConfiguration]
   );
-  const [availableAgents, setAvailableAgents] = (0, import_react14.useState)([]);
-  const [coagentStates, setCoagentStates] = (0, import_react14.useState)({});
-  const coagentStatesRef = (0, import_react14.useRef)({});
-  const setCoagentStatesWithRef = (0, import_react14.useCallback)(
+  const [availableAgents, setAvailableAgents] = (0, import_react15.useState)([]);
+  const [coagentStates, setCoagentStates] = (0, import_react15.useState)({});
+  const coagentStatesRef = (0, import_react15.useRef)({});
+  const setCoagentStatesWithRef = (0, import_react15.useCallback)(
     (value) => {
       const newValue = typeof value === "function" ? value(coagentStatesRef.current) : value;
       coagentStatesRef.current = newValue;
@@ -1704,8 +1990,8 @@ ${nonDocumentStrings}`;
       agentName: props.agent
     };
   }
-  const [agentSession, setAgentSession] = (0, import_react14.useState)(initialAgentSession);
-  (0, import_react14.useEffect)(() => {
+  const [agentSession, setAgentSession] = (0, import_react15.useState)(initialAgentSession);
+  (0, import_react15.useEffect)(() => {
     if (props.agent) {
       setAgentSession({
         agentName: props.agent
@@ -1715,7 +2001,7 @@ ${nonDocumentStrings}`;
     }
   }, [props.agent]);
   const { threadId, setThreadId: setInternalThreadId } = useThreads();
-  const setThreadId = (0, import_react14.useCallback)(
+  const setThreadId = (0, import_react15.useCallback)(
     (value) => {
       if (props.threadId) {
         throw new Error("Cannot call setThreadId() when threadId is provided via props.");
@@ -1724,11 +2010,11 @@ ${nonDocumentStrings}`;
     },
     [props.threadId]
   );
-  const [runId, setRunId] = (0, import_react14.useState)(null);
-  const chatAbortControllerRef = (0, import_react14.useRef)(null);
+  const [runId, setRunId] = (0, import_react15.useState)(null);
+  const chatAbortControllerRef = (0, import_react15.useRef)(null);
   const showDevConsole = shouldShowDevConsole(props.showDevConsole);
-  const [interruptActions, _setInterruptActions] = (0, import_react14.useState)({});
-  const setInterruptAction = (0, import_react14.useCallback)((action) => {
+  const [interruptActions, _setInterruptActions] = (0, import_react15.useState)({});
+  const setInterruptAction = (0, import_react15.useCallback)((action) => {
     _setInterruptActions((prev) => {
       var _a2;
       if (action == null || !action.id) {
@@ -1739,14 +2025,14 @@ ${nonDocumentStrings}`;
       });
     });
   }, []);
-  const removeInterruptAction = (0, import_react14.useCallback)((actionId) => {
+  const removeInterruptAction = (0, import_react15.useCallback)((actionId) => {
     _setInterruptActions((prev) => {
       const _a2 = prev, { [actionId]: _ } = _a2, rest = __objRest(_a2, [__restKey(actionId)]);
       return rest;
     });
   }, []);
-  const [interruptEventQueue, setInterruptEventQueue] = (0, import_react14.useState)({});
-  const addInterruptEvent = (0, import_react14.useCallback)((queuedEvent) => {
+  const [interruptEventQueue, setInterruptEventQueue] = (0, import_react15.useState)({});
+  const addInterruptEvent = (0, import_react15.useCallback)((queuedEvent) => {
     setInterruptEventQueue((prev) => {
       const threadQueue = prev[queuedEvent.threadId] || [];
       return __spreadProps(__spreadValues({}, prev), {
@@ -1754,7 +2040,7 @@ ${nonDocumentStrings}`;
       });
     });
   }, []);
-  const resolveInterruptEvent = (0, import_react14.useCallback)(
+  const resolveInterruptEvent = (0, import_react15.useCallback)(
     (threadId2, eventId, response) => {
       setInterruptEventQueue((prev) => {
         const threadQueue = prev[threadId2] || [];
@@ -1767,20 +2053,20 @@ ${nonDocumentStrings}`;
     },
     []
   );
-  const memoizedChildren = (0, import_react14.useMemo)(() => children, [children]);
-  const [bannerError, setBannerError] = (0, import_react14.useState)(null);
-  const agentLock = (0, import_react14.useMemo)(() => {
+  const memoizedChildren = (0, import_react15.useMemo)(() => children, [children]);
+  const [bannerError, setBannerError] = (0, import_react15.useState)(null);
+  const agentLock = (0, import_react15.useMemo)(() => {
     var _a2;
     return (_a2 = props.agent) != null ? _a2 : null;
   }, [props.agent]);
-  const forwardedParameters = (0, import_react14.useMemo)(
+  const forwardedParameters = (0, import_react15.useMemo)(
     () => {
       var _a2;
       return (_a2 = props.forwardedParameters) != null ? _a2 : {};
     },
     [props.forwardedParameters]
   );
-  const updateExtensions = (0, import_react14.useCallback)(
+  const updateExtensions = (0, import_react15.useCallback)(
     (newExtensions) => {
       setExtensions((prev) => {
         const resolved = typeof newExtensions === "function" ? newExtensions(prev) : newExtensions;
@@ -1792,7 +2078,7 @@ ${nonDocumentStrings}`;
     },
     [setExtensions]
   );
-  const updateAuthStates = (0, import_react14.useCallback)(
+  const updateAuthStates = (0, import_react15.useCallback)(
     (newAuthStates) => {
       setAuthStates((prev) => {
         const resolved = typeof newAuthStates === "function" ? newAuthStates(prev) : newAuthStates;
@@ -1804,8 +2090,8 @@ ${nonDocumentStrings}`;
     },
     [setAuthStates]
   );
-  const handleSetRegisteredActions = (0, import_react14.useCallback)((actionConfig) => {
-    const key = actionConfig.action.name || (0, import_shared12.randomUUID)();
+  const handleSetRegisteredActions = (0, import_react15.useCallback)((actionConfig) => {
+    const key = actionConfig.action.name || (0, import_shared13.randomUUID)();
     setRegisteredActionConfigs((prev) => {
       const newMap = new Map(prev);
       newMap.set(key, actionConfig);
@@ -1813,21 +2099,21 @@ ${nonDocumentStrings}`;
     });
     return key;
   }, []);
-  const handleRemoveRegisteredAction = (0, import_react14.useCallback)((actionKey) => {
+  const handleRemoveRegisteredAction = (0, import_react15.useCallback)((actionKey) => {
     setRegisteredActionConfigs((prev) => {
       const newMap = new Map(prev);
       newMap.delete(actionKey);
       return newMap;
     });
   }, []);
-  const RegisteredActionsRenderer = (0, import_react14.useMemo)(() => {
+  const RegisteredActionsRenderer = (0, import_react15.useMemo)(() => {
     return () => /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_jsx_runtime7.Fragment, { children: Array.from(registeredActionConfigs.entries()).map(([key, config]) => {
       const Component = config.component;
       return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(Component, { action: config.action }, key);
     }) });
   }, [registeredActionConfigs]);
   return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
-    import_react15.CopilotChatConfigurationProvider,
+    import_react16.CopilotChatConfigurationProvider,
     {
       agentId: (_b = props.agent) != null ? _b : "default",
       threadId,
@@ -1894,6 +2180,7 @@ ${nonDocumentStrings}`;
           },
           children: [
             /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(CopilotListeners, {}),
+            /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(CopilotKitErrorBridge, {}),
             /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)(CoAgentStateRendersProvider, { children: [
               /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(MessagesTapProvider, { children: /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)(CopilotMessages, { children: [
                 memoizedChildren,
@@ -1948,12 +2235,12 @@ function validateProps(props) {
   const cloudFeatures = Object.keys(props).filter((key) => key.endsWith("_c"));
   const hasApiKey = props.publicApiKey || props.publicLicenseKey;
   if (!props.runtimeUrl && !hasApiKey) {
-    throw new import_shared12.ConfigurationError(
+    throw new import_shared13.ConfigurationError(
       "Missing required prop: 'runtimeUrl' or 'publicApiKey' or 'publicLicenseKey'"
     );
   }
   if (cloudFeatures.length > 0 && !hasApiKey) {
-    throw new import_shared12.MissingPublicApiKeyError(
+    throw new import_shared13.MissingPublicApiKeyError(
       `Missing required prop: 'publicApiKey' or 'publicLicenseKey' to use cloud features: ${cloudFeatures.map(formatFeatureName).join(", ")}`
     );
   }