@copilotkitnext/react 1.51.3-next.2 → 1.51.3-next.4

package/dist/index.js

@@ -2514,16 +2514,18 @@ function useRenderCustomMessages() {
       return null;
     }
     const { message, position } = params;
-    const runId = copilotkit.getRunIdForMessage(agentId, threadId, message.id);
+    const resolvedRunId = copilotkit.getRunIdForMessage(agentId, threadId, message.id) ?? copilotkit.getRunIdsForThread(agentId, threadId).slice(-1)[0];
+    const runId = resolvedRunId ?? `missing-run-id:${message.id}`;
     const agent = copilotkit.getAgent(agentId);
     if (!agent) {
       throw new Error("Agent not found");
     }
-    const messagesIdsInRun = agent.messages.filter((msg) => copilotkit.getRunIdForMessage(agentId, threadId, msg.id) === runId).map((msg) => msg.id);
-    const messageIndex = agent.messages.findIndex((msg) => msg.id === message.id) ?? 0;
-    const messageIndexInRun = Math.min(messagesIdsInRun.indexOf(message.id), 0);
-    const numberOfMessagesInRun = messagesIdsInRun.length;
-    const stateSnapshot = copilotkit.getStateByRun(agentId, threadId, runId);
+    const messagesIdsInRun = resolvedRunId ? agent.messages.filter((msg) => copilotkit.getRunIdForMessage(agentId, threadId, msg.id) === resolvedRunId).map((msg) => msg.id) : [message.id];
+    const rawMessageIndex = agent.messages.findIndex((msg) => msg.id === message.id);
+    const messageIndex = rawMessageIndex >= 0 ? rawMessageIndex : 0;
+    const messageIndexInRun = resolvedRunId ? Math.max(messagesIdsInRun.indexOf(message.id), 0) : 0;
+    const numberOfMessagesInRun = resolvedRunId ? messagesIdsInRun.length : 1;
+    const stateSnapshot = resolvedRunId ? copilotkit.getStateByRun(agentId, threadId, resolvedRunId) : void 0;
     let result = null;
     for (const renderer of customMessageRenderers) {
       if (!renderer.render) {
@@ -3755,9 +3757,9 @@ function CopilotChatMessageView({
   }, [config?.agentId, copilotkit, forceUpdate]);
   const getStateSnapshotForMessage = (messageId) => {
     if (!config) return void 0;
-    const runId = copilotkit.getRunIdForMessage(config.agentId, config.threadId, messageId);
-    if (!runId) return void 0;
-    return copilotkit.getStateByRun(config.agentId, config.threadId, runId);
+    const resolvedRunId = copilotkit.getRunIdForMessage(config.agentId, config.threadId, messageId) ?? copilotkit.getRunIdsForThread(config.agentId, config.threadId).slice(-1)[0];
+    if (!resolvedRunId) return void 0;
+    return copilotkit.getStateByRun(config.agentId, config.threadId, resolvedRunId);
   };
   const messageElements = messages.flatMap((message) => {
     const elements = [];