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

package/dist/hooks/use-copilot-chat_internal.js

@@ -83,7 +83,7 @@ __export(use_copilot_chat_internal_exports, {
   useCopilotChatInternal: () => useCopilotChatInternal
 });
 module.exports = __toCommonJS(use_copilot_chat_internal_exports);
-var import_react13 = require("react");
+var import_react14 = require("react");
 
 // src/context/copilot-context.tsx
 var import_react = __toESM(require("react"));
@@ -472,7 +472,7 @@ function useLangGraphInterruptRender(agent) {
 }
 
 // src/hooks/use-copilot-chat_internal.ts
-var import_react14 = require("@copilotkitnext/react");
+var import_react15 = require("@copilotkitnext/react");
 
 // src/hooks/use-lazy-tool-renderer.tsx
 var import_react9 = require("@copilotkitnext/react");
@@ -503,8 +503,11 @@ function useLazyToolRenderer() {
 var import_client = require("@ag-ui/client");
 
 // src/hooks/use-coagent-state-render-bridge.tsx
-var import_react11 = require("@copilotkitnext/react");
-var import_react12 = require("react");
+var import_react12 = require("@copilotkitnext/react");
+var import_react13 = require("react");
+var import_shared4 = require("@copilotkit/shared");
+
+// src/hooks/use-coagent-state-render-bridge.helpers.ts
 var import_shared3 = require("@copilotkit/shared");
 function getStateWithoutConstantKeys(state) {
   if (!state)
@@ -527,16 +530,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_shared3.dataToUUID)(getStateWithoutConstantKeys(claim.stateSnapshot)) === (0, import_shared3.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_shared3.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_react11 = 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_react11.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_react11.useAgent)({ agentId });
-  const [nodeName, setNodeName] = (0, import_react12.useState)(void 0);
-  const [, forceUpdate] = (0, import_react12.useState)(0);
-  const runId = (_a = props.runId) != null ? _a : message.runId;
-  const effectiveRunId = runId || "pending";
-  (0, import_react12.useEffect)(() => {
+  const { agent } = (0, import_react12.useAgent)({ agentId });
+  const [nodeName, setNodeName] = (0, import_react13.useState)(void 0);
+  const [, forceUpdate] = (0, import_react13.useState)(0);
+  (0, import_react13.useEffect)(() => {
     if (!agent)
       return;
     const subscriber = {
@@ -559,103 +840,69 @@ function useCoagentStateRenderBridge(agentId, props) {
       unsubscribe();
     };
   }, [agentId, nodeName]);
-  const getStateRender = (0, import_react12.useCallback)(
+  const getStateRender = (0, import_react13.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_react13.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_shared3.dataToUUID)(getStateWithoutConstantKeys(c.stateSnapshot)) === (0, import_shared3.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_react12.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_react13.useMemo)(() => {
+    var _a2, _b;
     if (!stateRender || !stateRenderId) {
       return null;
     }
-    const snapshot = stateSnapshot ? (0, import_shared3.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_shared3.parseJson)(stateSnapshot, stateSnapshot) : (_b = agent == null ? void 0 : agent.state) != null ? _b : {},
+        state: stateSnapshot ? (0, import_shared4.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) {
@@ -671,13 +918,13 @@ function useCopilotChatInternal({
   onReloadMessages
 } = {}) {
   var _a, _b, _c;
-  const { copilotkit } = (0, import_react14.useCopilotKit)();
+  const { copilotkit } = (0, import_react15.useCopilotKit)();
   const { threadId, agentSession } = useCopilotContext();
-  const existingConfig = (0, import_react14.useCopilotChatConfiguration)();
-  const [agentAvailable, setAgentAvailable] = (0, import_react13.useState)(false);
+  const existingConfig = (0, import_react15.useCopilotChatConfiguration)();
+  const [agentAvailable, setAgentAvailable] = (0, import_react14.useState)(false);
   const resolvedAgentId = (_a = existingConfig == null ? void 0 : existingConfig.agentId) != null ? _a : "default";
-  const { agent } = (0, import_react14.useAgent)({ agentId: resolvedAgentId });
-  (0, import_react13.useEffect)(() => {
+  const { agent } = (0, import_react15.useAgent)({ agentId: resolvedAgentId });
+  (0, import_react14.useEffect)(() => {
     const connect = (agent2) => __async(this, null, function* () {
       setAgentAvailable(false);
       try {
@@ -686,7 +933,7 @@ function useCopilotChatInternal({
       } catch (error) {
         if (error instanceof import_client.AGUIConnectNotImplementedError) {
         } else {
-          throw error;
+          console.error("CopilotChat: connectAgent failed", error);
         }
       }
     });
@@ -697,7 +944,7 @@ function useCopilotChatInternal({
     return () => {
     };
   }, [existingConfig == null ? void 0 : existingConfig.threadId, agent, copilotkit, resolvedAgentId]);
-  (0, import_react13.useEffect)(() => {
+  (0, import_react14.useEffect)(() => {
     onInProgress == null ? void 0 : onInProgress(Boolean(agent == null ? void 0 : agent.isRunning));
   }, [agent == null ? void 0 : agent.isRunning, onInProgress]);
   const interrupt = useLangGraphInterruptRender(agent);
@@ -705,7 +952,7 @@ function useCopilotChatInternal({
     agent == null ? void 0 : agent.setMessages([]);
     agent == null ? void 0 : agent.setState(null);
   };
-  const deleteMessage = (0, import_react13.useCallback)(
+  const deleteMessage = (0, import_react14.useCallback)(
     (messageId) => {
       var _a2;
       const filteredMessages = ((_a2 = agent == null ? void 0 : agent.messages) != null ? _a2 : []).filter(
@@ -716,13 +963,13 @@ function useCopilotChatInternal({
     [agent == null ? void 0 : agent.setMessages, agent == null ? void 0 : agent.messages]
   );
   const latestDelete = useUpdatedRef(deleteMessage);
-  const latestDeleteFunc = (0, import_react13.useCallback)(
+  const latestDeleteFunc = (0, import_react14.useCallback)(
     (messageId) => {
       return latestDelete.current(messageId);
     },
     [latestDelete]
   );
-  const currentSuggestions = (0, import_react14.useSuggestions)({ agentId: resolvedAgentId });
+  const currentSuggestions = (0, import_react15.useSuggestions)({ agentId: resolvedAgentId });
   const reload = useAsyncCallback(
     (reloadMessageId) => __async(this, null, function* () {
       var _a2;
@@ -757,7 +1004,11 @@ function useCopilotChatInternal({
       }
       agent == null ? void 0 : agent.setMessages(historyCutoff);
       if (agent) {
-        copilotkit.runAgent({ agent });
+        try {
+          yield copilotkit.runAgent({ agent });
+        } catch (error) {
+          console.error("CopilotChat: runAgent failed during reload", error);
+        }
       }
       return;
     }),
@@ -797,7 +1048,7 @@ function useCopilotChatInternal({
     }),
     [latestSendMessageFunc]
   );
-  const latestSetMessagesFunc = (0, import_react13.useCallback)(
+  const latestSetMessagesFunc = (0, import_react14.useCallback)(
     (messages) => {
       var _a2, _b2;
       if (messages.every((message) => message instanceof import_runtime_client_gql2.Message)) {
@@ -820,7 +1071,7 @@ function useCopilotChatInternal({
     }),
     [latestReload, agent, onReloadMessages]
   );
-  const latestStopFunc = (0, import_react13.useCallback)(() => {
+  const latestStopFunc = (0, import_react14.useCallback)(() => {
     var _a2, _b2;
     onStopGeneration == null ? void 0 : onStopGeneration({
       currentAgentName: agent == null ? void 0 : agent.agentId,
@@ -829,11 +1080,11 @@ function useCopilotChatInternal({
     return (_b2 = agent == null ? void 0 : agent.abortRun) == null ? void 0 : _b2.call(agent);
   }, [onStopGeneration, agent]);
   const latestReset = useUpdatedRef(reset);
-  const latestResetFunc = (0, import_react13.useCallback)(() => {
+  const latestResetFunc = (0, import_react14.useCallback)(() => {
     return latestReset.current();
   }, [latestReset]);
   const lazyToolRendered = useLazyToolRenderer();
-  const renderCustomMessage = (0, import_react14.useRenderCustomMessages)();
+  const renderCustomMessage = (0, import_react15.useRenderCustomMessages)();
   const legacyCustomMessageRenderer = useLegacyCoagentRenderer({
     copilotkit,
     agent,
@@ -841,7 +1092,8 @@ function useCopilotChatInternal({
     threadId: (_b = existingConfig == null ? void 0 : existingConfig.threadId) != null ? _b : threadId
   });
   const allMessages = (_c = agent == null ? void 0 : agent.messages) != null ? _c : [];
-  const resolvedMessages = (0, import_react13.useMemo)(() => {
+  const resolvedMessages = (0, import_react14.useMemo)(() => {
+    var _a2, _b2;
     let processedMessages = allMessages.map((message) => {
       if (message.role !== "assistant") {
         return message;
@@ -854,31 +1106,78 @@ function useCopilotChatInternal({
         }
       }
       const bridgeRenderer = legacyCustomMessageRenderer || renderCustomMessage ? () => {
-        const customRender = renderCustomMessage == null ? void 0 : renderCustomMessage({
-          message,
-          position: "before"
-        });
-        if (customRender) {
-          return customRender;
+        var _a3;
+        if (legacyCustomMessageRenderer) {
+          return legacyCustomMessageRenderer({ message, position: "before" });
+        }
+        try {
+          return (_a3 = renderCustomMessage == null ? void 0 : renderCustomMessage({ message, position: "before" })) != null ? _a3 : null;
+        } catch (error) {
+          console.warn(
+            "[CopilotKit] renderCustomMessages failed, falling back to legacy renderer",
+            error
+          );
+          return null;
         }
-        return legacyCustomMessageRenderer == null ? void 0 : legacyCustomMessageRenderer({ message, position: "before" });
       } : null;
       if (bridgeRenderer) {
-        return __spreadProps(__spreadValues({}, message), { generativeUI: bridgeRenderer });
+        return __spreadProps(__spreadValues({}, message), {
+          generativeUI: bridgeRenderer,
+          generativeUIPosition: "before"
+        });
       }
       return message;
     });
     const hasAssistantMessages = processedMessages.some((msg) => msg.role === "assistant");
+    const canUseCustomRenderer = Boolean(
+      renderCustomMessage && ((_a2 = copilotkit == null ? void 0 : copilotkit.getAgent) == null ? void 0 : _a2.call(copilotkit, resolvedAgentId))
+    );
+    const placeholderRenderer = legacyCustomMessageRenderer ? legacyCustomMessageRenderer : canUseCustomRenderer ? renderCustomMessage : null;
+    const shouldRenderPlaceholder = Boolean(agent == null ? void 0 : agent.isRunning) || Boolean((agent == null ? void 0 : agent.state) && Object.keys(agent.state).length);
+    const effectiveThreadId = (_b2 = threadId != null ? threadId : agent == null ? void 0 : agent.threadId) != null ? _b2 : "default";
+    let latestUserIndex = -1;
+    for (let i = processedMessages.length - 1; i >= 0; i -= 1) {
+      if (processedMessages[i].role === "user") {
+        latestUserIndex = i;
+        break;
+      }
+    }
+    const latestUserMessageId = latestUserIndex >= 0 ? processedMessages[latestUserIndex].id : void 0;
+    const currentRunId = latestUserMessageId ? copilotkit.getRunIdForMessage(resolvedAgentId, effectiveThreadId, latestUserMessageId) || `pending:${latestUserMessageId}` : void 0;
+    const hasAssistantForCurrentRun = latestUserIndex >= 0 ? processedMessages.slice(latestUserIndex + 1).some((msg) => msg.role === "assistant") : hasAssistantMessages;
+    if (placeholderRenderer && shouldRenderPlaceholder && !hasAssistantForCurrentRun) {
+      const placeholderId = currentRunId ? `coagent-state-render-${resolvedAgentId}-${currentRunId}` : `coagent-state-render-${resolvedAgentId}`;
+      const placeholderMessage = {
+        id: placeholderId,
+        role: "assistant",
+        content: "",
+        name: "coagent-state-render",
+        runId: currentRunId
+      };
+      processedMessages = [
+        ...processedMessages,
+        __spreadProps(__spreadValues({}, placeholderMessage), {
+          generativeUIPosition: "before",
+          generativeUI: () => placeholderRenderer({
+            message: placeholderMessage,
+            position: "before"
+          })
+        })
+      ];
+    }
     return processedMessages;
   }, [
     agent == null ? void 0 : agent.messages,
     lazyToolRendered,
     allMessages,
     renderCustomMessage,
-    // legacyCustomMessageRenderer,
-    resolvedAgentId
+    legacyCustomMessageRenderer,
+    resolvedAgentId,
+    copilotkit,
+    agent == null ? void 0 : agent.isRunning,
+    agent == null ? void 0 : agent.state
   ]);
-  const renderedSuggestions = (0, import_react13.useMemo)(() => {
+  const renderedSuggestions = (0, import_react14.useMemo)(() => {
     if (Array.isArray(suggestions)) {
       return {
         suggestions: suggestions.map((s) => __spreadProps(__spreadValues({}, s), { isLoading: false })),
@@ -913,8 +1212,8 @@ function useCopilotChatInternal({
   };
 }
 function useUpdatedRef(value) {
-  const ref = (0, import_react13.useRef)(value);
-  (0, import_react13.useEffect)(() => {
+  const ref = (0, import_react14.useRef)(value);
+  (0, import_react14.useEffect)(() => {
     ref.current = value;
   }, [value]);
   return ref;
@@ -925,14 +1224,15 @@ function useLegacyCoagentRenderer({
   agentId,
   threadId
 }) {
-  return (0, import_react13.useMemo)(() => {
+  return (0, import_react14.useMemo)(() => {
     if (!copilotkit || !agent) {
       return null;
     }
     return ({ message, position }) => {
       var _a;
       const effectiveThreadId = (_a = threadId != null ? threadId : agent.threadId) != null ? _a : "default";
-      const existingRunId = copilotkit.getRunIdForMessage(agentId, effectiveThreadId, message.id);
+      const providedRunId = message.runId;
+      const existingRunId = providedRunId ? providedRunId : copilotkit.getRunIdForMessage(agentId, effectiveThreadId, message.id);
       const runId = existingRunId || `pending:${message.id}`;
       const messageIndex = Math.max(
         agent.messages.findIndex((msg) => msg.id === message.id),
@@ -948,7 +1248,7 @@ function useLegacyCoagentRenderer({
         agentId,
         stateSnapshot: message.state
       };
-      return (0, import_react13.createElement)(CoAgentStateRenderBridge, bridgeProps);
+      return (0, import_react14.createElement)(CoAgentStateRenderBridge, bridgeProps);
     };
   }, [agent, agentId, copilotkit, threadId]);
 }