@liveblocks/react-ui 2.25.0-aiprivatebeta7 → 2.25.0-aiprivatebeta8

package/dist/components/internal/AiChatAssistantMessage.js

@@ -1,67 +1,32 @@
-import { jsxs, jsx, Fragment } from 'react/jsx-runtime';
+import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { kInternal } from '@liveblocks/core';
 import { useClient } from '@liveblocks/react';
 import { useSignal } from '@liveblocks/react/_private';
 import { Lexer } from 'marked';
-import { memo, forwardRef, useState, useEffect, useMemo } from 'react';
-import { Button } from './Button.js';
-import { Tooltip } from './Tooltip.js';
+import { memo, forwardRef, useMemo, useState, useRef, useEffect, useCallback } from 'react';
+import '../../_private/index.js';
+import { useComponents } from '../../components.js';
 import { CheckIcon } from '../../icons/Check.js';
-import { ChevronDownIcon } from '../../icons/ChevronDown.js';
 import { ChevronRightIcon } from '../../icons/ChevronRight.js';
 import { CopyIcon } from '../../icons/Copy.js';
-import { RetryIcon } from '../../icons/Retry.js';
 import { WarningIcon } from '../../icons/Warning.js';
 import { useOverrides } from '../../overrides.js';
 import { Root, Trigger, Content } from '../../primitives/internal/Collapsible/index.js';
 import { BlockTokenComp } from '../../primitives/internal/Markdown.js';
 import { classNames } from '../../utils/class-names.js';
-import { TooltipProvider } from '@radix-ui/react-tooltip';
+import { Button } from './Button.js';
 
 const AiChatAssistantMessage = memo(
   forwardRef(
-    ({
-      message,
-      showActions = false,
-      showRegenerate = false,
-      copilotId,
-      className,
-      overrides,
-      ...props
-    }, forwardedRef) => {
+    ({ message, className, overrides, components, ...props }, forwardedRef) => {
       const $ = useOverrides(overrides);
       let children = null;
-      function MessageActions({ text }) {
-        if (!showActions)
-          return null;
-        return /* @__PURE__ */ jsxs("div", {
-          className: "lb-ai-chat-message-actions",
-          children: [
-            /* @__PURE__ */ jsx(Tooltip, {
-              content: $.AI_CHAT_MESSAGE_COPY,
-              children: /* @__PURE__ */ jsx(CopyTextButton, {
-                text,
-                label: $.AI_CHAT_MESSAGE_COPY
-              })
-            }),
-            showRegenerate && /* @__PURE__ */ jsx(Tooltip, {
-              content: $.AI_CHAT_MESSAGE_TRY_AGAIN,
-              children: /* @__PURE__ */ jsx(RegenerateMessageButton, {
-                chatId: message.chatId,
-                messageId: message.id,
-                copilotId,
-                label: $.AI_CHAT_MESSAGE_TRY_AGAIN
-              })
-            })
-          ]
-        });
-      }
       if (message.deletedAt !== void 0) {
         children = /* @__PURE__ */ jsx("div", {
           className: "lb-ai-chat-message-deleted",
           children: $.AI_CHAT_MESSAGE_DELETED
         });
-      } else if (message.status === "pending") {
+      } else if (message.status === "generating" || message.status === "awaiting-tool") {
         if (message.contentSoFar.length === 0) {
           children = /* @__PURE__ */ jsx("div", {
             className: "lb-ai-chat-message-thinking lb-ai-chat-pending",
@@ -70,52 +35,34 @@ const AiChatAssistantMessage = memo(
         } else {
           children = /* @__PURE__ */ jsx(AssistantMessageContent, {
             content: message.contentSoFar,
-            chatId: message.chatId
+            chatId: message.chatId,
+            messageId: message.id,
+            components
           });
         }
       } else if (message.status === "completed") {
-        const text = message.content.reduce((acc, part) => {
-          if (part.type === "text") {
-            return acc + part.text;
-          }
-          return acc;
-        }, "");
-        children = /* @__PURE__ */ jsxs(Fragment, {
-          children: [
-            /* @__PURE__ */ jsx(AssistantMessageContent, {
-              content: message.content,
-              chatId: message.chatId
-            }),
-            /* @__PURE__ */ jsx(MessageActions, {
-              text
-            })
-          ]
+        children = /* @__PURE__ */ jsx(AssistantMessageContent, {
+          content: message.content,
+          chatId: message.chatId,
+          messageId: message.id,
+          components
         });
       } else if (message.status === "failed") {
-        const text = message.contentSoFar.reduce((acc, part) => {
-          if (part.type === "text") {
-            return acc + part.text;
-          }
-          return acc;
-        }, "");
         if (message.errorReason === "Aborted by user") {
-          children = /* @__PURE__ */ jsxs(Fragment, {
-            children: [
-              /* @__PURE__ */ jsx(AssistantMessageContent, {
-                content: message.contentSoFar,
-                chatId: message.chatId
-              }),
-              /* @__PURE__ */ jsx(MessageActions, {
-                text
-              })
-            ]
+          children = /* @__PURE__ */ jsx(AssistantMessageContent, {
+            content: message.contentSoFar,
+            chatId: message.chatId,
+            messageId: message.id,
+            components
           });
         } else {
           children = /* @__PURE__ */ jsxs(Fragment, {
             children: [
               /* @__PURE__ */ jsx(AssistantMessageContent, {
                 content: message.contentSoFar,
-                chatId: message.chatId
+                chatId: message.chatId,
+                messageId: message.id,
+                components
               }),
               /* @__PURE__ */ jsxs("div", {
                 className: "lb-ai-chat-message-error",
@@ -126,71 +73,28 @@ const AiChatAssistantMessage = memo(
                   }),
                   message.errorReason
                 ]
-              }),
-              /* @__PURE__ */ jsx(MessageActions, {
-                text
               })
             ]
           });
         }
       }
-      return /* @__PURE__ */ jsx(TooltipProvider, {
-        children: /* @__PURE__ */ jsx("div", {
-          className: classNames(
-            "lb-ai-chat-message lb-ai-chat-assistant-message",
-            showActions === "hover" && "lb-ai-chat-message:show-actions-hover",
-            className
-          ),
-          ...props,
-          ref: forwardedRef,
-          children
-        })
+      return /* @__PURE__ */ jsx("div", {
+        className: classNames(
+          "lb-ai-chat-message lb-ai-chat-assistant-message",
+          className
+        ),
+        ...props,
+        ref: forwardedRef,
+        children
       });
     }
   )
 );
-function CopyTextButton({ text, label }) {
-  const [isCopied, setIsCopied] = useState(false);
-  useEffect(() => {
-    const timeoutId = setTimeout(() => {
-      setIsCopied(false);
-    }, 2e3);
-    return () => {
-      clearTimeout(timeoutId);
-    };
-  }, [isCopied]);
-  return /* @__PURE__ */ jsx(Button, {
-    onClick: function() {
-      navigator.clipboard.writeText(text);
-      setIsCopied(true);
-    },
-    className: "lb-ai-chat-message-action",
-    "aria-label": label,
-    icon: isCopied ? /* @__PURE__ */ jsx(CheckIcon, {}) : /* @__PURE__ */ jsx(CopyIcon, {})
-  });
-}
-function RegenerateMessageButton({
-  chatId,
-  messageId,
-  copilotId,
-  label
-}) {
-  const client = useClient();
-  return /* @__PURE__ */ jsx(Button, {
-    onClick: function() {
-      client[kInternal].ai.regenerateMessage(chatId, messageId, {
-        copilotId,
-        stream: true
-      });
-    },
-    className: "lb-ai-chat-message-action",
-    "aria-label": label,
-    icon: /* @__PURE__ */ jsx(RetryIcon, {})
-  });
-}
 function AssistantMessageContent({
   content,
-  chatId
+  chatId,
+  messageId,
+  components
 }) {
   const isReasoning = content.some((part) => part.type === "reasoning") && content.every((part) => part.type === "reasoning");
   return /* @__PURE__ */ jsx("div", {
@@ -200,20 +104,22 @@ function AssistantMessageContent({
         case "text": {
           return /* @__PURE__ */ jsx(TextPart, {
             text: part.text,
+            components,
             className: "lb-ai-chat-message-text"
           }, index);
         }
-        case "tool-call": {
-          return /* @__PURE__ */ jsx(ToolCallPart, {
+        case "tool-invocation": {
+          return /* @__PURE__ */ jsx(ToolInvocationPart, {
             chatId,
-            name: part.toolName,
-            args: part.args
+            messageId,
+            part
           }, index);
         }
         case "reasoning": {
           return /* @__PURE__ */ jsx(ReasoningPart, {
             text: part.text,
-            isPending: isReasoning
+            isPending: isReasoning,
+            components
           }, index);
         }
         default: {
@@ -224,7 +130,7 @@ function AssistantMessageContent({
   });
 }
 const TextPart = forwardRef(
-  ({ text, ...props }, forwardedRef) => {
+  ({ text, components, ...props }, forwardedRef) => {
     const tokens = useMemo(() => {
       return new Lexer().lex(text);
     }, [text]);
@@ -233,16 +139,72 @@ const TextPart = forwardRef(
       ...props,
       children: tokens.map((token, index) => {
         return /* @__PURE__ */ jsx(MemoizedBlockTokenComp, {
-          token
+          token,
+          components
         }, index);
       })
     });
   }
 );
+function CodeBlock({
+  language,
+  code
+}) {
+  const [isCopied, setCopied] = useState(false);
+  const timeoutRef = useRef(null);
+  useEffect(() => {
+    if (isCopied) {
+      timeoutRef.current = setTimeout(() => {
+        setCopied(false);
+      }, 1e3);
+    }
+    return () => {
+      if (timeoutRef.current) {
+        clearTimeout(timeoutRef.current);
+      }
+    };
+  }, [isCopied]);
+  return /* @__PURE__ */ jsxs("div", {
+    className: "lb-code-block",
+    children: [
+      /* @__PURE__ */ jsxs("div", {
+        className: "lb-code-block-header",
+        children: [
+          /* @__PURE__ */ jsx("span", {
+            className: "lb-code-block-title",
+            children: language ?? "Plain text"
+          }),
+          /* @__PURE__ */ jsx("div", {
+            className: "lb-code-block-header-actions",
+            children: /* @__PURE__ */ jsx(Button, {
+              className: "lb-code-block-header-action",
+              icon: isCopied ? /* @__PURE__ */ jsx(CheckIcon, {}) : /* @__PURE__ */ jsx(CopyIcon, {}),
+              onClick: () => {
+                setCopied(true);
+                navigator.clipboard.writeText(code);
+              }
+            })
+          })
+        ]
+      }),
+      /* @__PURE__ */ jsx("pre", {
+        className: "lb-code-block-content",
+        children: /* @__PURE__ */ jsx("code", {
+          children: code
+        })
+      })
+    ]
+  });
+}
 const MemoizedBlockTokenComp = memo(
-  function BlockTokenComp$1({ token }) {
+  function BlockTokenComp$1({
+    token,
+    components
+  }) {
+    const { Anchor } = useComponents(components);
     return /* @__PURE__ */ jsx(BlockTokenComp, {
-      token
+      token,
+      components: { CodeBlock, Anchor }
     });
   },
   (prevProps, nextProps) => {
@@ -257,27 +219,42 @@ const MemoizedBlockTokenComp = memo(
     return prevToken.raw === nextToken.raw;
   }
 );
-function ToolCallPart({
+function noop() {
+}
+function ToolInvocationPart({
   chatId,
-  name,
-  args
+  messageId,
+  part
 }) {
   const client = useClient();
-  const tool = useSignal(
-    client[kInternal].ai.signals.getToolDefinition\u03A3(chatId, name)
+  const ai = client[kInternal].ai;
+  const tool = useSignal(ai.signals.getToolDefinition\u03A3(chatId, part.toolName));
+  const respond = useCallback(
+    (result) => {
+      ai.setToolResult(
+        chatId,
+        messageId,
+        part.toolCallId,
+        result
+      );
+    },
+    [ai, chatId, messageId, part.toolCallId]
   );
   if (tool === void 0 || tool.render === void 0)
     return null;
+  const { type: _, ...rest } = part;
   return /* @__PURE__ */ jsx("div", {
     className: "lb-ai-chat-message-tool",
     children: /* @__PURE__ */ jsx(tool.render, {
-      args
+      ...rest,
+      respond: part.status === "executing" ? respond : noop
     })
   });
 }
 function ReasoningPart({
   text,
-  isPending
+  isPending,
+  components
 }) {
   const [isOpen, setIsOpen] = useState(false);
   return /* @__PURE__ */ jsxs(Root, {
@@ -293,14 +270,17 @@ function ReasoningPart({
         children: [
           "Reasoning",
           /* @__PURE__ */ jsx("span", {
-            className: "lb-icon-container",
-            children: isOpen ? /* @__PURE__ */ jsx(ChevronDownIcon, {}) : /* @__PURE__ */ jsx(ChevronRightIcon, {})
+            className: "lb-ai-chat-message-collapsible-chevron lb-icon-container",
+            children: /* @__PURE__ */ jsx(ChevronRightIcon, {})
           })
         ]
       }),
       /* @__PURE__ */ jsx(Content, {
         className: "lb-ai-chat-message-collapsible-content",
-        children: text
+        children: /* @__PURE__ */ jsx(TextPart, {
+          text,
+          components
+        })
       })
     ]
   });

package/dist/components/internal/AiChatAssistantMessage.js.map

@@ -1 +1 @@
-{"version":3,"file":"AiChatAssistantMessage.js","sources":["../../../src/components/internal/AiChatAssistantMessage.tsx"],"sourcesContent":["import {\n  type AiAssistantContentPart,\n  type CopilotId,\n  kInternal,\n  type MessageId,\n  type UiAssistantMessage,\n} from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { Lexer } from \"marked\";\nimport {\n  type ComponentProps,\n  forwardRef,\n  memo,\n  type ReactNode,\n  useEffect,\n  useMemo,\n  useState,\n} from \"react\";\n\nimport { Button } from \"../../components/internal/Button\";\nimport { Tooltip, TooltipProvider } from \"../../components/internal/Tooltip\";\nimport { CheckIcon } from \"../../icons/Check\";\nimport { ChevronDownIcon } from \"../../icons/ChevronDown\";\nimport { ChevronRightIcon } from \"../../icons/ChevronRight\";\nimport { CopyIcon } from \"../../icons/Copy\";\nimport { RetryIcon } from \"../../icons/Retry\";\nimport { WarningIcon } from \"../../icons/Warning\";\nimport {\n  type AiChatMessageOverrides,\n  type GlobalOverrides,\n  useOverrides,\n} from \"../../overrides\";\nimport * as CollapsiblePrimitive from \"../../primitives/internal/Collapsible\";\nimport {\n  type BlockToken,\n  BlockTokenComp as BlockTokenCompPrimitive,\n} from \"../../primitives/internal/Markdown\";\nimport { classNames } from \"../../utils/class-names\";\n\n/* -------------------------------------------------------------------------------------------------\n * AiChatAssistantMessage\n * -----------------------------------------------------------------------------------------------*/\nexport interface AiChatAssistantMessageProps extends ComponentProps<\"div\"> {\n  /**\n   * The message to display.\n   */\n  message: UiAssistantMessage;\n  /**\n   * Whether to show or hide message actions.\n   * @internal\n   */\n  showActions?: boolean | \"hover\";\n  /**\n   * Override the component's strings.\n   */\n  overrides?: Partial<GlobalOverrides & AiChatMessageOverrides>;\n  /**\n   * @internal\n   * Whether to show or hide the regenerate button.\n   */\n  showRegenerate?: boolean;\n  /**\n   * @internal\n   * The id of the copilot to use to regenerate the message. Only used if `showRegenerate` is true.\n   */\n  copilotId?: CopilotId;\n}\n\nexport const AiChatAssistantMessage = memo(\n  forwardRef<HTMLDivElement, AiChatAssistantMessageProps>(\n    (\n      {\n        message,\n        showActions = false,\n        showRegenerate = false,\n        copilotId,\n        className,\n        overrides,\n        ...props\n      },\n      forwardedRef\n    ) => {\n      const $ = useOverrides(overrides);\n\n      let children: ReactNode = null;\n\n      function MessageActions({ text }: { text: string }) {\n        if (!showActions) return null;\n\n        return (\n          <div className=\"lb-ai-chat-message-actions\">\n            <Tooltip content={$.AI_CHAT_MESSAGE_COPY}>\n              <CopyTextButton text={text} label={$.AI_CHAT_MESSAGE_COPY} />\n            </Tooltip>\n\n            {showRegenerate && (\n              <Tooltip content={$.AI_CHAT_MESSAGE_TRY_AGAIN}>\n                <RegenerateMessageButton\n                  chatId={message.chatId}\n                  messageId={message.id}\n                  copilotId={copilotId}\n                  label={$.AI_CHAT_MESSAGE_TRY_AGAIN}\n                />\n              </Tooltip>\n            )}\n          </div>\n        );\n      }\n\n      if (message.deletedAt !== undefined) {\n        children = (\n          <div className=\"lb-ai-chat-message-deleted\">\n            {$.AI_CHAT_MESSAGE_DELETED}\n          </div>\n        );\n      } else if (message.status === \"pending\") {\n        if (message.contentSoFar.length === 0) {\n          children = (\n            <div className=\"lb-ai-chat-message-thinking lb-ai-chat-pending\">\n              {$.AI_CHAT_MESSAGE_THINKING}\n            </div>\n          );\n        } else {\n          children = (\n            <AssistantMessageContent\n              content={message.contentSoFar}\n              chatId={message.chatId}\n            />\n          );\n        }\n      } else if (message.status === \"completed\") {\n        const text: string = message.content.reduce((acc, part) => {\n          if (part.type === \"text\") {\n            return acc + part.text;\n          }\n          return acc;\n        }, \"\");\n\n        children = (\n          <>\n            <AssistantMessageContent\n              content={message.content}\n              chatId={message.chatId}\n            />\n\n            <MessageActions text={text} />\n          </>\n        );\n      } else if (message.status === \"failed\") {\n        const text: string = message.contentSoFar.reduce((acc, part) => {\n          if (part.type === \"text\") {\n            return acc + part.text;\n          }\n          return acc;\n        }, \"\");\n\n        // Do not include the error message if the user aborted the request.\n        if (message.errorReason === \"Aborted by user\") {\n          children = (\n            <>\n              <AssistantMessageContent\n                content={message.contentSoFar}\n                chatId={message.chatId}\n              />\n              <MessageActions text={text} />\n            </>\n          );\n        } else {\n          children = (\n            <>\n              <AssistantMessageContent\n                content={message.contentSoFar}\n                chatId={message.chatId}\n              />\n\n              <div className=\"lb-ai-chat-message-error\">\n                <span className=\"lb-icon-container\">\n                  <WarningIcon />\n                </span>\n                {message.errorReason}\n              </div>\n\n              <MessageActions text={text} />\n            </>\n          );\n        }\n      }\n\n      return (\n        <TooltipProvider>\n          <div\n            className={classNames(\n              \"lb-ai-chat-message lb-ai-chat-assistant-message\",\n              showActions === \"hover\" &&\n                \"lb-ai-chat-message:show-actions-hover\",\n              className\n            )}\n            {...props}\n            ref={forwardedRef}\n          >\n            {children}\n          </div>\n        </TooltipProvider>\n      );\n    }\n  )\n);\n\nfunction CopyTextButton({ text, label }: { text: string; label: string }) {\n  const [isCopied, setIsCopied] = useState(false);\n\n  useEffect(() => {\n    const timeoutId = setTimeout(() => {\n      setIsCopied(false);\n    }, 2000);\n    return () => {\n      clearTimeout(timeoutId);\n    };\n  }, [isCopied]);\n\n  return (\n    <Button\n      onClick={function () {\n        navigator.clipboard.writeText(text);\n        setIsCopied(true);\n      }}\n      className=\"lb-ai-chat-message-action\"\n      aria-label={label}\n      icon={isCopied ? <CheckIcon /> : <CopyIcon />}\n    />\n  );\n}\n\nfunction RegenerateMessageButton({\n  chatId,\n  messageId,\n  copilotId,\n  label,\n}: {\n  chatId: string;\n  messageId: MessageId;\n  copilotId?: CopilotId;\n  label: string;\n}) {\n  const client = useClient();\n\n  return (\n    <Button\n      onClick={function () {\n        client[kInternal].ai.regenerateMessage(chatId, messageId, {\n          copilotId,\n          stream: true,\n        });\n      }}\n      className=\"lb-ai-chat-message-action\"\n      aria-label={label}\n      icon={<RetryIcon />}\n    />\n  );\n}\n\nfunction AssistantMessageContent({\n  content,\n  chatId,\n}: {\n  content: AiAssistantContentPart[];\n  chatId: string;\n}) {\n  // A message is considered to be in \"reasoning\" state if it only contains reasoning parts and no other parts.\n  const isReasoning =\n    content.some((part) => part.type === \"reasoning\") &&\n    content.every((part) => part.type === \"reasoning\");\n\n  return (\n    <div className=\"lb-ai-chat-message-content\">\n      {content.map((part, index) => {\n        switch (part.type) {\n          case \"text\": {\n            return (\n              <TextPart\n                key={index}\n                text={part.text}\n                className=\"lb-ai-chat-message-text\"\n              />\n            );\n          }\n          case \"tool-call\": {\n            return (\n              <ToolCallPart\n                key={index}\n                chatId={chatId}\n                name={part.toolName}\n                args={part.args}\n              />\n            );\n          }\n          case \"reasoning\": {\n            return (\n              <ReasoningPart\n                key={index}\n                text={part.text}\n                isPending={isReasoning}\n              />\n            );\n          }\n          default: {\n            return null;\n          }\n        }\n      })}\n    </div>\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * TextPart\n * -----------------------------------------------------------------------------------------------*/\ninterface TextPartProps extends ComponentProps<\"div\"> {\n  text: string;\n}\n\nconst TextPart = forwardRef<HTMLDivElement, TextPartProps>(\n  ({ text, ...props }, forwardedRef) => {\n    const tokens = useMemo(() => {\n      return new Lexer().lex(text);\n    }, [text]);\n\n    return (\n      <div ref={forwardedRef} {...props}>\n        {tokens.map((token, index) => {\n          return (\n            <MemoizedBlockTokenComp token={token as BlockToken} key={index} />\n          );\n        })}\n      </div>\n    );\n  }\n);\n\nconst MemoizedBlockTokenComp = memo(\n  function BlockTokenComp({ token }: { token: BlockToken }) {\n    return <BlockTokenCompPrimitive token={token} />;\n  },\n  (prevProps, nextProps) => {\n    const prevToken = prevProps.token;\n    const nextToken = nextProps.token;\n    if (prevToken.raw.length !== nextToken.raw.length) {\n      return false;\n    }\n    if (prevToken.type !== nextToken.type) {\n      return false;\n    }\n    return prevToken.raw === nextToken.raw;\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * ToolCallPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ToolCallPart({\n  chatId,\n  name,\n  args,\n}: {\n  chatId: string;\n  name: string;\n  args: any;\n}) {\n  const client = useClient();\n\n  const tool = useSignal(\n    client[kInternal].ai.signals.getToolDefinitionΣ(chatId, name)\n  );\n  if (tool === undefined || tool.render === undefined) return null;\n\n  return (\n    <div className=\"lb-ai-chat-message-tool\">\n      <tool.render args={args as unknown} />\n    </div>\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * ReasoningPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ReasoningPart({\n  text,\n  isPending,\n}: {\n  text: string;\n  isPending: boolean;\n}) {\n  const [isOpen, setIsOpen] = useState(false);\n  return (\n    <CollapsiblePrimitive.Root\n      className=\"lb-ai-chat-message-collapsible lb-ai-chat-message-reasoning\"\n      open={isOpen}\n      onOpenChange={setIsOpen}\n    >\n      <CollapsiblePrimitive.Trigger\n        className={classNames(\n          \"lb-ai-chat-message-collapsible-trigger\",\n          isPending && \"lb-ai-chat-pending\"\n        )}\n      >\n        Reasoning\n        <span className=\"lb-icon-container\">\n          {isOpen ? <ChevronDownIcon /> : <ChevronRightIcon />}\n        </span>\n      </CollapsiblePrimitive.Trigger>\n\n      <CollapsiblePrimitive.Content className=\"lb-ai-chat-message-collapsible-content\">\n        {text}\n      </CollapsiblePrimitive.Content>\n    </CollapsiblePrimitive.Root>\n  );\n}\n"],"names":["BlockTokenComp","BlockTokenCompPrimitive","CollapsiblePrimitive.Root","CollapsiblePrimitive.Trigger","CollapsiblePrimitive.Content"],"mappings":";;;;;;;;;;;;;;;;;;;;AAqEO,MAAM,sBAAyB,GAAA,IAAA;AAAA,EACpC,UAAA;AAAA,IACE,CACE;AAAA,MACE,OAAA;AAAA,MACA,WAAc,GAAA,KAAA;AAAA,MACd,cAAiB,GAAA,KAAA;AAAA,MACjB,SAAA;AAAA,MACA,SAAA;AAAA,MACA,SAAA;AAAA,MACG,GAAA,KAAA;AAAA,OAEL,YACG,KAAA;AACH,MAAM,MAAA,CAAA,GAAI,aAAa,SAAS,CAAA,CAAA;AAEhC,MAAA,IAAI,QAAsB,GAAA,IAAA,CAAA;AAE1B,MAAS,SAAA,cAAA,CAAe,EAAE,IAAA,EAA0B,EAAA;AAClD,QAAA,IAAI,CAAC,WAAA;AAAa,UAAO,OAAA,IAAA,CAAA;AAEzB,QAAA,uBACG,IAAA,CAAA,KAAA,EAAA;AAAA,UAAI,SAAU,EAAA,4BAAA;AAAA,UACb,QAAA,EAAA;AAAA,4BAAC,GAAA,CAAA,OAAA,EAAA;AAAA,cAAQ,SAAS,CAAE,CAAA,oBAAA;AAAA,cAClB,QAAC,kBAAA,GAAA,CAAA,cAAA,EAAA;AAAA,gBAAe,IAAA;AAAA,gBAAY,OAAO,CAAE,CAAA,oBAAA;AAAA,eAAsB,CAAA;AAAA,aAC7D,CAAA;AAAA,YAEC,kCACE,GAAA,CAAA,OAAA,EAAA;AAAA,cAAQ,SAAS,CAAE,CAAA,yBAAA;AAAA,cAClB,QAAC,kBAAA,GAAA,CAAA,uBAAA,EAAA;AAAA,gBACC,QAAQ,OAAQ,CAAA,MAAA;AAAA,gBAChB,WAAW,OAAQ,CAAA,EAAA;AAAA,gBACnB,SAAA;AAAA,gBACA,OAAO,CAAE,CAAA,yBAAA;AAAA,eACX,CAAA;AAAA,aACF,CAAA;AAAA,WAAA;AAAA,SAEJ,CAAA,CAAA;AAAA,OAEJ;AAEA,MAAI,IAAA,OAAA,CAAQ,cAAc,KAAW,CAAA,EAAA;AACnC,QAAA,QAAA,mBACG,GAAA,CAAA,KAAA,EAAA;AAAA,UAAI,SAAU,EAAA,4BAAA;AAAA,UACZ,QAAE,EAAA,CAAA,CAAA,uBAAA;AAAA,SACL,CAAA,CAAA;AAAA,OAEJ,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,SAAW,EAAA;AACvC,QAAI,IAAA,OAAA,CAAQ,YAAa,CAAA,MAAA,KAAW,CAAG,EAAA;AACrC,UAAA,QAAA,mBACG,GAAA,CAAA,KAAA,EAAA;AAAA,YAAI,SAAU,EAAA,gDAAA;AAAA,YACZ,QAAE,EAAA,CAAA,CAAA,wBAAA;AAAA,WACL,CAAA,CAAA;AAAA,SAEG,MAAA;AACL,UAAA,QAAA,mBACG,GAAA,CAAA,uBAAA,EAAA;AAAA,YACC,SAAS,OAAQ,CAAA,YAAA;AAAA,YACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,WAClB,CAAA,CAAA;AAAA,SAEJ;AAAA,OACF,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,WAAa,EAAA;AACzC,QAAA,MAAM,OAAe,OAAQ,CAAA,OAAA,CAAQ,MAAO,CAAA,CAAC,KAAK,IAAS,KAAA;AACzD,UAAI,IAAA,IAAA,CAAK,SAAS,MAAQ,EAAA;AACxB,YAAA,OAAO,MAAM,IAAK,CAAA,IAAA,CAAA;AAAA,WACpB;AACA,UAAO,OAAA,GAAA,CAAA;AAAA,WACN,EAAE,CAAA,CAAA;AAEL,QACE,QAAA,mBAAA,IAAA,CAAA,QAAA,EAAA;AAAA,UACE,QAAA,EAAA;AAAA,4BAAC,GAAA,CAAA,uBAAA,EAAA;AAAA,cACC,SAAS,OAAQ,CAAA,OAAA;AAAA,cACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,aAClB,CAAA;AAAA,4BAEC,GAAA,CAAA,cAAA,EAAA;AAAA,cAAe,IAAA;AAAA,aAAY,CAAA;AAAA,WAAA;AAAA,SAC9B,CAAA,CAAA;AAAA,OAEJ,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,QAAU,EAAA;AACtC,QAAA,MAAM,OAAe,OAAQ,CAAA,YAAA,CAAa,MAAO,CAAA,CAAC,KAAK,IAAS,KAAA;AAC9D,UAAI,IAAA,IAAA,CAAK,SAAS,MAAQ,EAAA;AACxB,YAAA,OAAO,MAAM,IAAK,CAAA,IAAA,CAAA;AAAA,WACpB;AACA,UAAO,OAAA,GAAA,CAAA;AAAA,WACN,EAAE,CAAA,CAAA;AAGL,QAAI,IAAA,OAAA,CAAQ,gBAAgB,iBAAmB,EAAA;AAC7C,UACE,QAAA,mBAAA,IAAA,CAAA,QAAA,EAAA;AAAA,YACE,QAAA,EAAA;AAAA,8BAAC,GAAA,CAAA,uBAAA,EAAA;AAAA,gBACC,SAAS,OAAQ,CAAA,YAAA;AAAA,gBACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,eAClB,CAAA;AAAA,8BACC,GAAA,CAAA,cAAA,EAAA;AAAA,gBAAe,IAAA;AAAA,eAAY,CAAA;AAAA,aAAA;AAAA,WAC9B,CAAA,CAAA;AAAA,SAEG,MAAA;AACL,UACE,QAAA,mBAAA,IAAA,CAAA,QAAA,EAAA;AAAA,YACE,QAAA,EAAA;AAAA,8BAAC,GAAA,CAAA,uBAAA,EAAA;AAAA,gBACC,SAAS,OAAQ,CAAA,YAAA;AAAA,gBACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,eAClB,CAAA;AAAA,8BAEC,IAAA,CAAA,KAAA,EAAA;AAAA,gBAAI,SAAU,EAAA,0BAAA;AAAA,gBACb,QAAA,EAAA;AAAA,kCAAC,GAAA,CAAA,MAAA,EAAA;AAAA,oBAAK,SAAU,EAAA,mBAAA;AAAA,oBACd,8BAAC,WAAY,EAAA,EAAA,CAAA;AAAA,mBACf,CAAA;AAAA,kBACC,OAAQ,CAAA,WAAA;AAAA,iBAAA;AAAA,eACX,CAAA;AAAA,8BAEC,GAAA,CAAA,cAAA,EAAA;AAAA,gBAAe,IAAA;AAAA,eAAY,CAAA;AAAA,aAAA;AAAA,WAC9B,CAAA,CAAA;AAAA,SAEJ;AAAA,OACF;AAEA,MAAA,uBACG,GAAA,CAAA,eAAA,EAAA;AAAA,QACC,QAAC,kBAAA,GAAA,CAAA,KAAA,EAAA;AAAA,UACC,SAAW,EAAA,UAAA;AAAA,YACT,iDAAA;AAAA,YACA,gBAAgB,OACd,IAAA,uCAAA;AAAA,YACF,SAAA;AAAA,WACF;AAAA,UACC,GAAG,KAAA;AAAA,UACJ,GAAK,EAAA,YAAA;AAAA,UAEJ,QAAA;AAAA,SACH,CAAA;AAAA,OACF,CAAA,CAAA;AAAA,KAEJ;AAAA,GACF;AACF,EAAA;AAEA,SAAS,cAAe,CAAA,EAAE,IAAM,EAAA,KAAA,EAA0C,EAAA;AACxE,EAAA,MAAM,CAAC,QAAA,EAAU,WAAW,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AAE9C,EAAA,SAAA,CAAU,MAAM;AACd,IAAM,MAAA,SAAA,GAAY,WAAW,MAAM;AACjC,MAAA,WAAA,CAAY,KAAK,CAAA,CAAA;AAAA,OAChB,GAAI,CAAA,CAAA;AACP,IAAA,OAAO,MAAM;AACX,MAAA,YAAA,CAAa,SAAS,CAAA,CAAA;AAAA,KACxB,CAAA;AAAA,GACF,EAAG,CAAC,QAAQ,CAAC,CAAA,CAAA;AAEb,EAAA,uBACG,GAAA,CAAA,MAAA,EAAA;AAAA,IACC,SAAS,WAAY;AACnB,MAAU,SAAA,CAAA,SAAA,CAAU,UAAU,IAAI,CAAA,CAAA;AAClC,MAAA,WAAA,CAAY,IAAI,CAAA,CAAA;AAAA,KAClB;AAAA,IACA,SAAU,EAAA,2BAAA;AAAA,IACV,YAAY,EAAA,KAAA;AAAA,IACZ,MAAM,QAAW,mBAAA,GAAA,CAAC,SAAU,EAAA,EAAA,CAAA,uBAAM,QAAS,EAAA,EAAA,CAAA;AAAA,GAC7C,CAAA,CAAA;AAEJ,CAAA;AAEA,SAAS,uBAAwB,CAAA;AAAA,EAC/B,MAAA;AAAA,EACA,SAAA;AAAA,EACA,SAAA;AAAA,EACA,KAAA;AACF,CAKG,EAAA;AACD,EAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AAEzB,EAAA,uBACG,GAAA,CAAA,MAAA,EAAA;AAAA,IACC,SAAS,WAAY;AACnB,MAAA,MAAA,CAAO,SAAW,CAAA,CAAA,EAAA,CAAG,iBAAkB,CAAA,MAAA,EAAQ,SAAW,EAAA;AAAA,QACxD,SAAA;AAAA,QACA,MAAQ,EAAA,IAAA;AAAA,OACT,CAAA,CAAA;AAAA,KACH;AAAA,IACA,SAAU,EAAA,2BAAA;AAAA,IACV,YAAY,EAAA,KAAA;AAAA,IACZ,IAAA,sBAAO,SAAU,EAAA,EAAA,CAAA;AAAA,GACnB,CAAA,CAAA;AAEJ,CAAA;AAEA,SAAS,uBAAwB,CAAA;AAAA,EAC/B,OAAA;AAAA,EACA,MAAA;AACF,CAGG,EAAA;AAED,EAAA,MAAM,WACJ,GAAA,OAAA,CAAQ,IAAK,CAAA,CAAC,SAAS,IAAK,CAAA,IAAA,KAAS,WAAW,CAAA,IAChD,QAAQ,KAAM,CAAA,CAAC,IAAS,KAAA,IAAA,CAAK,SAAS,WAAW,CAAA,CAAA;AAEnD,EAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,4BAAA;AAAA,IACZ,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAC5B,MAAA,QAAQ,KAAK,IAAM;AAAA,QACjB,KAAK,MAAQ,EAAA;AACX,UAAA,uBACG,GAAA,CAAA,QAAA,EAAA;AAAA,YAEC,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,SAAU,EAAA,yBAAA;AAAA,WAAA,EAFL,KAGP,CAAA,CAAA;AAAA,SAEJ;AAAA,QACA,KAAK,WAAa,EAAA;AAChB,UAAA,uBACG,GAAA,CAAA,YAAA,EAAA;AAAA,YAEC,MAAA;AAAA,YACA,MAAM,IAAK,CAAA,QAAA;AAAA,YACX,MAAM,IAAK,CAAA,IAAA;AAAA,WAAA,EAHN,KAIP,CAAA,CAAA;AAAA,SAEJ;AAAA,QACA,KAAK,WAAa,EAAA;AAChB,UAAA,uBACG,GAAA,CAAA,aAAA,EAAA;AAAA,YAEC,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,SAAW,EAAA,WAAA;AAAA,WAAA,EAFN,KAGP,CAAA,CAAA;AAAA,SAEJ;AAAA,QACA,SAAS;AACP,UAAO,OAAA,IAAA,CAAA;AAAA,SACT;AAAA,OACF;AAAA,KACD,CAAA;AAAA,GACH,CAAA,CAAA;AAEJ,CAAA;AASA,MAAM,QAAW,GAAA,UAAA;AAAA,EACf,CAAC,EAAE,IAAS,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AACpC,IAAM,MAAA,MAAA,GAAS,QAAQ,MAAM;AAC3B,MAAA,OAAO,IAAI,KAAA,EAAQ,CAAA,GAAA,CAAI,IAAI,CAAA,CAAA;AAAA,KAC7B,EAAG,CAAC,IAAI,CAAC,CAAA,CAAA;AAET,IAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,MAAI,GAAK,EAAA,YAAA;AAAA,MAAe,GAAG,KAAA;AAAA,MACzB,QAAO,EAAA,MAAA,CAAA,GAAA,CAAI,CAAC,KAAA,EAAO,KAAU,KAAA;AAC5B,QAAA,uBACG,GAAA,CAAA,sBAAA,EAAA;AAAA,UAAuB,KAAA;AAAA,SAAA,EAAiC,KAAO,CAAA,CAAA;AAAA,OAEnE,CAAA;AAAA,KACH,CAAA,CAAA;AAAA,GAEJ;AACF,CAAA,CAAA;AAEA,MAAM,sBAAyB,GAAA,IAAA;AAAA,EAC7B,SAASA,gBAAA,CAAe,EAAE,KAAA,EAAgC,EAAA;AACxD,IAAA,uBAAQ,GAAA,CAAAC,cAAA,EAAA;AAAA,MAAwB,KAAA;AAAA,KAAc,CAAA,CAAA;AAAA,GAChD;AAAA,EACA,CAAC,WAAW,SAAc,KAAA;AACxB,IAAA,MAAM,YAAY,SAAU,CAAA,KAAA,CAAA;AAC5B,IAAA,MAAM,YAAY,SAAU,CAAA,KAAA,CAAA;AAC5B,IAAA,IAAI,SAAU,CAAA,GAAA,CAAI,MAAW,KAAA,SAAA,CAAU,IAAI,MAAQ,EAAA;AACjD,MAAO,OAAA,KAAA,CAAA;AAAA,KACT;AACA,IAAI,IAAA,SAAA,CAAU,IAAS,KAAA,SAAA,CAAU,IAAM,EAAA;AACrC,MAAO,OAAA,KAAA,CAAA;AAAA,KACT;AACA,IAAO,OAAA,SAAA,CAAU,QAAQ,SAAU,CAAA,GAAA,CAAA;AAAA,GACrC;AACF,CAAA,CAAA;AAKA,SAAS,YAAa,CAAA;AAAA,EACpB,MAAA;AAAA,EACA,IAAA;AAAA,EACA,IAAA;AACF,CAIG,EAAA;AACD,EAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AAEzB,EAAA,MAAM,IAAO,GAAA,SAAA;AAAA,IACX,OAAO,SAAW,CAAA,CAAA,EAAA,CAAG,OAAQ,CAAA,uBAAA,CAAmB,QAAQ,IAAI,CAAA;AAAA,GAC9D,CAAA;AACA,EAAI,IAAA,IAAA,KAAS,KAAa,CAAA,IAAA,IAAA,CAAK,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AAE5D,EAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,yBAAA;AAAA,IACb,QAAA,kBAAA,GAAA,CAAC,KAAK,MAAL,EAAA;AAAA,MAAY,IAAA;AAAA,KAAuB,CAAA;AAAA,GACtC,CAAA,CAAA;AAEJ,CAAA;AAKA,SAAS,aAAc,CAAA;AAAA,EACrB,IAAA;AAAA,EACA,SAAA;AACF,CAGG,EAAA;AACD,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AAC1C,EACE,uBAAA,IAAA,CAACC,IAAA,EAAA;AAAA,IACC,SAAU,EAAA,6DAAA;AAAA,IACV,IAAM,EAAA,MAAA;AAAA,IACN,YAAc,EAAA,SAAA;AAAA,IAEd,QAAA,EAAA;AAAA,sBAAA,IAAA,CAACC,OAAA,EAAA;AAAA,QACC,SAAW,EAAA,UAAA;AAAA,UACT,wCAAA;AAAA,UACA,SAAa,IAAA,oBAAA;AAAA,SACf;AAAA,QACD,QAAA,EAAA;AAAA,UAAA,WAAA;AAAA,0BAEE,GAAA,CAAA,MAAA,EAAA;AAAA,YAAK,SAAU,EAAA,mBAAA;AAAA,YACb,QAAS,EAAA,MAAA,mBAAA,GAAA,CAAC,eAAgB,EAAA,EAAA,CAAA,uBAAM,gBAAiB,EAAA,EAAA,CAAA;AAAA,WACpD,CAAA;AAAA,SAAA;AAAA,OACF,CAAA;AAAA,sBAEA,GAAA,CAACC,OAAA,EAAA;AAAA,QAA6B,SAAU,EAAA,wCAAA;AAAA,QACrC,QAAA,EAAA,IAAA;AAAA,OACH,CAAA;AAAA,KAAA;AAAA,GACF,CAAA,CAAA;AAEJ;;;;"}
+{"version":3,"file":"AiChatAssistantMessage.js","sources":["../../../src/components/internal/AiChatAssistantMessage.tsx"],"sourcesContent":["import type {\n  AiAssistantContentPart,\n  AiToolInvocationPart,\n  Json,\n  MessageId,\n  UiAssistantMessage,\n} from \"@liveblocks/core\";\nimport { kInternal } from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { Lexer } from \"marked\";\nimport {\n  type ComponentProps,\n  forwardRef,\n  memo,\n  type ReactNode,\n  useCallback,\n  useEffect,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\n\nimport { Button } from \"../../_private\";\nimport { type GlobalComponents, useComponents } from \"../../components\";\nimport { CheckIcon } from \"../../icons/Check\";\nimport { ChevronRightIcon } from \"../../icons/ChevronRight\";\nimport { CopyIcon } from \"../../icons/Copy\";\nimport { WarningIcon } from \"../../icons/Warning\";\nimport {\n  type AiChatMessageOverrides,\n  type GlobalOverrides,\n  useOverrides,\n} from \"../../overrides\";\nimport * as CollapsiblePrimitive from \"../../primitives/internal/Collapsible\";\nimport {\n  type BlockToken,\n  BlockTokenComp as BlockTokenCompPrimitive,\n  type MarkdownComponents,\n} from \"../../primitives/internal/Markdown\";\nimport { classNames } from \"../../utils/class-names\";\n\n/* -------------------------------------------------------------------------------------------------\n * AiChatAssistantMessage\n * -----------------------------------------------------------------------------------------------*/\nexport interface AiChatAssistantMessageProps extends ComponentProps<\"div\"> {\n  /**\n   * The message to display.\n   */\n  message: UiAssistantMessage;\n\n  /**\n   * Override the component's strings.\n   */\n  overrides?: Partial<GlobalOverrides & AiChatMessageOverrides>;\n\n  /**\n   * Override the component's components.\n   */\n  components?: Partial<GlobalComponents>;\n}\n\nexport const AiChatAssistantMessage = memo(\n  forwardRef<HTMLDivElement, AiChatAssistantMessageProps>(\n    ({ message, className, overrides, components, ...props }, forwardedRef) => {\n      const $ = useOverrides(overrides);\n\n      let children: ReactNode = null;\n\n      if (message.deletedAt !== undefined) {\n        children = (\n          <div className=\"lb-ai-chat-message-deleted\">\n            {$.AI_CHAT_MESSAGE_DELETED}\n          </div>\n        );\n      } else if (\n        message.status === \"generating\" ||\n        message.status === \"awaiting-tool\"\n      ) {\n        if (message.contentSoFar.length === 0) {\n          children = (\n            <div className=\"lb-ai-chat-message-thinking lb-ai-chat-pending\">\n              {$.AI_CHAT_MESSAGE_THINKING}\n            </div>\n          );\n        } else {\n          children = (\n            <AssistantMessageContent\n              content={message.contentSoFar}\n              chatId={message.chatId}\n              messageId={message.id}\n              components={components}\n            />\n          );\n        }\n      } else if (message.status === \"completed\") {\n        children = (\n          <AssistantMessageContent\n            content={message.content}\n            chatId={message.chatId}\n            messageId={message.id}\n            components={components}\n          />\n        );\n      } else if (message.status === \"failed\") {\n        // Do not include the error message if the user aborted the request.\n        if (message.errorReason === \"Aborted by user\") {\n          children = (\n            <AssistantMessageContent\n              content={message.contentSoFar}\n              chatId={message.chatId}\n              messageId={message.id}\n              components={components}\n            />\n          );\n        } else {\n          children = (\n            <>\n              <AssistantMessageContent\n                content={message.contentSoFar}\n                chatId={message.chatId}\n                messageId={message.id}\n                components={components}\n              />\n\n              <div className=\"lb-ai-chat-message-error\">\n                <span className=\"lb-icon-container\">\n                  <WarningIcon />\n                </span>\n                {message.errorReason}\n              </div>\n            </>\n          );\n        }\n      }\n\n      return (\n        <div\n          className={classNames(\n            \"lb-ai-chat-message lb-ai-chat-assistant-message\",\n            className\n          )}\n          {...props}\n          ref={forwardedRef}\n        >\n          {children}\n        </div>\n      );\n    }\n  )\n);\n\nfunction AssistantMessageContent({\n  content,\n  chatId,\n  messageId,\n  components,\n}: {\n  content: AiAssistantContentPart[];\n  chatId: string;\n  messageId: MessageId;\n  components: Partial<GlobalComponents> | undefined;\n}) {\n  // A message is considered to be in \"reasoning\" state if it only contains reasoning parts and no other parts.\n  const isReasoning =\n    content.some((part) => part.type === \"reasoning\") &&\n    content.every((part) => part.type === \"reasoning\");\n\n  return (\n    <div className=\"lb-ai-chat-message-content\">\n      {content.map((part, index) => {\n        switch (part.type) {\n          case \"text\": {\n            return (\n              <TextPart\n                key={index}\n                text={part.text}\n                components={components}\n                className=\"lb-ai-chat-message-text\"\n              />\n            );\n          }\n          case \"tool-invocation\": {\n            return (\n              <ToolInvocationPart\n                key={index}\n                chatId={chatId}\n                messageId={messageId}\n                part={part}\n              />\n            );\n          }\n          case \"reasoning\": {\n            return (\n              <ReasoningPart\n                key={index}\n                text={part.text}\n                isPending={isReasoning}\n                components={components}\n              />\n            );\n          }\n          default: {\n            return null;\n          }\n        }\n      })}\n    </div>\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * TextPart\n * -----------------------------------------------------------------------------------------------*/\ninterface TextPartProps extends ComponentProps<\"div\"> {\n  text: string;\n  components: Partial<GlobalComponents> | undefined;\n}\n\nconst TextPart = forwardRef<HTMLDivElement, TextPartProps>(\n  ({ text, components, ...props }, forwardedRef) => {\n    const tokens = useMemo(() => {\n      return new Lexer().lex(text);\n    }, [text]);\n\n    return (\n      <div ref={forwardedRef} {...props}>\n        {tokens.map((token, index) => {\n          return (\n            <MemoizedBlockTokenComp\n              token={token as BlockToken}\n              key={index}\n              components={components}\n            />\n          );\n        })}\n      </div>\n    );\n  }\n);\n\n// TODO: Improve (better copy handling, tooltips, etc)\nfunction CodeBlock({\n  language,\n  code,\n}: ComponentProps<MarkdownComponents[\"CodeBlock\"]>) {\n  const [isCopied, setCopied] = useState(false);\n  const timeoutRef = useRef<ReturnType<typeof setTimeout> | null>(null);\n\n  useEffect(() => {\n    if (isCopied) {\n      timeoutRef.current = setTimeout(() => {\n        setCopied(false);\n      }, 1000);\n    }\n\n    return () => {\n      if (timeoutRef.current) {\n        clearTimeout(timeoutRef.current);\n      }\n    };\n  }, [isCopied]);\n\n  return (\n    <div className=\"lb-code-block\">\n      <div className=\"lb-code-block-header\">\n        <span className=\"lb-code-block-title\">{language ?? \"Plain text\"}</span>\n        <div className=\"lb-code-block-header-actions\">\n          <Button\n            className=\"lb-code-block-header-action\"\n            icon={isCopied ? <CheckIcon /> : <CopyIcon />}\n            onClick={() => {\n              setCopied(true);\n              navigator.clipboard.writeText(code);\n            }}\n          />\n        </div>\n      </div>\n      <pre className=\"lb-code-block-content\">\n        <code>{code}</code>\n      </pre>\n    </div>\n  );\n}\n\nconst MemoizedBlockTokenComp = memo(\n  function BlockTokenComp({\n    token,\n    components,\n  }: {\n    token: BlockToken;\n    components?: Partial<GlobalComponents>;\n  }) {\n    const { Anchor } = useComponents(components);\n\n    return (\n      <BlockTokenCompPrimitive\n        token={token}\n        components={{ CodeBlock, Anchor }}\n      />\n    );\n  },\n  (prevProps, nextProps) => {\n    const prevToken = prevProps.token;\n    const nextToken = nextProps.token;\n    if (prevToken.raw.length !== nextToken.raw.length) {\n      return false;\n    }\n    if (prevToken.type !== nextToken.type) {\n      return false;\n    }\n    return prevToken.raw === nextToken.raw;\n  }\n);\n\nfunction noop() {\n  // Do nothing\n}\n\n/* -------------------------------------------------------------------------------------------------\n * ToolInvocationPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ToolInvocationPart({\n  chatId,\n  messageId,\n  part,\n}: {\n  chatId: string;\n  messageId: MessageId;\n  part: AiToolInvocationPart;\n}) {\n  const client = useClient();\n  const ai = client[kInternal].ai;\n  const tool = useSignal(ai.signals.getToolDefinitionΣ(chatId, part.toolName));\n  const respond = useCallback(\n    (result: Json) => {\n      ai.setToolResult(\n        chatId,\n        messageId,\n        part.toolCallId,\n        result\n        // TODO Pass in AiGenerationOptions here?\n      );\n    },\n    [ai, chatId, messageId, part.toolCallId]\n  );\n\n  if (tool === undefined || tool.render === undefined) return null;\n\n  const { type: _, ...rest } = part;\n  return (\n    <div className=\"lb-ai-chat-message-tool\">\n      <tool.render\n        {...rest}\n        respond={\n          // It only makes sense and is safe to call `respond()` in \"executing\" state.\n          part.status === \"executing\" ? respond : noop\n        }\n      />\n    </div>\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * ReasoningPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ReasoningPart({\n  text,\n  isPending,\n  components,\n}: {\n  text: string;\n  isPending: boolean;\n  components: Partial<GlobalComponents> | undefined;\n}) {\n  const [isOpen, setIsOpen] = useState(false);\n  return (\n    <CollapsiblePrimitive.Root\n      className=\"lb-ai-chat-message-collapsible lb-ai-chat-message-reasoning\"\n      open={isOpen}\n      onOpenChange={setIsOpen}\n    >\n      <CollapsiblePrimitive.Trigger\n        className={classNames(\n          \"lb-ai-chat-message-collapsible-trigger\",\n          isPending && \"lb-ai-chat-pending\"\n        )}\n      >\n        {/* TODO: If `isPending` is true, show \"Reasoning…\"/\"Thinking…\", otherwise show \"Reasoned/thought for x seconds\"? */}\n        Reasoning\n        <span className=\"lb-ai-chat-message-collapsible-chevron lb-icon-container\">\n          <ChevronRightIcon />\n        </span>\n      </CollapsiblePrimitive.Trigger>\n\n      <CollapsiblePrimitive.Content className=\"lb-ai-chat-message-collapsible-content\">\n        <TextPart text={text} components={components} />\n      </CollapsiblePrimitive.Content>\n    </CollapsiblePrimitive.Root>\n  );\n}\n"],"names":["BlockTokenComp","BlockTokenCompPrimitive","CollapsiblePrimitive.Root","CollapsiblePrimitive.Trigger","CollapsiblePrimitive.Content"],"mappings":";;;;;;;;;;;;;;;;;;AA8DO,MAAM,sBAAyB,GAAA,IAAA;AAAA,EACpC,UAAA;AAAA,IACE,CAAC,EAAE,OAAS,EAAA,SAAA,EAAW,WAAW,UAAe,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AACzE,MAAM,MAAA,CAAA,GAAI,aAAa,SAAS,CAAA,CAAA;AAEhC,MAAA,IAAI,QAAsB,GAAA,IAAA,CAAA;AAE1B,MAAI,IAAA,OAAA,CAAQ,cAAc,KAAW,CAAA,EAAA;AACnC,QAAA,QAAA,mBACG,GAAA,CAAA,KAAA,EAAA;AAAA,UAAI,SAAU,EAAA,4BAAA;AAAA,UACZ,QAAE,EAAA,CAAA,CAAA,uBAAA;AAAA,SACL,CAAA,CAAA;AAAA,iBAGF,OAAQ,CAAA,MAAA,KAAW,YACnB,IAAA,OAAA,CAAQ,WAAW,eACnB,EAAA;AACA,QAAI,IAAA,OAAA,CAAQ,YAAa,CAAA,MAAA,KAAW,CAAG,EAAA;AACrC,UAAA,QAAA,mBACG,GAAA,CAAA,KAAA,EAAA;AAAA,YAAI,SAAU,EAAA,gDAAA;AAAA,YACZ,QAAE,EAAA,CAAA,CAAA,wBAAA;AAAA,WACL,CAAA,CAAA;AAAA,SAEG,MAAA;AACL,UAAA,QAAA,mBACG,GAAA,CAAA,uBAAA,EAAA;AAAA,YACC,SAAS,OAAQ,CAAA,YAAA;AAAA,YACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,YAChB,WAAW,OAAQ,CAAA,EAAA;AAAA,YACnB,UAAA;AAAA,WACF,CAAA,CAAA;AAAA,SAEJ;AAAA,OACF,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,WAAa,EAAA;AACzC,QAAA,QAAA,mBACG,GAAA,CAAA,uBAAA,EAAA;AAAA,UACC,SAAS,OAAQ,CAAA,OAAA;AAAA,UACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,UAChB,WAAW,OAAQ,CAAA,EAAA;AAAA,UACnB,UAAA;AAAA,SACF,CAAA,CAAA;AAAA,OAEJ,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,QAAU,EAAA;AAEtC,QAAI,IAAA,OAAA,CAAQ,gBAAgB,iBAAmB,EAAA;AAC7C,UAAA,QAAA,mBACG,GAAA,CAAA,uBAAA,EAAA;AAAA,YACC,SAAS,OAAQ,CAAA,YAAA;AAAA,YACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,YAChB,WAAW,OAAQ,CAAA,EAAA;AAAA,YACnB,UAAA;AAAA,WACF,CAAA,CAAA;AAAA,SAEG,MAAA;AACL,UACE,QAAA,mBAAA,IAAA,CAAA,QAAA,EAAA;AAAA,YACE,QAAA,EAAA;AAAA,8BAAC,GAAA,CAAA,uBAAA,EAAA;AAAA,gBACC,SAAS,OAAQ,CAAA,YAAA;AAAA,gBACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,gBAChB,WAAW,OAAQ,CAAA,EAAA;AAAA,gBACnB,UAAA;AAAA,eACF,CAAA;AAAA,8BAEC,IAAA,CAAA,KAAA,EAAA;AAAA,gBAAI,SAAU,EAAA,0BAAA;AAAA,gBACb,QAAA,EAAA;AAAA,kCAAC,GAAA,CAAA,MAAA,EAAA;AAAA,oBAAK,SAAU,EAAA,mBAAA;AAAA,oBACd,8BAAC,WAAY,EAAA,EAAA,CAAA;AAAA,mBACf,CAAA;AAAA,kBACC,OAAQ,CAAA,WAAA;AAAA,iBAAA;AAAA,eACX,CAAA;AAAA,aAAA;AAAA,WACF,CAAA,CAAA;AAAA,SAEJ;AAAA,OACF;AAEA,MAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,QACC,SAAW,EAAA,UAAA;AAAA,UACT,iDAAA;AAAA,UACA,SAAA;AAAA,SACF;AAAA,QACC,GAAG,KAAA;AAAA,QACJ,GAAK,EAAA,YAAA;AAAA,QAEJ,QAAA;AAAA,OACH,CAAA,CAAA;AAAA,KAEJ;AAAA,GACF;AACF,EAAA;AAEA,SAAS,uBAAwB,CAAA;AAAA,EAC/B,OAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,UAAA;AACF,CAKG,EAAA;AAED,EAAA,MAAM,WACJ,GAAA,OAAA,CAAQ,IAAK,CAAA,CAAC,SAAS,IAAK,CAAA,IAAA,KAAS,WAAW,CAAA,IAChD,QAAQ,KAAM,CAAA,CAAC,IAAS,KAAA,IAAA,CAAK,SAAS,WAAW,CAAA,CAAA;AAEnD,EAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,4BAAA;AAAA,IACZ,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAC5B,MAAA,QAAQ,KAAK,IAAM;AAAA,QACjB,KAAK,MAAQ,EAAA;AACX,UAAA,uBACG,GAAA,CAAA,QAAA,EAAA;AAAA,YAEC,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,UAAA;AAAA,YACA,SAAU,EAAA,yBAAA;AAAA,WAAA,EAHL,KAIP,CAAA,CAAA;AAAA,SAEJ;AAAA,QACA,KAAK,iBAAmB,EAAA;AACtB,UAAA,uBACG,GAAA,CAAA,kBAAA,EAAA;AAAA,YAEC,MAAA;AAAA,YACA,SAAA;AAAA,YACA,IAAA;AAAA,WAAA,EAHK,KAIP,CAAA,CAAA;AAAA,SAEJ;AAAA,QACA,KAAK,WAAa,EAAA;AAChB,UAAA,uBACG,GAAA,CAAA,aAAA,EAAA;AAAA,YAEC,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,SAAW,EAAA,WAAA;AAAA,YACX,UAAA;AAAA,WAAA,EAHK,KAIP,CAAA,CAAA;AAAA,SAEJ;AAAA,QACA,SAAS;AACP,UAAO,OAAA,IAAA,CAAA;AAAA,SACT;AAAA,OACF;AAAA,KACD,CAAA;AAAA,GACH,CAAA,CAAA;AAEJ,CAAA;AAUA,MAAM,QAAW,GAAA,UAAA;AAAA,EACf,CAAC,EAAE,IAAA,EAAM,UAAe,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AAChD,IAAM,MAAA,MAAA,GAAS,QAAQ,MAAM;AAC3B,MAAA,OAAO,IAAI,KAAA,EAAQ,CAAA,GAAA,CAAI,IAAI,CAAA,CAAA;AAAA,KAC7B,EAAG,CAAC,IAAI,CAAC,CAAA,CAAA;AAET,IAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,MAAI,GAAK,EAAA,YAAA;AAAA,MAAe,GAAG,KAAA;AAAA,MACzB,QAAO,EAAA,MAAA,CAAA,GAAA,CAAI,CAAC,KAAA,EAAO,KAAU,KAAA;AAC5B,QAAA,uBACG,GAAA,CAAA,sBAAA,EAAA;AAAA,UACC,KAAA;AAAA,UAEA,UAAA;AAAA,SAAA,EADK,KAEP,CAAA,CAAA;AAAA,OAEH,CAAA;AAAA,KACH,CAAA,CAAA;AAAA,GAEJ;AACF,CAAA,CAAA;AAGA,SAAS,SAAU,CAAA;AAAA,EACjB,QAAA;AAAA,EACA,IAAA;AACF,CAAoD,EAAA;AAClD,EAAA,MAAM,CAAC,QAAA,EAAU,SAAS,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AAC5C,EAAM,MAAA,UAAA,GAAa,OAA6C,IAAI,CAAA,CAAA;AAEpE,EAAA,SAAA,CAAU,MAAM;AACd,IAAA,IAAI,QAAU,EAAA;AACZ,MAAW,UAAA,CAAA,OAAA,GAAU,WAAW,MAAM;AACpC,QAAA,SAAA,CAAU,KAAK,CAAA,CAAA;AAAA,SACd,GAAI,CAAA,CAAA;AAAA,KACT;AAEA,IAAA,OAAO,MAAM;AACX,MAAA,IAAI,WAAW,OAAS,EAAA;AACtB,QAAA,YAAA,CAAa,WAAW,OAAO,CAAA,CAAA;AAAA,OACjC;AAAA,KACF,CAAA;AAAA,GACF,EAAG,CAAC,QAAQ,CAAC,CAAA,CAAA;AAEb,EAAA,uBACG,IAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,eAAA;AAAA,IACb,QAAA,EAAA;AAAA,sBAAC,IAAA,CAAA,KAAA,EAAA;AAAA,QAAI,SAAU,EAAA,sBAAA;AAAA,QACb,QAAA,EAAA;AAAA,0BAAC,GAAA,CAAA,MAAA,EAAA;AAAA,YAAK,SAAU,EAAA,qBAAA;AAAA,YAAuB,QAAY,EAAA,QAAA,IAAA,YAAA;AAAA,WAAa,CAAA;AAAA,0BAC/D,GAAA,CAAA,KAAA,EAAA;AAAA,YAAI,SAAU,EAAA,8BAAA;AAAA,YACb,QAAC,kBAAA,GAAA,CAAA,MAAA,EAAA;AAAA,cACC,SAAU,EAAA,6BAAA;AAAA,cACV,MAAM,QAAW,mBAAA,GAAA,CAAC,SAAU,EAAA,EAAA,CAAA,uBAAM,QAAS,EAAA,EAAA,CAAA;AAAA,cAC3C,SAAS,MAAM;AACb,gBAAA,SAAA,CAAU,IAAI,CAAA,CAAA;AACd,gBAAU,SAAA,CAAA,SAAA,CAAU,UAAU,IAAI,CAAA,CAAA;AAAA,eACpC;AAAA,aACF,CAAA;AAAA,WACF,CAAA;AAAA,SAAA;AAAA,OACF,CAAA;AAAA,sBACC,GAAA,CAAA,KAAA,EAAA;AAAA,QAAI,SAAU,EAAA,uBAAA;AAAA,QACb,QAAC,kBAAA,GAAA,CAAA,MAAA,EAAA;AAAA,UAAM,QAAA,EAAA,IAAA;AAAA,SAAK,CAAA;AAAA,OACd,CAAA;AAAA,KAAA;AAAA,GACF,CAAA,CAAA;AAEJ,CAAA;AAEA,MAAM,sBAAyB,GAAA,IAAA;AAAA,EAC7B,SAASA,gBAAe,CAAA;AAAA,IACtB,KAAA;AAAA,IACA,UAAA;AAAA,GAIC,EAAA;AACD,IAAA,MAAM,EAAE,MAAA,EAAW,GAAA,aAAA,CAAc,UAAU,CAAA,CAAA;AAE3C,IAAA,uBACG,GAAA,CAAAC,cAAA,EAAA;AAAA,MACC,KAAA;AAAA,MACA,UAAA,EAAY,EAAE,SAAA,EAAW,MAAO,EAAA;AAAA,KAClC,CAAA,CAAA;AAAA,GAEJ;AAAA,EACA,CAAC,WAAW,SAAc,KAAA;AACxB,IAAA,MAAM,YAAY,SAAU,CAAA,KAAA,CAAA;AAC5B,IAAA,MAAM,YAAY,SAAU,CAAA,KAAA,CAAA;AAC5B,IAAA,IAAI,SAAU,CAAA,GAAA,CAAI,MAAW,KAAA,SAAA,CAAU,IAAI,MAAQ,EAAA;AACjD,MAAO,OAAA,KAAA,CAAA;AAAA,KACT;AACA,IAAI,IAAA,SAAA,CAAU,IAAS,KAAA,SAAA,CAAU,IAAM,EAAA;AACrC,MAAO,OAAA,KAAA,CAAA;AAAA,KACT;AACA,IAAO,OAAA,SAAA,CAAU,QAAQ,SAAU,CAAA,GAAA,CAAA;AAAA,GACrC;AACF,CAAA,CAAA;AAEA,SAAS,IAAO,GAAA;AAEhB,CAAA;AAKA,SAAS,kBAAmB,CAAA;AAAA,EAC1B,MAAA;AAAA,EACA,SAAA;AAAA,EACA,IAAA;AACF,CAIG,EAAA;AACD,EAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AACzB,EAAM,MAAA,EAAA,GAAK,OAAO,SAAW,CAAA,CAAA,EAAA,CAAA;AAC7B,EAAM,MAAA,IAAA,GAAO,UAAU,EAAG,CAAA,OAAA,CAAQ,wBAAmB,MAAQ,EAAA,IAAA,CAAK,QAAQ,CAAC,CAAA,CAAA;AAC3E,EAAA,MAAM,OAAU,GAAA,WAAA;AAAA,IACd,CAAC,MAAiB,KAAA;AAChB,MAAG,EAAA,CAAA,aAAA;AAAA,QACD,MAAA;AAAA,QACA,SAAA;AAAA,QACA,IAAK,CAAA,UAAA;AAAA,QACL,MAAA;AAAA,OAEF,CAAA;AAAA,KACF;AAAA,IACA,CAAC,EAAA,EAAI,MAAQ,EAAA,SAAA,EAAW,KAAK,UAAU,CAAA;AAAA,GACzC,CAAA;AAEA,EAAI,IAAA,IAAA,KAAS,KAAa,CAAA,IAAA,IAAA,CAAK,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AAE5D,EAAA,MAAM,EAAE,IAAA,EAAM,CAAM,EAAA,GAAA,IAAA,EAAS,GAAA,IAAA,CAAA;AAC7B,EAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,yBAAA;AAAA,IACb,QAAA,kBAAA,GAAA,CAAC,KAAK,MAAL,EAAA;AAAA,MACE,GAAG,IAAA;AAAA,MACJ,OAEE,EAAA,IAAA,CAAK,MAAW,KAAA,WAAA,GAAc,OAAU,GAAA,IAAA;AAAA,KAE5C,CAAA;AAAA,GACF,CAAA,CAAA;AAEJ,CAAA;AAKA,SAAS,aAAc,CAAA;AAAA,EACrB,IAAA;AAAA,EACA,SAAA;AAAA,EACA,UAAA;AACF,CAIG,EAAA;AACD,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AAC1C,EACE,uBAAA,IAAA,CAACC,IAAA,EAAA;AAAA,IACC,SAAU,EAAA,6DAAA;AAAA,IACV,IAAM,EAAA,MAAA;AAAA,IACN,YAAc,EAAA,SAAA;AAAA,IAEd,QAAA,EAAA;AAAA,sBAAA,IAAA,CAACC,OAAA,EAAA;AAAA,QACC,SAAW,EAAA,UAAA;AAAA,UACT,wCAAA;AAAA,UACA,SAAa,IAAA,oBAAA;AAAA,SACf;AAAA,QAEqH,QAAA,EAAA;AAAA,UAAA,WAAA;AAAA,0BAEpH,GAAA,CAAA,MAAA,EAAA;AAAA,YAAK,SAAU,EAAA,0DAAA;AAAA,YACd,8BAAC,gBAAiB,EAAA,EAAA,CAAA;AAAA,WACpB,CAAA;AAAA,SAAA;AAAA,OACF,CAAA;AAAA,sBAEA,GAAA,CAACC,OAAA,EAAA;AAAA,QAA6B,SAAU,EAAA,wCAAA;AAAA,QACtC,QAAC,kBAAA,GAAA,CAAA,QAAA,EAAA;AAAA,UAAS,IAAA;AAAA,UAAY,UAAA;AAAA,SAAwB,CAAA;AAAA,OAChD,CAAA;AAAA,KAAA;AAAA,GACF,CAAA,CAAA;AAEJ;;;;"}

package/dist/components/internal/AiChatComposer.cjs

@@ -26,6 +26,7 @@ const AiChatComposer = react.forwardRef(
     branchId,
     copilotId,
     stream = true,
+    onUserMessageCreate,
     ...props
   }, forwardedRef) => {
     const $ = overrides.useOverrides(overrides$1);
@@ -36,32 +37,43 @@ const AiChatComposer = react.forwardRef(
         return null;
       return lastMessage.id;
     }, []);
-    const getPendingMessage = react.useCallback((messages) => {
+    const getAbortableMessageId = react.useCallback((messages) => {
       return messages.find(
-        (m) => m.role === "assistant" && m.status === "pending"
+        (m) => m.role === "assistant" && (m.status === "generating" || m.status === "awaiting-tool")
       )?.id;
     }, []);
-    const pendingMessage = _private.useSignal(
-      client[core.kInternal].ai.signals.getChatMessagesForBranch\u03A3(chatId, branchId),
-      getPendingMessage
-    );
-    const lastMessageId = _private.useSignal(
-      client[core.kInternal].ai.signals.getChatMessagesForBranch\u03A3(chatId, branchId),
-      getLastMessageId
+    const messages\u03A3 = client[core.kInternal].ai.signals.getChatMessagesForBranch\u03A3(
+      chatId,
+      branchId
     );
+    const abortableMessageId = _private.useSignal(messages\u03A3, getAbortableMessageId);
+    const lastMessageId = _private.useSignal(messages\u03A3, getLastMessageId);
     const handleComposerSubmit = react.useCallback(
       (message, event) => {
-        if (pendingMessage !== void 0) {
+        if (abortableMessageId !== void 0) {
           event.preventDefault();
           return;
         }
         onComposerSubmit?.(message, event);
         if (event.isDefaultPrevented())
           return;
-        client[core.kInternal].ai.addUserMessageAndAsk(
+        const content = [{ type: "text", text: message.text }];
+        const newMessageId = client[core.kInternal].ai[core.kInternal].context.messagesStore.createOptimistically(
           chatId,
+          "user",
           lastMessageId,
-          message.text,
+          content
+        );
+        onUserMessageCreate?.({ id: newMessageId });
+        const targetMessageId = client[core.kInternal].ai[core.kInternal].context.messagesStore.createOptimistically(
+          chatId,
+          "assistant",
+          newMessageId
+        );
+        client[core.kInternal].ai.askUserMessageInChat(
+          chatId,
+          { id: newMessageId, parentMessageId: lastMessageId, content },
+          targetMessageId,
           {
             stream,
             copilotId
@@ -70,10 +82,11 @@ const AiChatComposer = react.forwardRef(
       },
       [
         onComposerSubmit,
+        onUserMessageCreate,
         client,
         chatId,
         lastMessageId,
-        pendingMessage,
+        abortableMessageId,
         stream,
         copilotId
       ]
@@ -84,7 +97,6 @@ const AiChatComposer = react.forwardRef(
           "lb-ai-chat-composer lb-ai-chat-composer-form",
           className
         ),
-        chatId,
         dir: $.dir,
         ...props,
         disabled,
@@ -107,7 +119,7 @@ const AiChatComposer = react.forwardRef(
                 }),
                 /* @__PURE__ */ jsxRuntime.jsx("div", {
                   className: "lb-ai-chat-composer-actions",
-                  children: pendingMessage === void 0 ? /* @__PURE__ */ jsxRuntime.jsx(Tooltip.ShortcutTooltip, {
+                  children: abortableMessageId === void 0 ? /* @__PURE__ */ jsxRuntime.jsx(Tooltip.ShortcutTooltip, {
                     content: $.AI_CHAT_COMPOSER_SEND,
                     shortcut: "Enter",
                     children: /* @__PURE__ */ jsxRuntime.jsx(index.AiChatComposerSubmit, {
@@ -127,7 +139,7 @@ const AiChatComposer = react.forwardRef(
                       onPointerDown: (event) => event.preventDefault(),
                       onClick: (event) => {
                         event.stopPropagation();
-                        client[core.kInternal].ai.abort(pendingMessage);
+                        client[core.kInternal].ai.abort(abortableMessageId);
                       },
                       className: "lb-ai-chat-composer-action",
                       variant: "secondary",