@copilotkit/react-core 1.51.3-next.3 → 1.51.3-next.5

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # ui
 
+## 1.51.3-next.5
+
+### Patch Changes
+
+- e89fe98: fix(v1.x): address issue causing onSubmitMessage to fire after agent response
+  - @copilotkit/runtime-client-gql@1.51.3-next.5
+  - @copilotkit/shared@1.51.3-next.5
+  - @copilotkitnext/core@1.51.3-next.5
+  - @copilotkitnext/react@1.51.3-next.5
+
+## 1.51.3-next.4
+
+### Patch Changes
+
+- @copilotkit/runtime-client-gql@1.51.3-next.4
+- @copilotkit/shared@1.51.3-next.4
+- @copilotkitnext/core@1.51.3-next.4
+- @copilotkitnext/react@1.51.3-next.4
+
 ## 1.51.3-next.3
 
 ### Patch Changes

package/dist/{chunk-OBXQSB7S.mjs → chunk-3HJ2O2MN.mjs}

@@ -9,7 +9,7 @@ import {
 } from "./chunk-BVBFFI7K.mjs";
 import {
   CoAgentStateRenderBridge
-} from "./chunk-QNUAXSDP.mjs";
+} from "./chunk-3X3I7OJV.mjs";
 import {
   useCopilotContext
 } from "./chunk-AFNWX62Q.mjs";
@@ -139,6 +139,14 @@ function useCopilotChatInternal({
       if (options == null ? void 0 : options.clearSuggestions) {
         copilotkit.clearSuggestions(resolvedAgentId);
       }
+      if (onSubmitMessage) {
+        const content = typeof message.content === "string" ? message.content : message.content && "text" in message.content ? message.content.text : message.content && "filename" in message.content ? message.content.filename : "";
+        try {
+          yield onSubmitMessage(content);
+        } catch (error) {
+          console.error("Error in onSubmitMessage:", error);
+        }
+      }
       agent == null ? void 0 : agent.addMessage(message);
       if (followUp) {
         try {
@@ -147,10 +155,6 @@ function useCopilotChatInternal({
           console.error("CopilotChat: runAgent failed", error);
         }
       }
-      if (onSubmitMessage) {
-        const content = typeof message.content === "string" ? message.content : message.content && "text" in message.content ? message.content.text : message.content && "filename" in message.content ? message.content.filename : "";
-        onSubmitMessage(content);
-      }
     }),
     [agent, copilotkit, resolvedAgentId, onSubmitMessage]
   );
@@ -346,4 +350,4 @@ export {
   useCopilotChatInternal,
   defaultSystemMessage
 };
-//# sourceMappingURL=chunk-OBXQSB7S.mjs.map
+//# sourceMappingURL=chunk-3HJ2O2MN.mjs.map

package/dist/chunk-3HJ2O2MN.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-copilot-chat_internal.ts"],"sourcesContent":["import { useRef, useEffect, useCallback, useMemo, useState, createElement } from \"react\";\nimport { useCopilotContext } from \"../context/copilot-context\";\nimport { SystemMessageFunction } from \"../types\";\nimport { useAsyncCallback } from \"../components/error-boundary/error-utils\";\nimport { Message } from \"@copilotkit/shared\";\nimport { gqlToAGUI, Message as DeprecatedGqlMessage } from \"@copilotkit/runtime-client-gql\";\nimport { useLangGraphInterruptRender } from \"./use-langgraph-interrupt-render\";\nimport {\n  useAgent,\n  useCopilotChatConfiguration,\n  useCopilotKit,\n  useRenderCustomMessages,\n  useSuggestions,\n} from \"@copilotkitnext/react\";\nimport { Suggestion } from \"@copilotkitnext/core\";\nimport { useLazyToolRenderer } from \"./use-lazy-tool-renderer\";\nimport { AbstractAgent, AGUIConnectNotImplementedError } from \"@ag-ui/client\";\nimport {\n  CoAgentStateRenderBridge,\n  type CoAgentStateRenderBridgeProps,\n} from \"./use-coagent-state-render-bridge\";\n\n/**\n * The type of suggestions to use in the chat.\n *\n * `auto` - Suggestions are generated automatically.\n * `manual` - Suggestions are controlled programmatically.\n * `SuggestionItem[]` - Static suggestions array.\n */\nexport type ChatSuggestions = \"auto\" | \"manual\" | Omit<Suggestion, \"isLoading\">[];\n\nexport interface AppendMessageOptions {\n  /**\n   * Whether to run the chat completion after appending the message. Defaults to `true`.\n   */\n  followUp?: boolean;\n  /**\n   * Whether to clear the suggestions after appending the message. Defaults to `true`.\n   */\n  clearSuggestions?: boolean;\n}\n\nexport interface OnStopGenerationArguments {\n  /**\n   * The name of the currently executing agent.\n   */\n  currentAgentName: string | undefined;\n\n  /**\n   * The messages in the chat.\n   */\n  messages: Message[];\n}\n\nexport type OnReloadMessagesArguments = OnStopGenerationArguments & {\n  /**\n   * The message on which \"regenerate\" was pressed\n   */\n  messageId: string;\n};\n\nexport type OnStopGeneration = (args: OnStopGenerationArguments) => void;\n\nexport type OnReloadMessages = (args: OnReloadMessagesArguments) => void;\n\nexport interface UseCopilotChatOptions {\n  /**\n   * A unique identifier for the chat. If not provided, a random one will be\n   * generated. When provided, the `useChat` hook with the same `id` will\n   * have shared states across components.\n   */\n  id?: string;\n\n  /**\n   * HTTP headers to be sent with the API request.\n   */\n  headers?: Record<string, string> | Headers;\n\n  /**\n   * Initial messages to populate the chat with.\n   */\n  initialMessages?: Message[];\n\n  /**\n   * A function to generate the system message. Defaults to `defaultSystemMessage`.\n   */\n  makeSystemMessage?: SystemMessageFunction;\n\n  /**\n   * Disables inclusion of CopilotKit’s default system message. When true, no system message is sent (this also suppresses any custom message from <code>makeSystemMessage</code>).\n   */\n  disableSystemMessage?: boolean;\n  /**\n   * Controls the behavior of suggestions in the chat interface.\n   *\n   * `auto` (default) - Suggestions are generated automatically:\n   *   - When the chat is first opened (empty state)\n   *   - After each message exchange completes\n   *   - Uses configuration from `useCopilotChatSuggestions` hooks\n   *\n   * `manual` - Suggestions are controlled programmatically:\n   *   - Use `setSuggestions()` to set custom suggestions\n   *   - Use `generateSuggestions()` to trigger AI generation\n   *   - Access via `useCopilotChat` hook\n   *\n   * `SuggestionItem[]` - Static suggestions array:\n   *   - Always shows the same suggestions\n   *   - No AI generation involved\n   */\n  suggestions?: ChatSuggestions;\n\n  onInProgress?: (isLoading: boolean) => void;\n  onSubmitMessage?: (messageContent: string) => Promise<void> | void;\n  onStopGeneration?: OnStopGeneration;\n  onReloadMessages?: OnReloadMessages;\n}\n\nexport interface MCPServerConfig {\n  endpoint: string;\n  apiKey?: string;\n}\n\n// Old suggestion item interface, for returning from useCopilotChatInternal\ninterface SuggestionItem {\n  title: string;\n  message: string;\n  partial?: boolean;\n  className?: string;\n}\n\nexport interface UseCopilotChatReturn {\n  /**\n   * @deprecated use `messages` instead, this is an old non ag-ui version of the messages\n   * Array of messages currently visible in the chat interface\n   *\n   * This is the visible messages, not the raw messages from the runtime client.\n   */\n  visibleMessages: DeprecatedGqlMessage[];\n\n  /**\n   * The messages that are currently in the chat in AG-UI format.\n   */\n  messages: Message[];\n\n  /** @deprecated use `sendMessage` in `useCopilotChatHeadless_c` instead. This will be removed in a future major version. */\n  appendMessage: (message: DeprecatedGqlMessage, options?: AppendMessageOptions) => Promise<void>;\n\n  /**\n   * Send a new message to the chat\n   *\n   * ```tsx\n   * await sendMessage({\n   *   id: \"123\",\n   *   role: \"user\",\n   *   content: \"Hello, process this request\",\n   * });\n   * ```\n   */\n  sendMessage: (message: Message, options?: AppendMessageOptions) => Promise<void>;\n\n  /**\n   * Replace all messages in the chat\n   *\n   * ```tsx\n   * setMessages([\n   *   { id: \"123\", role: \"user\", content: \"Hello, process this request\" },\n   *   { id: \"456\", role: \"assistant\", content: \"Hello, I'm the assistant\" },\n   * ]);\n   * ```\n   *\n   * **Deprecated** non-ag-ui version:\n   *\n   * ```tsx\n   * setMessages([\n   *   new TextMessage({\n   *     content: \"Hello, process this request\",\n   *     role: gqlRole.User,\n   *   }),\n   *   new TextMessage({\n   *     content: \"Hello, I'm the assistant\",\n   *     role: gqlRole.Assistant,\n   * ]);\n   * ```\n   *\n   */\n  setMessages: (messages: Message[] | DeprecatedGqlMessage[]) => void;\n\n  /**\n   * Remove a specific message by ID\n   *\n   * ```tsx\n   * deleteMessage(\"123\");\n   * ```\n   */\n  deleteMessage: (messageId: string) => void;\n\n  /**\n   * Regenerate the response for a specific message\n   *\n   * ```tsx\n   * reloadMessages(\"123\");\n   * ```\n   */\n  reloadMessages: (messageId: string) => Promise<void>;\n\n  /**\n   * Stop the current message generation\n   *\n   * ```tsx\n   * if (isLoading) {\n   *   stopGeneration();\n   * }\n   * ```\n   */\n  stopGeneration: () => void;\n\n  /**\n   * Clear all messages and reset chat state\n   *\n   * ```tsx\n   * reset();\n   * console.log(messages); // []\n   * ```\n   */\n  reset: () => void;\n\n  /**\n   * Whether the chat is currently generating a response\n   *\n   * ```tsx\n   * if (isLoading) {\n   *   console.log(\"Loading...\");\n   * } else {\n   *   console.log(\"Not loading\");\n   * }\n   */\n  isLoading: boolean;\n\n  /**\n   * Whether the chat agent is available to generate responses\n   *\n   * ```tsx\n   * if (isAvailable) {\n   *   console.log(\"Loading...\");\n   * } else {\n   *   console.log(\"Not loading\");\n   * }\n   */\n  isAvailable: boolean;\n\n  /** Manually trigger chat completion (advanced usage) */\n  runChatCompletion: () => Promise<Message[]>;\n\n  /** MCP (Model Context Protocol) server configurations */\n  mcpServers: MCPServerConfig[];\n\n  /** Update MCP server configurations */\n  setMcpServers: (mcpServers: MCPServerConfig[]) => void;\n\n  /**\n   * Current suggestions array\n   * Use this to read the current suggestions or in conjunction with setSuggestions for manual control\n   */\n  suggestions: Suggestion[];\n\n  /**\n   * Manually set suggestions\n   * Useful for manual mode or custom suggestion workflows\n   */\n  setSuggestions: (suggestions: Omit<Suggestion, \"isLoading\">[]) => void;\n\n  /**\n   * Trigger AI-powered suggestion generation\n   * Uses configurations from useCopilotChatSuggestions hooks\n   * Respects global debouncing - only one generation can run at a time\n   *\n   * ```tsx\n   * generateSuggestions();\n   * console.log(suggestions); // [suggestion1, suggestion2, suggestion3]\n   * ```\n   */\n  generateSuggestions: () => Promise<void>;\n\n  /**\n   * Clear all current suggestions\n   * Also resets suggestion generation state\n   */\n  resetSuggestions: () => void;\n\n  /** Whether suggestions are currently being generated */\n  isLoadingSuggestions: boolean;\n\n  /** Interrupt content for human-in-the-loop workflows */\n  interrupt: string | React.ReactElement | null;\n\n  agent?: ReturnType<typeof useAgent>[\"agent\"];\n\n  threadId?: string;\n}\n\nexport function useCopilotChatInternal({\n  suggestions,\n  onInProgress,\n  onSubmitMessage,\n  onStopGeneration,\n  onReloadMessages,\n}: UseCopilotChatOptions = {}): UseCopilotChatReturn {\n  const { copilotkit } = useCopilotKit();\n  const { threadId, agentSession } = useCopilotContext();\n  const existingConfig = useCopilotChatConfiguration();\n  const [agentAvailable, setAgentAvailable] = useState(false);\n\n  // Apply priority: props > existing config > defaults\n  const resolvedAgentId = existingConfig?.agentId ?? \"default\";\n  const { agent } = useAgent({ agentId: resolvedAgentId });\n\n  useEffect(() => {\n    const connect = async (agent: AbstractAgent) => {\n      setAgentAvailable(false);\n      try {\n        await copilotkit.connectAgent({ agent });\n        setAgentAvailable(true);\n      } catch (error) {\n        if (error instanceof AGUIConnectNotImplementedError) {\n          // connect not implemented, ignore\n        } else {\n          throw error;\n        }\n      }\n    };\n    if (agent && existingConfig?.threadId && agent.threadId !== existingConfig.threadId) {\n      agent.threadId = existingConfig.threadId;\n      connect(agent);\n    }\n    return () => {};\n  }, [existingConfig?.threadId, agent, copilotkit, resolvedAgentId]);\n\n  useEffect(() => {\n    onInProgress?.(Boolean(agent?.isRunning));\n  }, [agent?.isRunning, onInProgress]);\n\n  const interrupt = useLangGraphInterruptRender(agent);\n\n  const reset = () => {\n    agent?.setMessages([]);\n    agent?.setState(null);\n  };\n\n  const deleteMessage = useCallback(\n    (messageId: string) => {\n      const filteredMessages = (agent?.messages ?? []).filter(\n        (message) => message.id !== messageId,\n      );\n      agent?.setMessages(filteredMessages);\n    },\n    [agent?.setMessages, agent?.messages],\n  );\n\n  const latestDelete = useUpdatedRef(deleteMessage);\n  const latestDeleteFunc = useCallback(\n    (messageId: string) => {\n      return latestDelete.current(messageId);\n    },\n    [latestDelete],\n  );\n\n  const currentSuggestions = useSuggestions({ agentId: resolvedAgentId });\n\n  const reload = useAsyncCallback(\n    async (reloadMessageId: string): Promise<void> => {\n      const messages = agent?.messages ?? [];\n      // TODO: get isLoading\n      const isLoading = false;\n      if (isLoading || messages.length === 0) {\n        return;\n      }\n\n      const reloadMessageIndex = messages.findIndex((msg) => msg.id === reloadMessageId);\n      if (reloadMessageIndex === -1) {\n        console.warn(`Message with id ${reloadMessageId} not found`);\n        return;\n      }\n\n      const reloadMessageRole = messages[reloadMessageIndex].role;\n      if (reloadMessageRole !== \"assistant\") {\n        console.warn(`Regenerate cannot be performed on ${reloadMessageRole} role`);\n        return;\n      }\n      let historyCutoff: Message[] = [messages[0]];\n\n      if (messages.length > 2 && reloadMessageIndex !== 0) {\n        // message to regenerate from is now first.\n        // Work backwards to find the first the closest user message\n        const lastUserMessageBeforeRegenerate = messages\n          .slice(0, reloadMessageIndex)\n          .reverse()\n          .find((msg) => msg.role === \"user\");\n\n        if (!lastUserMessageBeforeRegenerate) {\n          historyCutoff = [messages[0]];\n        } else {\n          const indexOfLastUserMessageBeforeRegenerate = messages.findIndex(\n            (msg) => msg.id === lastUserMessageBeforeRegenerate.id,\n          );\n          // Include the user message, remove everything after it\n          historyCutoff = messages.slice(0, indexOfLastUserMessageBeforeRegenerate + 1);\n        }\n      } else if (messages.length > 2 && reloadMessageIndex === 0) {\n        historyCutoff = [messages[0], messages[1]];\n      }\n\n      agent?.setMessages(historyCutoff);\n\n      if (agent) {\n        copilotkit.runAgent({ agent });\n      }\n      return;\n    },\n    [agent?.setMessages, copilotkit?.runAgent],\n  );\n\n  const latestSendMessageFunc = useAsyncCallback(\n    async (message: Message, options?: AppendMessageOptions) => {\n      if (!agent) return;\n      const followUp = options?.followUp ?? true;\n      if (options?.clearSuggestions) {\n        copilotkit.clearSuggestions(resolvedAgentId);\n      }\n\n      // Call onSubmitMessage BEFORE adding message and running agent\n      // This allows users to perform actions (e.g., open chat window) before agent starts processing\n      if (onSubmitMessage) {\n        const content =\n          typeof message.content === \"string\"\n            ? message.content\n            : message.content && \"text\" in message.content\n              ? message.content.text\n              : message.content && \"filename\" in message.content\n                ? message.content.filename\n                : \"\";\n        try {\n          await onSubmitMessage(content);\n        } catch (error) {\n          console.error(\"Error in onSubmitMessage:\", error);\n        }\n      }\n\n      agent?.addMessage(message);\n      if (followUp) {\n        try {\n          await copilotkit.runAgent({ agent });\n        } catch (error) {\n          console.error(\"CopilotChat: runAgent failed\", error);\n        }\n      }\n    },\n    [agent, copilotkit, resolvedAgentId, onSubmitMessage],\n  );\n\n  const latestAppendFunc = useAsyncCallback(\n    async (message: DeprecatedGqlMessage, options?: AppendMessageOptions) => {\n      return latestSendMessageFunc(gqlToAGUI([message])[0], options);\n    },\n    [latestSendMessageFunc],\n  );\n\n  const latestSetMessagesFunc = useCallback(\n    (messages: Message[] | DeprecatedGqlMessage[]) => {\n      if (messages.every((message) => message instanceof DeprecatedGqlMessage)) {\n        return agent?.setMessages?.(gqlToAGUI(messages));\n      }\n      return agent?.setMessages?.(messages);\n    },\n    [agent?.setMessages, agent],\n  );\n\n  const latestReload = useUpdatedRef(reload);\n  const latestReloadFunc = useAsyncCallback(\n    async (messageId: string) => {\n      onReloadMessages?.({\n        messageId,\n        currentAgentName: agent?.agentId,\n        messages: agent?.messages ?? [],\n      });\n      return await latestReload.current(messageId);\n    },\n    [latestReload, agent, onReloadMessages],\n  );\n\n  const latestStopFunc = useCallback(() => {\n    onStopGeneration?.({\n      currentAgentName: agent?.agentId,\n      messages: agent?.messages ?? [],\n    });\n    return agent?.abortRun?.();\n  }, [onStopGeneration, agent]);\n\n  const latestReset = useUpdatedRef(reset);\n  const latestResetFunc = useCallback(() => {\n    return latestReset.current();\n  }, [latestReset]);\n\n  const lazyToolRendered = useLazyToolRenderer();\n  const renderCustomMessage = useRenderCustomMessages();\n  const legacyCustomMessageRenderer = useLegacyCoagentRenderer({\n    copilotkit,\n    agent,\n    agentId: resolvedAgentId,\n    threadId: existingConfig?.threadId ?? threadId,\n  });\n  const allMessages = agent?.messages ?? [];\n  const resolvedMessages = useMemo(() => {\n    let processedMessages = allMessages.map((message) => {\n      if (message.role !== \"assistant\") {\n        return message;\n      }\n\n      const lazyRendered = lazyToolRendered(message, allMessages);\n      if (lazyRendered) {\n        const renderedGenUi = lazyRendered();\n        if (renderedGenUi) {\n          return { ...message, generativeUI: () => renderedGenUi };\n        }\n      }\n\n      const bridgeRenderer =\n        legacyCustomMessageRenderer || renderCustomMessage\n          ? () => {\n              const customRender = renderCustomMessage?.({\n                message,\n                position: \"before\",\n              });\n              if (customRender) {\n                return customRender;\n              }\n              return legacyCustomMessageRenderer?.({ message, position: \"before\" });\n            }\n          : null;\n\n      if (bridgeRenderer) {\n        return { ...message, generativeUI: bridgeRenderer };\n      }\n      return message;\n    });\n\n    const hasAssistantMessages = processedMessages.some((msg) => msg.role === \"assistant\");\n\n    // TODO: what is this?\n    // if (legacyCustomMessageRenderer && !hasAssistantMessages) {\n    //   const placeholderId = `coagent-state-render-${resolvedAgentId}`;\n    //   const placeholderMessage: Message = {\n    //     id: placeholderId,\n    //     role: \"assistant\",\n    //     content: \"\",\n    //     name: \"coagent-state-render\",\n    //   };\n    //   processedMessages = [\n    //     ...processedMessages,\n    //     {\n    //       ...placeholderMessage,\n    //       generativeUI: () =>\n    //         legacyCustomMessageRenderer({\n    //           message: placeholderMessage,\n    //           position: \"before\",\n    //         }),\n    //     } as Message,\n    //   ];\n    // }\n\n    return processedMessages;\n  }, [\n    agent?.messages,\n    lazyToolRendered,\n    allMessages,\n    renderCustomMessage,\n    // legacyCustomMessageRenderer,\n    resolvedAgentId,\n  ]);\n\n  const renderedSuggestions = useMemo(() => {\n    if (Array.isArray(suggestions)) {\n      return {\n        suggestions: suggestions.map((s) => ({ ...s, isLoading: false })),\n        isLoading: false,\n      };\n    }\n    return currentSuggestions;\n  }, [suggestions, currentSuggestions]);\n\n  // @ts-ignore\n  return {\n    messages: resolvedMessages,\n    sendMessage: latestSendMessageFunc,\n    appendMessage: latestAppendFunc,\n    setMessages: latestSetMessagesFunc,\n    reloadMessages: latestReloadFunc,\n    stopGeneration: latestStopFunc,\n    reset: latestResetFunc,\n    deleteMessage: latestDeleteFunc,\n    isAvailable: agentAvailable,\n    isLoading: Boolean(agent?.isRunning),\n    // mcpServers,\n    // setMcpServers,\n    suggestions: renderedSuggestions.suggestions,\n    setSuggestions: (suggestions: Omit<Suggestion, \"isLoading\">[]) =>\n      copilotkit.addSuggestionsConfig({ suggestions }),\n    generateSuggestions: async () => copilotkit.reloadSuggestions(resolvedAgentId),\n    resetSuggestions: () => copilotkit.clearSuggestions(resolvedAgentId),\n    isLoadingSuggestions: renderedSuggestions.isLoading,\n    interrupt,\n    agent,\n    threadId,\n  };\n}\n\n// store `value` in a ref and update\n// it whenever it changes.\nfunction useUpdatedRef<T>(value: T) {\n  const ref = useRef(value);\n\n  useEffect(() => {\n    ref.current = value;\n  }, [value]);\n\n  return ref;\n}\n\ntype LegacyRenderParams = {\n  message: Message;\n  position: \"before\" | \"after\";\n};\n\ntype LegacyRenderer = ((args: LegacyRenderParams) => any) | null;\n\nfunction useLegacyCoagentRenderer({\n  copilotkit,\n  agent,\n  agentId,\n  threadId,\n}: {\n  copilotkit: ReturnType<typeof useCopilotKit>[\"copilotkit\"];\n  agent?: AbstractAgent;\n  agentId: string;\n  threadId?: string;\n}): LegacyRenderer {\n  return useMemo(() => {\n    if (!copilotkit || !agent) {\n      return null;\n    }\n\n    return ({ message, position }: LegacyRenderParams) => {\n      const effectiveThreadId = threadId ?? agent.threadId ?? \"default\";\n      const existingRunId = copilotkit.getRunIdForMessage(agentId, effectiveThreadId, message.id);\n      const runId = existingRunId || `pending:${message.id}`;\n      const messageIndex = Math.max(\n        agent.messages.findIndex((msg) => msg.id === message.id),\n        0,\n      );\n\n      const bridgeProps: CoAgentStateRenderBridgeProps = {\n        message: message as any,\n        position,\n        runId,\n        messageIndex,\n        messageIndexInRun: 0,\n        numberOfMessagesInRun: 1,\n        agentId,\n        stateSnapshot: (message as any).state,\n      };\n\n      return createElement(CoAgentStateRenderBridge, bridgeProps) as any;\n    };\n  }, [agent, agentId, copilotkit, threadId]);\n}\n\nexport function defaultSystemMessage(\n  contextString: string,\n  additionalInstructions?: string,\n): string {\n  return (\n    `\nPlease act as an efficient, competent, conscientious, and industrious professional assistant.\n\nHelp the user achieve their goals, and you do so in a way that is as efficient as possible, without unnecessary fluff, but also without sacrificing professionalism.\nAlways be polite and respectful, and prefer brevity over verbosity.\n\nThe user has provided you with the following context:\n\\`\\`\\`\n${contextString}\n\\`\\`\\`\n\nThey have also provided you with functions you can call to initiate actions on their behalf, or functions you can call to receive more information.\n\nPlease assist them as best you can.\n\nYou can ask them for clarifying questions if needed, but don't be annoying about it. If you can reasonably 'fill in the blanks' yourself, do so.\n\nIf you would like to call a function, call it without saying anything else.\nIn case of a function error:\n- If this error stems from incorrect function parameters or syntax, you may retry with corrected arguments.\n- If the error's source is unclear or seems unrelated to your input, do not attempt further retries.\n` + (additionalInstructions ? `\\n\\n${additionalInstructions}` : \"\")\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAAA,SAAS,QAAQ,WAAW,aAAa,SAAS,UAAU,qBAAqB;AAKjF,SAAS,WAAW,WAAW,4BAA4B;AAE3D;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAGP,SAAwB,sCAAsC;AA4RvD,SAAS,uBAAuB;AAAA,EACrC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,IAA2B,CAAC,GAAyB;AAlTrD;AAmTE,QAAM,EAAE,WAAW,IAAI,cAAc;AACrC,QAAM,EAAE,UAAU,aAAa,IAAI,kBAAkB;AACrD,QAAM,iBAAiB,4BAA4B;AACnD,QAAM,CAAC,gBAAgB,iBAAiB,IAAI,SAAS,KAAK;AAG1D,QAAM,mBAAkB,sDAAgB,YAAhB,YAA2B;AACnD,QAAM,EAAE,MAAM,IAAI,SAAS,EAAE,SAAS,gBAAgB,CAAC;AAEvD,YAAU,MAAM;AACd,UAAM,UAAU,CAAOA,WAAyB;AAC9C,wBAAkB,KAAK;AACvB,UAAI;AACF,cAAM,WAAW,aAAa,EAAE,OAAAA,OAAM,CAAC;AACvC,0BAAkB,IAAI;AAAA,MACxB,SAAS,OAAP;AACA,YAAI,iBAAiB,gCAAgC;AAAA,QAErD,OAAO;AACL,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IACF;AACA,QAAI,UAAS,iDAAgB,aAAY,MAAM,aAAa,eAAe,UAAU;AACnF,YAAM,WAAW,eAAe;AAChC,cAAQ,KAAK;AAAA,IACf;AACA,WAAO,MAAM;AAAA,IAAC;AAAA,EAChB,GAAG,CAAC,iDAAgB,UAAU,OAAO,YAAY,eAAe,CAAC;AAEjE,YAAU,MAAM;AACd,iDAAe,QAAQ,+BAAO,SAAS;AAAA,EACzC,GAAG,CAAC,+BAAO,WAAW,YAAY,CAAC;AAEnC,QAAM,YAAY,4BAA4B,KAAK;AAEnD,QAAM,QAAQ,MAAM;AAClB,mCAAO,YAAY,CAAC;AACpB,mCAAO,SAAS;AAAA,EAClB;AAEA,QAAM,gBAAgB;AAAA,IACpB,CAAC,cAAsB;AA7V3B,UAAAC;AA8VM,YAAM,qBAAoBA,MAAA,+BAAO,aAAP,OAAAA,MAAmB,CAAC,GAAG;AAAA,QAC/C,CAAC,YAAY,QAAQ,OAAO;AAAA,MAC9B;AACA,qCAAO,YAAY;AAAA,IACrB;AAAA,IACA,CAAC,+BAAO,aAAa,+BAAO,QAAQ;AAAA,EACtC;AAEA,QAAM,eAAe,cAAc,aAAa;AAChD,QAAM,mBAAmB;AAAA,IACvB,CAAC,cAAsB;AACrB,aAAO,aAAa,QAAQ,SAAS;AAAA,IACvC;AAAA,IACA,CAAC,YAAY;AAAA,EACf;AAEA,QAAM,qBAAqB,eAAe,EAAE,SAAS,gBAAgB,CAAC;AAEtE,QAAM,SAAS;AAAA,IACb,CAAO,oBAA2C;AAjXtD,UAAAA;AAkXM,YAAM,YAAWA,MAAA,+BAAO,aAAP,OAAAA,MAAmB,CAAC;AAErC,YAAM,YAAY;AAClB,UAAI,aAAa,SAAS,WAAW,GAAG;AACtC;AAAA,MACF;AAEA,YAAM,qBAAqB,SAAS,UAAU,CAAC,QAAQ,IAAI,OAAO,eAAe;AACjF,UAAI,uBAAuB,IAAI;AAC7B,gBAAQ,KAAK,mBAAmB,2BAA2B;AAC3D;AAAA,MACF;AAEA,YAAM,oBAAoB,SAAS,kBAAkB,EAAE;AACvD,UAAI,sBAAsB,aAAa;AACrC,gBAAQ,KAAK,qCAAqC,wBAAwB;AAC1E;AAAA,MACF;AACA,UAAI,gBAA2B,CAAC,SAAS,CAAC,CAAC;AAE3C,UAAI,SAAS,SAAS,KAAK,uBAAuB,GAAG;AAGnD,cAAM,kCAAkC,SACrC,MAAM,GAAG,kBAAkB,EAC3B,QAAQ,EACR,KAAK,CAAC,QAAQ,IAAI,SAAS,MAAM;AAEpC,YAAI,CAAC,iCAAiC;AACpC,0BAAgB,CAAC,SAAS,CAAC,CAAC;AAAA,QAC9B,OAAO;AACL,gBAAM,yCAAyC,SAAS;AAAA,YACtD,CAAC,QAAQ,IAAI,OAAO,gCAAgC;AAAA,UACtD;AAEA,0BAAgB,SAAS,MAAM,GAAG,yCAAyC,CAAC;AAAA,QAC9E;AAAA,MACF,WAAW,SAAS,SAAS,KAAK,uBAAuB,GAAG;AAC1D,wBAAgB,CAAC,SAAS,CAAC,GAAG,SAAS,CAAC,CAAC;AAAA,MAC3C;AAEA,qCAAO,YAAY;AAEnB,UAAI,OAAO;AACT,mBAAW,SAAS,EAAE,MAAM,CAAC;AAAA,MAC/B;AACA;AAAA,IACF;AAAA,IACA,CAAC,+BAAO,aAAa,yCAAY,QAAQ;AAAA,EAC3C;AAEA,QAAM,wBAAwB;AAAA,IAC5B,CAAO,SAAkB,YAAmC;AAtahE,UAAAA;AAuaM,UAAI,CAAC;AAAO;AACZ,YAAM,YAAWA,MAAA,mCAAS,aAAT,OAAAA,MAAqB;AACtC,UAAI,mCAAS,kBAAkB;AAC7B,mBAAW,iBAAiB,eAAe;AAAA,MAC7C;AAIA,UAAI,iBAAiB;AACnB,cAAM,UACJ,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,QAAQ,WAAW,UAAU,QAAQ,UACnC,QAAQ,QAAQ,OAChB,QAAQ,WAAW,cAAc,QAAQ,UACvC,QAAQ,QAAQ,WAChB;AACV,YAAI;AACF,gBAAM,gBAAgB,OAAO;AAAA,QAC/B,SAAS,OAAP;AACA,kBAAQ,MAAM,6BAA6B,KAAK;AAAA,QAClD;AAAA,MACF;AAEA,qCAAO,WAAW;AAClB,UAAI,UAAU;AACZ,YAAI;AACF,gBAAM,WAAW,SAAS,EAAE,MAAM,CAAC;AAAA,QACrC,SAAS,OAAP;AACA,kBAAQ,MAAM,gCAAgC,KAAK;AAAA,QACrD;AAAA,MACF;AAAA,IACF;AAAA,IACA,CAAC,OAAO,YAAY,iBAAiB,eAAe;AAAA,EACtD;AAEA,QAAM,mBAAmB;AAAA,IACvB,CAAO,SAA+B,YAAmC;AACvE,aAAO,sBAAsB,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG,OAAO;AAAA,IAC/D;AAAA,IACA,CAAC,qBAAqB;AAAA,EACxB;AAEA,QAAM,wBAAwB;AAAA,IAC5B,CAAC,aAAiD;AAndtD,UAAAA,KAAAC;AAodM,UAAI,SAAS,MAAM,CAAC,YAAY,mBAAmB,oBAAoB,GAAG;AACxE,gBAAOD,MAAA,+BAAO,gBAAP,gBAAAA,IAAA,YAAqB,UAAU,QAAQ;AAAA,MAChD;AACA,cAAOC,MAAA,+BAAO,gBAAP,gBAAAA,IAAA,YAAqB;AAAA,IAC9B;AAAA,IACA,CAAC,+BAAO,aAAa,KAAK;AAAA,EAC5B;AAEA,QAAM,eAAe,cAAc,MAAM;AACzC,QAAM,mBAAmB;AAAA,IACvB,CAAO,cAAsB;AA9djC,UAAAD;AA+dM,2DAAmB;AAAA,QACjB;AAAA,QACA,kBAAkB,+BAAO;AAAA,QACzB,WAAUA,MAAA,+BAAO,aAAP,OAAAA,MAAmB,CAAC;AAAA,MAChC;AACA,aAAO,MAAM,aAAa,QAAQ,SAAS;AAAA,IAC7C;AAAA,IACA,CAAC,cAAc,OAAO,gBAAgB;AAAA,EACxC;AAEA,QAAM,iBAAiB,YAAY,MAAM;AAze3C,QAAAA,KAAAC;AA0eI,yDAAmB;AAAA,MACjB,kBAAkB,+BAAO;AAAA,MACzB,WAAUD,MAAA,+BAAO,aAAP,OAAAA,MAAmB,CAAC;AAAA,IAChC;AACA,YAAOC,MAAA,+BAAO,aAAP,gBAAAA,IAAA;AAAA,EACT,GAAG,CAAC,kBAAkB,KAAK,CAAC;AAE5B,QAAM,cAAc,cAAc,KAAK;AACvC,QAAM,kBAAkB,YAAY,MAAM;AACxC,WAAO,YAAY,QAAQ;AAAA,EAC7B,GAAG,CAAC,WAAW,CAAC;AAEhB,QAAM,mBAAmB,oBAAoB;AAC7C,QAAM,sBAAsB,wBAAwB;AACpD,QAAM,8BAA8B,yBAAyB;AAAA,IAC3D;AAAA,IACA;AAAA,IACA,SAAS;AAAA,IACT,WAAU,sDAAgB,aAAhB,YAA4B;AAAA,EACxC,CAAC;AACD,QAAM,eAAc,oCAAO,aAAP,YAAmB,CAAC;AACxC,QAAM,mBAAmB,QAAQ,MAAM;AACrC,QAAI,oBAAoB,YAAY,IAAI,CAAC,YAAY;AACnD,UAAI,QAAQ,SAAS,aAAa;AAChC,eAAO;AAAA,MACT;AAEA,YAAM,eAAe,iBAAiB,SAAS,WAAW;AAC1D,UAAI,cAAc;AAChB,cAAM,gBAAgB,aAAa;AACnC,YAAI,eAAe;AACjB,iBAAO,iCAAK,UAAL,EAAc,cAAc,MAAM,cAAc;AAAA,QACzD;AAAA,MACF;AAEA,YAAM,iBACJ,+BAA+B,sBAC3B,MAAM;AACJ,cAAM,eAAe,2DAAsB;AAAA,UACzC;AAAA,UACA,UAAU;AAAA,QACZ;AACA,YAAI,cAAc;AAChB,iBAAO;AAAA,QACT;AACA,eAAO,2EAA8B,EAAE,SAAS,UAAU,SAAS;AAAA,MACrE,IACA;AAEN,UAAI,gBAAgB;AAClB,eAAO,iCAAK,UAAL,EAAc,cAAc,eAAe;AAAA,MACpD;AACA,aAAO;AAAA,IACT,CAAC;AAED,UAAM,uBAAuB,kBAAkB,KAAK,CAAC,QAAQ,IAAI,SAAS,WAAW;AAwBrF,WAAO;AAAA,EACT,GAAG;AAAA,IACD,+BAAO;AAAA,IACP;AAAA,IACA;AAAA,IACA;AAAA;AAAA,IAEA;AAAA,EACF,CAAC;AAED,QAAM,sBAAsB,QAAQ,MAAM;AACxC,QAAI,MAAM,QAAQ,WAAW,GAAG;AAC9B,aAAO;AAAA,QACL,aAAa,YAAY,IAAI,CAAC,MAAO,iCAAK,IAAL,EAAQ,WAAW,MAAM,EAAE;AAAA,QAChE,WAAW;AAAA,MACb;AAAA,IACF;AACA,WAAO;AAAA,EACT,GAAG,CAAC,aAAa,kBAAkB,CAAC;AAGpC,SAAO;AAAA,IACL,UAAU;AAAA,IACV,aAAa;AAAA,IACb,eAAe;AAAA,IACf,aAAa;AAAA,IACb,gBAAgB;AAAA,IAChB,gBAAgB;AAAA,IAChB,OAAO;AAAA,IACP,eAAe;AAAA,IACf,aAAa;AAAA,IACb,WAAW,QAAQ,+BAAO,SAAS;AAAA;AAAA;AAAA,IAGnC,aAAa,oBAAoB;AAAA,IACjC,gBAAgB,CAACC,iBACf,WAAW,qBAAqB,EAAE,aAAAA,aAAY,CAAC;AAAA,IACjD,qBAAqB,MAAS;AAAG,wBAAW,kBAAkB,eAAe;AAAA;AAAA,IAC7E,kBAAkB,MAAM,WAAW,iBAAiB,eAAe;AAAA,IACnE,sBAAsB,oBAAoB;AAAA,IAC1C;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAIA,SAAS,cAAiB,OAAU;AAClC,QAAM,MAAM,OAAO,KAAK;AAExB,YAAU,MAAM;AACd,QAAI,UAAU;AAAA,EAChB,GAAG,CAAC,KAAK,CAAC;AAEV,SAAO;AACT;AASA,SAAS,yBAAyB;AAAA,EAChC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAKmB;AACjB,SAAO,QAAQ,MAAM;AACnB,QAAI,CAAC,cAAc,CAAC,OAAO;AACzB,aAAO;AAAA,IACT;AAEA,WAAO,CAAC,EAAE,SAAS,SAAS,MAA0B;AA1oB1D;AA2oBM,YAAM,qBAAoB,mCAAY,MAAM,aAAlB,YAA8B;AACxD,YAAM,gBAAgB,WAAW,mBAAmB,SAAS,mBAAmB,QAAQ,EAAE;AAC1F,YAAM,QAAQ,iBAAiB,WAAW,QAAQ;AAClD,YAAM,eAAe,KAAK;AAAA,QACxB,MAAM,SAAS,UAAU,CAAC,QAAQ,IAAI,OAAO,QAAQ,EAAE;AAAA,QACvD;AAAA,MACF;AAEA,YAAM,cAA6C;AAAA,QACjD;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,mBAAmB;AAAA,QACnB,uBAAuB;AAAA,QACvB;AAAA,QACA,eAAgB,QAAgB;AAAA,MAClC;AAEA,aAAO,cAAc,0BAA0B,WAAW;AAAA,IAC5D;AAAA,EACF,GAAG,CAAC,OAAO,SAAS,YAAY,QAAQ,CAAC;AAC3C;AAEO,SAAS,qBACd,eACA,wBACQ;AACR,SACE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAaG,yBAAyB;AAAA;AAAA,EAAO,2BAA2B;AAEhE;","names":["agent","_a","_b","suggestions"]}

package/dist/{chunk-QNUAXSDP.mjs → chunk-3X3I7OJV.mjs}

@@ -36,12 +36,16 @@ function useCoagentStateRenderBridge(agentId, props) {
   const { coAgentStateRenders, claimsRef } = useCoAgentStateRenders();
   const { agent } = useAgent({ agentId });
   const [nodeName, setNodeName] = useState(void 0);
+  const [, forceUpdate] = useState(0);
   const runId = (_a = props.runId) != null ? _a : message.runId;
   const effectiveRunId = runId || "pending";
   useEffect(() => {
     if (!agent)
       return;
     const subscriber = {
+      onStateChanged: () => {
+        forceUpdate((value) => value + 1);
+      },
       onStepStartedEvent: ({ event }) => {
         if (event.stepName !== nodeName) {
           setNodeName(event.stepName);
@@ -119,12 +123,14 @@ function useCoagentStateRenderBridge(agentId, props) {
     if (!canRender) {
       return null;
     }
-    if (snapshot && !claimsRef.current[message.id].locked) {
-      if (stateSnapshot) {
-        claimsRef.current[message.id].stateSnapshot = snapshot;
-        claimsRef.current[message.id].locked = true;
-      } else {
+    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) {
@@ -163,4 +169,4 @@ export {
   useCoagentStateRenderBridge,
   CoAgentStateRenderBridge
 };
-//# sourceMappingURL=chunk-QNUAXSDP.mjs.map
+//# sourceMappingURL=chunk-3X3I7OJV.mjs.map

package/dist/chunk-3X3I7OJV.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/use-coagent-state-render-bridge.tsx"],"sourcesContent":["import { ReactCustomMessageRendererPosition, useAgent } from \"@copilotkitnext/react\";\nimport { useCallback, useEffect, useMemo, useState } from \"react\";\nimport type { AgentSubscriber } from \"@ag-ui/client\";\nimport { useCoAgentStateRenders } from \"../context\";\nimport { dataToUUID, parseJson } from \"@copilotkit/shared\";\n\nfunction getStateWithoutConstantKeys(state: any) {\n  if (!state) return {};\n  const { messages, tools, copilotkit, ...stateWithoutConstantKeys } = state;\n  return stateWithoutConstantKeys;\n}\n\n// Function that compares states, without the constant keys\nfunction areStatesEquals(a: any, b: any) {\n  if ((a && !b) || (!a && b)) return false;\n  const { messages, tools, copilotkit, ...aWithoutConstantKeys } = a;\n  const {\n    messages: bMessages,\n    tools: bTools,\n    copilotkit: bCopilotkit,\n    ...bWithoutConstantKeys\n  } = b;\n\n  return JSON.stringify(aWithoutConstantKeys) === JSON.stringify(bWithoutConstantKeys);\n}\n\n/**\n * Bridge hook that connects agent state renders to chat messages.\n *\n * ## Purpose\n * This hook finds matching state render configurations (registered via useCoAgentStateRender)\n * and returns UI to render in chat.\n * It ensures each state render appears bound to a specific message, preventing duplicates while\n * allowing re-binding when the underlying state changes significantly.\n *\n * ## Message-ID-Based Claiming System\n *\n * ### The Problem\n * Multiple bridge component instances render simultaneously (one per message). Without coordination,\n * they would all try to render the same state render, causing duplicates.\n *\n * ### The Solution: Message-ID Claims with State Comparison\n * Each state render is \"claimed\" by exactly one **message ID** (not runId):\n *\n * **Claim Structure**: `claimsRef.current[messageId] = { stateRenderId, runId, stateSnapshot, locked }`\n *\n * **Primary binding is by messageId because**:\n * - runId is not always available immediately (starts as \"pending\")\n * - messageId is the stable identifier throughout the message lifecycle\n * - Claims persist across component remounts via context ref\n *\n * ### Claiming Logic Flow\n *\n * 1. **Message already has a claim**:\n *    - Check if the claim matches the current stateRenderId\n *    - If yes → render (this message owns this render)\n *    - Update runId if it was \"pending\" and now available\n *\n * 2. **State render claimed by another message**:\n *    - Compare state snapshots (ignoring constant keys: messages, tools, copilotkit)\n *    - If states are identical → block rendering (duplicate)\n *    - **If states are different → allow claiming** (new data, new message)\n *    - This handles cases where the same render type shows different states in different messages\n *\n * 3. **Unclaimed state render**:\n *    - Only allow claiming if runId is \"pending\" (initial render)\n *    - If runId is real but no claim exists → block (edge case protection)\n *    - Create new claim: `claimsRef.current[messageId] = { stateRenderId, runId }`\n *\n * ### State Snapshot Locking\n *\n * Once a state snapshot is captured and locked for a message:\n * - The UI always renders with the locked snapshot (not live agent.state)\n * - Prevents UI from appearing \"wiped\" during state transitions\n * - Locked when: stateSnapshot prop is available (from message persistence)\n * - Unlocked state: can still update from live agent.state\n *\n * ### Synchronous Claiming (Ref-based)\n *\n * Claims are stored in a context-level ref (not React state):\n * - Multiple bridges render in the same tick\n * - State updates are async - would allow duplicates before update completes\n * - Ref provides immediate, synchronous claim checking\n * - Survives component remounts (stored in context, not component)\n *\n * ## Flow Example\n *\n * ```\n * Time 1: Message A renders, runId=undefined, state={progress: 50%}\n *   → effectiveRunId = \"pending\"\n *   → Claims: claimsRef[\"msgA\"] = { stateRenderId: \"tasks\", runId: \"pending\", stateSnapshot: {progress: 50%} }\n *   → Renders UI with 50% progress\n *\n * Time 2: Message B renders, runId=undefined, same state\n *   → Checks: \"tasks\" already claimed by msgA with same state\n *   → Returns null (blocked - duplicate)\n *\n * Time 3: Real runId appears (e.g., \"run-123\")\n *   → Updates claim: claimsRef[\"msgA\"].runId = \"run-123\"\n *   → Message A continues rendering\n *\n * Time 4: Agent processes more, state={progress: 100%}\n *   → Message A: locked to 50% (stateSnapshot locked)\n *   → Message C renders with state={progress: 100%}\n *   → Checks: \"tasks\" claimed by msgA but state is DIFFERENT (50% vs 100%)\n *   → Allows new claim: claimsRef[\"msgC\"] = { stateRenderId: \"tasks\", runId: \"run-123\", stateSnapshot: {progress: 100%} }\n *   → Both messages render independently with their own snapshots\n * ```\n */\nexport interface CoAgentStateRenderBridgeProps {\n  message: any;\n  position: ReactCustomMessageRendererPosition;\n  runId: string;\n  messageIndex: number;\n  messageIndexInRun: number;\n  numberOfMessagesInRun: number;\n  agentId: string;\n  stateSnapshot: any;\n}\n\nexport function useCoagentStateRenderBridge(agentId: string, props: CoAgentStateRenderBridgeProps) {\n  const { stateSnapshot, messageIndexInRun, message } = props;\n  const { coAgentStateRenders, claimsRef } = useCoAgentStateRenders();\n  const { agent } = useAgent({ agentId });\n  const [nodeName, setNodeName] = useState<string | undefined>(undefined);\n  const [, forceUpdate] = useState(0);\n\n  const runId = props.runId ?? message.runId;\n  const effectiveRunId = runId || \"pending\";\n\n  useEffect(() => {\n    if (!agent) return;\n    const subscriber: AgentSubscriber = {\n      onStateChanged: () => {\n        forceUpdate((value) => value + 1);\n      },\n      onStepStartedEvent: ({ event }) => {\n        if (event.stepName !== nodeName) {\n          setNodeName(event.stepName);\n        }\n      },\n      onStepFinishedEvent: ({ event }) => {\n        if (event.stepName === nodeName) {\n          setNodeName(undefined);\n        }\n      },\n    };\n\n    const { unsubscribe } = agent.subscribe(subscriber);\n    return () => {\n      unsubscribe();\n    };\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [agentId, nodeName]);\n\n  const getStateRender = useCallback(\n    (messageId: string) => {\n      return Object.entries(coAgentStateRenders).find(([stateRenderId, stateRender]) => {\n        if (claimsRef.current[messageId]) {\n          return stateRenderId === claimsRef.current[messageId].stateRenderId;\n        }\n        const matchingAgentName = stateRender.name === agentId;\n        const matchesNodeContext = stateRender.nodeName ? stateRender.nodeName === nodeName : true;\n        return matchingAgentName && matchesNodeContext;\n      });\n    },\n    [coAgentStateRenders, nodeName, agentId],\n  );\n\n  // Message ID-based claim system - A state render can only be claimed by one message ID\n  const handleRenderRequest = ({\n    stateRenderId,\n    messageId,\n    runId,\n    stateSnapshot: renderSnapshot,\n  }: {\n    stateRenderId: string;\n    messageId: string;\n    runId?: string;\n    stateSnapshot?: any;\n  }): boolean => {\n    // Check if this message has already claimed this state render\n    if (claimsRef.current[messageId]) {\n      const canRender = claimsRef.current[messageId].stateRenderId === stateRenderId;\n\n      // Update runId if it doesn't exist\n      if (\n        canRender &&\n        runId &&\n        (!claimsRef.current[messageId].runId || claimsRef.current[messageId].runId === \"pending\")\n      ) {\n        claimsRef.current[messageId].runId = runId;\n      }\n\n      return canRender;\n    }\n\n    // Do not allow render if any other message has claimed this state render\n    const renderClaimedByOtherMessage = Object.values(claimsRef.current).find(\n      (c) =>\n        c.stateRenderId === stateRenderId &&\n        dataToUUID(getStateWithoutConstantKeys(c.stateSnapshot)) ===\n          dataToUUID(getStateWithoutConstantKeys(renderSnapshot)),\n    );\n    if (renderClaimedByOtherMessage) {\n      // If:\n      //   - state render already claimed\n      //   - snapshot exists in the claiming object and is different from current,\n      if (\n        renderSnapshot &&\n        renderClaimedByOtherMessage.stateSnapshot &&\n        !areStatesEquals(renderClaimedByOtherMessage.stateSnapshot, renderSnapshot)\n      ) {\n        claimsRef.current[messageId] = { stateRenderId, runId };\n        return true;\n      }\n      return false;\n    }\n\n    // No existing claim anywhere yet – allow this message to claim even if we already know the runId.\n    if (!runId) {\n      return false;\n    }\n\n    claimsRef.current[messageId] = { stateRenderId, runId };\n    return true;\n  };\n\n  return useMemo(() => {\n    if (messageIndexInRun !== 0) {\n      return null;\n    }\n\n    const [stateRenderId, stateRender] = getStateRender(message.id) ?? [];\n\n    if (!stateRender || !stateRenderId) {\n      return null;\n    }\n\n    // Is there any state we can use?\n    const snapshot = stateSnapshot ? parseJson(stateSnapshot, stateSnapshot) : agent?.state;\n\n    // Synchronously check/claim - returns true if this message can render\n    const canRender = handleRenderRequest({\n      stateRenderId,\n      messageId: message.id,\n      runId: effectiveRunId,\n      stateSnapshot: snapshot,\n    });\n    if (!canRender) {\n      return null;\n    }\n\n    // If we found state, and given that now there's a claim for the current message, let's save it in the claim\n    if (snapshot) {\n      const existingSnapshot = claimsRef.current[message.id].stateSnapshot;\n      const snapshotChanged =\n        stateSnapshot &&\n        existingSnapshot !== undefined &&\n        !areStatesEquals(existingSnapshot, snapshot);\n\n      if (!claimsRef.current[message.id].locked || snapshotChanged) {\n        claimsRef.current[message.id].stateSnapshot = snapshot;\n        if (stateSnapshot) {\n          claimsRef.current[message.id].locked = true;\n        }\n      }\n    }\n\n    if (stateRender.handler) {\n      stateRender.handler({\n        state: stateSnapshot ? parseJson(stateSnapshot, stateSnapshot) : (agent?.state ?? {}),\n        nodeName: nodeName ?? \"\",\n      });\n    }\n\n    if (stateRender.render) {\n      const status = agent?.isRunning ? \"inProgress\" : \"complete\";\n\n      if (typeof stateRender.render === \"string\") return stateRender.render;\n\n      return stateRender.render({\n        status,\n        // Always use state from claim, to make sure the state does not seem \"wiped\" for a fraction of a second\n        state: claimsRef.current[message.id].stateSnapshot ?? {},\n        nodeName: nodeName ?? \"\",\n      });\n    }\n  }, [\n    getStateRender,\n    stateSnapshot,\n    agent?.state,\n    agent?.isRunning,\n    nodeName,\n    effectiveRunId,\n    message.id,\n    messageIndexInRun,\n  ]);\n}\n\nexport function CoAgentStateRenderBridge(props: CoAgentStateRenderBridgeProps) {\n  return useCoagentStateRenderBridge(props.agentId, props);\n}\n"],"mappings":";;;;;;;;AAAA,SAA6C,gBAAgB;AAC7D,SAAS,aAAa,WAAW,SAAS,gBAAgB;AAG1D,SAAS,YAAY,iBAAiB;AAEtC,SAAS,4BAA4B,OAAY;AAC/C,MAAI,CAAC;AAAO,WAAO,CAAC;AACpB,QAAqE,YAA7D,YAAU,OAAO,WAR3B,IAQuE,IAA7B,qCAA6B,IAA7B,CAAhC,YAAU,SAAO;AACzB,SAAO;AACT;AAGA,SAAS,gBAAgB,GAAQ,GAAQ;AACvC,MAAK,KAAK,CAAC,KAAO,CAAC,KAAK;AAAI,WAAO;AACnC,QAAiE,QAAzD,YAAU,OAAO,WAf3B,IAemE,IAAzB,iCAAyB,IAAzB,CAAhC,YAAU,SAAO;AACzB,QAKI,QAJF;AAAA,cAAU;AAAA,IACV,OAAO;AAAA,IACP,YAAY;AAAA,EAnBhB,IAqBM,IADC,iCACD,IADC;AAAA,IAHH;AAAA,IACA;AAAA,IACA;AAAA;AAIF,SAAO,KAAK,UAAU,oBAAoB,MAAM,KAAK,UAAU,oBAAoB;AACrF;AAgGO,SAAS,4BAA4B,SAAiB,OAAsC;AAxHnG;AAyHE,QAAM,EAAE,eAAe,mBAAmB,QAAQ,IAAI;AACtD,QAAM,EAAE,qBAAqB,UAAU,IAAI,uBAAuB;AAClE,QAAM,EAAE,MAAM,IAAI,SAAS,EAAE,QAAQ,CAAC;AACtC,QAAM,CAAC,UAAU,WAAW,IAAI,SAA6B,MAAS;AACtE,QAAM,CAAC,EAAE,WAAW,IAAI,SAAS,CAAC;AAElC,QAAM,SAAQ,WAAM,UAAN,YAAe,QAAQ;AACrC,QAAM,iBAAiB,SAAS;AAEhC,YAAU,MAAM;AACd,QAAI,CAAC;AAAO;AACZ,UAAM,aAA8B;AAAA,MAClC,gBAAgB,MAAM;AACpB,oBAAY,CAAC,UAAU,QAAQ,CAAC;AAAA,MAClC;AAAA,MACA,oBAAoB,CAAC,EAAE,MAAM,MAAM;AACjC,YAAI,MAAM,aAAa,UAAU;AAC/B,sBAAY,MAAM,QAAQ;AAAA,QAC5B;AAAA,MACF;AAAA,MACA,qBAAqB,CAAC,EAAE,MAAM,MAAM;AAClC,YAAI,MAAM,aAAa,UAAU;AAC/B,sBAAY,MAAS;AAAA,QACvB;AAAA,MACF;AAAA,IACF;AAEA,UAAM,EAAE,YAAY,IAAI,MAAM,UAAU,UAAU;AAClD,WAAO,MAAM;AACX,kBAAY;AAAA,IACd;AAAA,EAEF,GAAG,CAAC,SAAS,QAAQ,CAAC;AAEtB,QAAM,iBAAiB;AAAA,IACrB,CAAC,cAAsB;AACrB,aAAO,OAAO,QAAQ,mBAAmB,EAAE,KAAK,CAAC,CAAC,eAAe,WAAW,MAAM;AAChF,YAAI,UAAU,QAAQ,SAAS,GAAG;AAChC,iBAAO,kBAAkB,UAAU,QAAQ,SAAS,EAAE;AAAA,QACxD;AACA,cAAM,oBAAoB,YAAY,SAAS;AAC/C,cAAM,qBAAqB,YAAY,WAAW,YAAY,aAAa,WAAW;AACtF,eAAO,qBAAqB;AAAA,MAC9B,CAAC;AAAA,IACH;AAAA,IACA,CAAC,qBAAqB,UAAU,OAAO;AAAA,EACzC;AAGA,QAAM,sBAAsB,CAAC;AAAA,IAC3B;AAAA,IACA;AAAA,IACA,OAAAA;AAAA,IACA,eAAe;AAAA,EACjB,MAKe;AAEb,QAAI,UAAU,QAAQ,SAAS,GAAG;AAChC,YAAM,YAAY,UAAU,QAAQ,SAAS,EAAE,kBAAkB;AAGjE,UACE,aACAA,WACC,CAAC,UAAU,QAAQ,SAAS,EAAE,SAAS,UAAU,QAAQ,SAAS,EAAE,UAAU,YAC/E;AACA,kBAAU,QAAQ,SAAS,EAAE,QAAQA;AAAA,MACvC;AAEA,aAAO;AAAA,IACT;AAGA,UAAM,8BAA8B,OAAO,OAAO,UAAU,OAAO,EAAE;AAAA,MACnE,CAAC,MACC,EAAE,kBAAkB,iBACpB,WAAW,4BAA4B,EAAE,aAAa,CAAC,MACrD,WAAW,4BAA4B,cAAc,CAAC;AAAA,IAC5D;AACA,QAAI,6BAA6B;AAI/B,UACE,kBACA,4BAA4B,iBAC5B,CAAC,gBAAgB,4BAA4B,eAAe,cAAc,GAC1E;AACA,kBAAU,QAAQ,SAAS,IAAI,EAAE,eAAe,OAAAA,OAAM;AACtD,eAAO;AAAA,MACT;AACA,aAAO;AAAA,IACT;AAGA,QAAI,CAACA,QAAO;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,SAAS,IAAI,EAAE,eAAe,OAAAA,OAAM;AACtD,WAAO;AAAA,EACT;AAEA,SAAO,QAAQ,MAAM;AApOvB,QAAAC,KAAA;AAqOI,QAAI,sBAAsB,GAAG;AAC3B,aAAO;AAAA,IACT;AAEA,UAAM,CAAC,eAAe,WAAW,KAAIA,MAAA,eAAe,QAAQ,EAAE,MAAzB,OAAAA,MAA8B,CAAC;AAEpE,QAAI,CAAC,eAAe,CAAC,eAAe;AAClC,aAAO;AAAA,IACT;AAGA,UAAM,WAAW,gBAAgB,UAAU,eAAe,aAAa,IAAI,+BAAO;AAGlF,UAAM,YAAY,oBAAoB;AAAA,MACpC;AAAA,MACA,WAAW,QAAQ;AAAA,MACnB,OAAO;AAAA,MACP,eAAe;AAAA,IACjB,CAAC;AACD,QAAI,CAAC,WAAW;AACd,aAAO;AAAA,IACT;AAGA,QAAI,UAAU;AACZ,YAAM,mBAAmB,UAAU,QAAQ,QAAQ,EAAE,EAAE;AACvD,YAAM,kBACJ,iBACA,qBAAqB,UACrB,CAAC,gBAAgB,kBAAkB,QAAQ;AAE7C,UAAI,CAAC,UAAU,QAAQ,QAAQ,EAAE,EAAE,UAAU,iBAAiB;AAC5D,kBAAU,QAAQ,QAAQ,EAAE,EAAE,gBAAgB;AAC9C,YAAI,eAAe;AACjB,oBAAU,QAAQ,QAAQ,EAAE,EAAE,SAAS;AAAA,QACzC;AAAA,MACF;AAAA,IACF;AAEA,QAAI,YAAY,SAAS;AACvB,kBAAY,QAAQ;AAAA,QAClB,OAAO,gBAAgB,UAAU,eAAe,aAAa,KAAK,oCAAO,UAAP,YAAgB,CAAC;AAAA,QACnF,UAAU,8BAAY;AAAA,MACxB,CAAC;AAAA,IACH;AAEA,QAAI,YAAY,QAAQ;AACtB,YAAM,UAAS,+BAAO,aAAY,eAAe;AAEjD,UAAI,OAAO,YAAY,WAAW;AAAU,eAAO,YAAY;AAE/D,aAAO,YAAY,OAAO;AAAA,QACxB;AAAA;AAAA,QAEA,QAAO,eAAU,QAAQ,QAAQ,EAAE,EAAE,kBAA9B,YAA+C,CAAC;AAAA,QACvD,UAAU,8BAAY;AAAA,MACxB,CAAC;AAAA,IACH;AAAA,EACF,GAAG;AAAA,IACD;AAAA,IACA;AAAA,IACA,+BAAO;AAAA,IACP,+BAAO;AAAA,IACP;AAAA,IACA;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,EACF,CAAC;AACH;AAEO,SAAS,yBAAyB,OAAsC;AAC7E,SAAO,4BAA4B,MAAM,SAAS,KAAK;AACzD;","names":["runId","_a"]}

package/dist/{chunk-M7U7RZLU.mjs → chunk-HZWTHNXL.mjs}

@@ -1,6 +1,6 @@
 import {
   useCopilotChatInternal
-} from "./chunk-OBXQSB7S.mjs";
+} from "./chunk-3HJ2O2MN.mjs";
 
 // src/hooks/use-copilot-chat.ts
 function useCopilotChat(options = {}) {
@@ -33,4 +33,4 @@ function useCopilotChat(options = {}) {
 export {
   useCopilotChat
 };
-//# sourceMappingURL=chunk-M7U7RZLU.mjs.map
+//# sourceMappingURL=chunk-HZWTHNXL.mjs.map

package/dist/{chunk-4MLKBZGB.mjs → chunk-K2GJI3PT.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-4CEQJ2X6.mjs";
 import {
   defaultCopilotContextCategories
-} from "./chunk-2VA5EV2P.mjs";
+} from "./chunk-THCOOWDH.mjs";
 import {
   __async,
   __spreadValues
@@ -117,4 +117,4 @@ ${instructions}
 export {
   CopilotTask
 };
-//# sourceMappingURL=chunk-4MLKBZGB.mjs.map
+//# sourceMappingURL=chunk-K2GJI3PT.mjs.map

package/dist/{chunk-W5BKJN46.mjs → chunk-KINHFG63.mjs}

@@ -1,6 +1,6 @@
 import {
   useCopilotChatInternal
-} from "./chunk-OBXQSB7S.mjs";
+} from "./chunk-3HJ2O2MN.mjs";
 import {
   useCopilotContext
 } from "./chunk-AFNWX62Q.mjs";
@@ -83,4 +83,4 @@ function useCopilotChatHeadless_c(options = {}) {
 export {
   useCopilotChatHeadless_c
 };
-//# sourceMappingURL=chunk-W5BKJN46.mjs.map
+//# sourceMappingURL=chunk-KINHFG63.mjs.map

package/dist/{chunk-F555TVE4.mjs → chunk-LMHB2D4J.mjs}

@@ -4,8 +4,8 @@ import { randomUUID } from "@copilotkit/shared";
 import { jsx } from "react/jsx-runtime";
 var ThreadsContext = createContext(void 0);
 function ThreadsProvider({ children, threadId: explicitThreadId }) {
-  const [internalThreadId, setThreadId] = useState(explicitThreadId != null ? explicitThreadId : randomUUID());
-  const threadId = internalThreadId;
+  const [internalThreadId, setThreadId] = useState(() => randomUUID());
+  const threadId = explicitThreadId != null ? explicitThreadId : internalThreadId;
   return /* @__PURE__ */ jsx(
     ThreadsContext.Provider,
     {
@@ -30,4 +30,4 @@ export {
   ThreadsProvider,
   useThreads
 };
-//# sourceMappingURL=chunk-F555TVE4.mjs.map
+//# sourceMappingURL=chunk-LMHB2D4J.mjs.map

package/dist/chunk-LMHB2D4J.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/context/threads-context.tsx"],"sourcesContent":["import React, { createContext, useContext, useState, ReactNode, SetStateAction } from \"react\";\nimport { randomUUID } from \"@copilotkit/shared\";\n\nexport interface ThreadsContextValue {\n  threadId: string;\n  setThreadId: (value: SetStateAction<string>) => void;\n}\n\nconst ThreadsContext = createContext<ThreadsContextValue | undefined>(undefined);\n\nexport interface ThreadsProviderProps {\n  children: ReactNode;\n  threadId?: string;\n}\n\nexport function ThreadsProvider({ children, threadId: explicitThreadId }: ThreadsProviderProps) {\n  const [internalThreadId, setThreadId] = useState<string>(() => randomUUID());\n\n  const threadId = explicitThreadId ?? internalThreadId;\n\n  return (\n    <ThreadsContext.Provider\n      value={{\n        threadId,\n        setThreadId,\n      }}\n    >\n      {children}\n    </ThreadsContext.Provider>\n  );\n}\n\nexport function useThreads() {\n  const context = useContext(ThreadsContext);\n  if (!context) {\n    throw new Error(\"useThreads must be used within ThreadsProvider\");\n  }\n  return context;\n}\n\nexport { ThreadsContext };\n"],"mappings":";AAAA,SAAgB,eAAe,YAAY,gBAA2C;AACtF,SAAS,kBAAkB;AAoBvB;AAbJ,IAAM,iBAAiB,cAA+C,MAAS;AAOxE,SAAS,gBAAgB,EAAE,UAAU,UAAU,iBAAiB,GAAyB;AAC9F,QAAM,CAAC,kBAAkB,WAAW,IAAI,SAAiB,MAAM,WAAW,CAAC;AAE3E,QAAM,WAAW,8CAAoB;AAErC,SACE;AAAA,IAAC,eAAe;AAAA,IAAf;AAAA,MACC,OAAO;AAAA,QACL;AAAA,QACA;AAAA,MACF;AAAA,MAEC;AAAA;AAAA,EACH;AAEJ;AAEO,SAAS,aAAa;AAC3B,QAAM,UAAU,WAAW,cAAc;AACzC,MAAI,CAAC,SAAS;AACZ,UAAM,IAAI,MAAM,gDAAgD;AAAA,EAClE;AACA,SAAO;AACT;","names":[]}

package/dist/{chunk-2VA5EV2P.mjs → chunk-THCOOWDH.mjs}

@@ -16,11 +16,11 @@ import {
 } from "./chunk-YYN33GSG.mjs";
 import {
   CoAgentStateRenderBridge
-} from "./chunk-QNUAXSDP.mjs";
+} from "./chunk-3X3I7OJV.mjs";
 import {
   ThreadsProvider,
   useThreads
-} from "./chunk-F555TVE4.mjs";
+} from "./chunk-LMHB2D4J.mjs";
 import {
   CopilotListeners
 } from "./chunk-WF65O6HX.mjs";
@@ -548,4 +548,4 @@ export {
   CopilotKitInternal,
   defaultCopilotContextCategories
 };
-//# sourceMappingURL=chunk-2VA5EV2P.mjs.map
+//# sourceMappingURL=chunk-THCOOWDH.mjs.map

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

@@ -1236,8 +1236,8 @@ var import_shared8 = require("@copilotkit/shared");
 var import_jsx_runtime6 = require("react/jsx-runtime");
 var ThreadsContext = (0, import_react9.createContext)(void 0);
 function ThreadsProvider({ children, threadId: explicitThreadId }) {
-  const [internalThreadId, setThreadId] = (0, import_react9.useState)(explicitThreadId != null ? explicitThreadId : (0, import_shared8.randomUUID)());
-  const threadId = internalThreadId;
+  const [internalThreadId, setThreadId] = (0, import_react9.useState)(() => (0, import_shared8.randomUUID)());
+  const threadId = explicitThreadId != null ? explicitThreadId : internalThreadId;
   return /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
     ThreadsContext.Provider,
     {
@@ -1286,12 +1286,16 @@ function useCoagentStateRenderBridge(agentId, 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)(() => {
     if (!agent)
       return;
     const subscriber = {
+      onStateChanged: () => {
+        forceUpdate((value) => value + 1);
+      },
       onStepStartedEvent: ({ event }) => {
         if (event.stepName !== nodeName) {
           setNodeName(event.stepName);
@@ -1369,12 +1373,14 @@ function useCoagentStateRenderBridge(agentId, props) {
     if (!canRender) {
       return null;
     }
-    if (snapshot && !claimsRef.current[message.id].locked) {
-      if (stateSnapshot) {
-        claimsRef.current[message.id].stateSnapshot = snapshot;
-        claimsRef.current[message.id].locked = true;
-      } else {
+    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) {