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

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

@@ -1 +1 @@
-{"version":3,"file":"AiChatAssistantMessage.cjs","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":["memo","forwardRef","overrides","useOverrides","jsxs","jsx","Tooltip","Fragment","WarningIcon","TooltipProvider","classNames","useState","useEffect","Button","CheckIcon","CopyIcon","useClient","kInternal","RetryIcon","useMemo","Lexer","BlockTokenCompPrimitive","useSignal","CollapsiblePrimitive.Root","CollapsiblePrimitive.Trigger","ChevronDownIcon","ChevronRightIcon","CollapsiblePrimitive.Content"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAqEO,MAAM,sBAAyB,GAAAA,UAAA;AAAA,EACpCC,gBAAA;AAAA,IACE,CACE;AAAA,MACE,OAAA;AAAA,MACA,WAAc,GAAA,KAAA;AAAA,MACd,cAAiB,GAAA,KAAA;AAAA,MACjB,SAAA;AAAA,MACA,SAAA;AAAA,iBACAC,WAAA;AAAA,MACG,GAAA,KAAA;AAAA,OAEL,YACG,KAAA;AACH,MAAM,MAAA,CAAA,GAAIC,uBAAaD,WAAS,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,uBACGE,eAAA,CAAA,KAAA,EAAA;AAAA,UAAI,SAAU,EAAA,4BAAA;AAAA,UACb,QAAA,EAAA;AAAA,4BAACC,cAAA,CAAAC,eAAA,EAAA;AAAA,cAAQ,SAAS,CAAE,CAAA,oBAAA;AAAA,cAClB,QAAC,kBAAAD,cAAA,CAAA,cAAA,EAAA;AAAA,gBAAe,IAAA;AAAA,gBAAY,OAAO,CAAE,CAAA,oBAAA;AAAA,eAAsB,CAAA;AAAA,aAC7D,CAAA;AAAA,YAEC,kCACEA,cAAA,CAAAC,eAAA,EAAA;AAAA,cAAQ,SAAS,CAAE,CAAA,yBAAA;AAAA,cAClB,QAAC,kBAAAD,cAAA,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,mBACGA,cAAA,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,mBACGA,cAAA,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,mBACGA,cAAA,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,mBAAAD,eAAA,CAAAG,mBAAA,EAAA;AAAA,UACE,QAAA,EAAA;AAAA,4BAACF,cAAA,CAAA,uBAAA,EAAA;AAAA,cACC,SAAS,OAAQ,CAAA,OAAA;AAAA,cACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,aAClB,CAAA;AAAA,4BAECA,cAAA,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,mBAAAD,eAAA,CAAAG,mBAAA,EAAA;AAAA,YACE,QAAA,EAAA;AAAA,8BAACF,cAAA,CAAA,uBAAA,EAAA;AAAA,gBACC,SAAS,OAAQ,CAAA,YAAA;AAAA,gBACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,eAClB,CAAA;AAAA,8BACCA,cAAA,CAAA,cAAA,EAAA;AAAA,gBAAe,IAAA;AAAA,eAAY,CAAA;AAAA,aAAA;AAAA,WAC9B,CAAA,CAAA;AAAA,SAEG,MAAA;AACL,UACE,QAAA,mBAAAD,eAAA,CAAAG,mBAAA,EAAA;AAAA,YACE,QAAA,EAAA;AAAA,8BAACF,cAAA,CAAA,uBAAA,EAAA;AAAA,gBACC,SAAS,OAAQ,CAAA,YAAA;AAAA,gBACjB,QAAQ,OAAQ,CAAA,MAAA;AAAA,eAClB,CAAA;AAAA,8BAECD,eAAA,CAAA,KAAA,EAAA;AAAA,gBAAI,SAAU,EAAA,0BAAA;AAAA,gBACb,QAAA,EAAA;AAAA,kCAACC,cAAA,CAAA,MAAA,EAAA;AAAA,oBAAK,SAAU,EAAA,mBAAA;AAAA,oBACd,yCAACG,mBAAY,EAAA,EAAA,CAAA;AAAA,mBACf,CAAA;AAAA,kBACC,OAAQ,CAAA,WAAA;AAAA,iBAAA;AAAA,eACX,CAAA;AAAA,8BAECH,cAAA,CAAA,cAAA,EAAA;AAAA,gBAAe,IAAA;AAAA,eAAY,CAAA;AAAA,aAAA;AAAA,WAC9B,CAAA,CAAA;AAAA,SAEJ;AAAA,OACF;AAEA,MAAA,uBACGA,cAAA,CAAAI,gCAAA,EAAA;AAAA,QACC,QAAC,kBAAAJ,cAAA,CAAA,KAAA,EAAA;AAAA,UACC,SAAW,EAAAK,qBAAA;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,GAAIC,eAAS,KAAK,CAAA,CAAA;AAE9C,EAAAC,eAAA,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,uBACGP,cAAA,CAAAQ,aAAA,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,mBAAAR,cAAA,CAACS,eAAU,EAAA,EAAA,CAAA,kCAAMC,aAAS,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,SAASC,iBAAU,EAAA,CAAA;AAEzB,EAAA,uBACGX,cAAA,CAAAQ,aAAA,EAAA;AAAA,IACC,SAAS,WAAY;AACnB,MAAA,MAAA,CAAOI,cAAW,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,iCAAOC,eAAU,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,uBACGb,cAAA,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,uBACGA,cAAA,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,uBACGA,cAAA,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,uBACGA,cAAA,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,GAAAJ,gBAAA;AAAA,EACf,CAAC,EAAE,IAAS,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AACpC,IAAM,MAAA,MAAA,GAASkB,cAAQ,MAAM;AAC3B,MAAA,OAAO,IAAIC,YAAA,EAAQ,CAAA,GAAA,CAAI,IAAI,CAAA,CAAA;AAAA,KAC7B,EAAG,CAAC,IAAI,CAAC,CAAA,CAAA;AAET,IAAA,uBACGf,cAAA,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,uBACGA,cAAA,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,GAAAL,UAAA;AAAA,EAC7B,SAAS,cAAA,CAAe,EAAE,KAAA,EAAgC,EAAA;AACxD,IAAA,uBAAQK,cAAA,CAAAgB,uBAAA,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,SAASL,iBAAU,EAAA,CAAA;AAEzB,EAAA,MAAM,IAAO,GAAAM,kBAAA;AAAA,IACX,OAAOL,cAAW,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,uBACGZ,cAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,yBAAA;AAAA,IACb,QAAA,kBAAAA,cAAA,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,GAAIM,eAAS,KAAK,CAAA,CAAA;AAC1C,EACE,uBAAAP,eAAA,CAACmB,UAAA,EAAA;AAAA,IACC,SAAU,EAAA,6DAAA;AAAA,IACV,IAAM,EAAA,MAAA;AAAA,IACN,YAAc,EAAA,SAAA;AAAA,IAEd,QAAA,EAAA;AAAA,sBAAAnB,eAAA,CAACoB,aAAA,EAAA;AAAA,QACC,SAAW,EAAAd,qBAAA;AAAA,UACT,wCAAA;AAAA,UACA,SAAa,IAAA,oBAAA;AAAA,SACf;AAAA,QACD,QAAA,EAAA;AAAA,UAAA,WAAA;AAAA,0BAEEL,cAAA,CAAA,MAAA,EAAA;AAAA,YAAK,SAAU,EAAA,mBAAA;AAAA,YACb,QAAS,EAAA,MAAA,mBAAAA,cAAA,CAACoB,2BAAgB,EAAA,EAAA,CAAA,kCAAMC,6BAAiB,EAAA,EAAA,CAAA;AAAA,WACpD,CAAA;AAAA,SAAA;AAAA,OACF,CAAA;AAAA,sBAEArB,cAAA,CAACsB,aAAA,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.cjs","sources":["../../../src/components/internal/AiChatAssistantMessage.tsx"],"sourcesContent":["import type { UiAssistantMessage } from \"@liveblocks/core\";\nimport {\n  type ComponentProps,\n  forwardRef,\n  memo,\n  type PropsWithChildren,\n  type ReactNode,\n  useState,\n} from \"react\";\n\nimport { ComponentsProvider, type GlobalComponents } from \"../../components\";\nimport { ChevronRightIcon } from \"../../icons/ChevronRight\";\nimport { WarningIcon } from \"../../icons/Warning\";\nimport {\n  type AiChatMessageOverrides,\n  type GlobalOverrides,\n  useOverrides,\n} from \"../../overrides\";\nimport * as AiMessage from \"../../primitives/AiMessage\";\nimport type {\n  AiMessageContentReasoningPartProps,\n  AiMessageContentTextPartProps,\n} from \"../../primitives/AiMessage/types\";\nimport * as Collapsible from \"../../primitives/Collapsible\";\nimport { classNames } from \"../../utils/class-names\";\nimport { Prose } from \"./Prose\";\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 = <AssistantMessageContent message={message} />;\n        }\n      } else if (message.status === \"completed\") {\n        children = <AssistantMessageContent message={message} />;\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 = <AssistantMessageContent message={message} />;\n        } else {\n          children = (\n            <>\n              <AssistantMessageContent message={message} />\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          <ComponentsProvider components={components}>\n            {children}\n          </ComponentsProvider>\n        </div>\n      );\n    }\n  )\n);\n\nfunction AssistantMessageContent({ message }: { message: UiAssistantMessage }) {\n  return (\n    <AiMessage.Content\n      message={message}\n      components={{\n        TextPart,\n        ReasoningPart,\n        ToolInvocationPart,\n      }}\n      className=\"lb-ai-chat-message-content\"\n    />\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * TextPart\n * -----------------------------------------------------------------------------------------------*/\nfunction TextPart({ part }: AiMessageContentTextPartProps) {\n  return <Prose content={part.text} className=\"lb-ai-chat-message-text\" />;\n}\n\n/* -------------------------------------------------------------------------------------------------\n * ReasoningPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ReasoningPart({\n  part,\n  isStreaming,\n}: AiMessageContentReasoningPartProps) {\n  const [isOpen, setIsOpen] = useState(false);\n  return (\n    <Collapsible.Root\n      className=\"lb-collapsible lb-ai-chat-message-reasoning\"\n      open={isOpen}\n      onOpenChange={setIsOpen}\n    >\n      <Collapsible.Trigger\n        className={classNames(\n          \"lb-collapsible-trigger\",\n          isStreaming && \"lb-ai-chat-pending\"\n        )}\n      >\n        {/* TODO: If `isStreaming` is true, show \"Reasoning…\"/\"Thinking…\", otherwise show \"Reasoned/thought for x seconds\"? */}\n        Reasoning\n        <span className=\"lb-collapsible-chevron lb-icon-container\">\n          <ChevronRightIcon />\n        </span>\n      </Collapsible.Trigger>\n\n      <Collapsible.Content className=\"lb-collapsible-content\">\n        <Prose content={part.text} />\n      </Collapsible.Content>\n    </Collapsible.Root>\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * ToolInvocationPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ToolInvocationPart({ children }: PropsWithChildren) {\n  return <div className=\"lb-ai-chat-message-tool-invocation\">{children}</div>;\n}\n"],"names":["memo","forwardRef","overrides","components","useOverrides","jsx","jsxs","Fragment","WarningIcon","classNames","ComponentsProvider","AiMessage.Content","Prose","useState","Collapsible.Root","Collapsible.Trigger","ChevronRightIcon","Collapsible.Content"],"mappings":";;;;;;;;;;;;;AA+CO,MAAM,sBAAyB,GAAAA,UAAA;AAAA,EACpCC,gBAAA;AAAA,IACE,CAAC,EAAE,OAAS,EAAA,SAAA,aAAWC,yBAAWC,YAAe,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AACzE,MAAM,MAAA,CAAA,GAAIC,uBAAaF,WAAS,CAAA,CAAA;AAEhC,MAAA,IAAI,QAAsB,GAAA,IAAA,CAAA;AAE1B,MAAI,IAAA,OAAA,CAAQ,cAAc,KAAW,CAAA,EAAA;AACnC,QAAA,QAAA,mBACGG,cAAA,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,mBACGA,cAAA,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,mBAAYA,cAAA,CAAA,uBAAA,EAAA;AAAA,YAAwB,OAAA;AAAA,WAAkB,CAAA,CAAA;AAAA,SACxD;AAAA,OACF,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,WAAa,EAAA;AACzC,QAAA,QAAA,mBAAYA,cAAA,CAAA,uBAAA,EAAA;AAAA,UAAwB,OAAA;AAAA,SAAkB,CAAA,CAAA;AAAA,OACxD,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,QAAU,EAAA;AAEtC,QAAI,IAAA,OAAA,CAAQ,gBAAgB,iBAAmB,EAAA;AAC7C,UAAA,QAAA,mBAAYA,cAAA,CAAA,uBAAA,EAAA;AAAA,YAAwB,OAAA;AAAA,WAAkB,CAAA,CAAA;AAAA,SACjD,MAAA;AACL,UACE,QAAA,mBAAAC,eAAA,CAAAC,mBAAA,EAAA;AAAA,YACE,QAAA,EAAA;AAAA,8BAACF,cAAA,CAAA,uBAAA,EAAA;AAAA,gBAAwB,OAAA;AAAA,eAAkB,CAAA;AAAA,8BAE1CC,eAAA,CAAA,KAAA,EAAA;AAAA,gBAAI,SAAU,EAAA,0BAAA;AAAA,gBACb,QAAA,EAAA;AAAA,kCAACD,cAAA,CAAA,MAAA,EAAA;AAAA,oBAAK,SAAU,EAAA,mBAAA;AAAA,oBACd,yCAACG,mBAAY,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,uBACGH,cAAA,CAAA,KAAA,EAAA;AAAA,QACC,SAAW,EAAAI,qBAAA;AAAA,UACT,iDAAA;AAAA,UACA,SAAA;AAAA,SACF;AAAA,QACC,GAAG,KAAA;AAAA,QACJ,GAAK,EAAA,YAAA;AAAA,QAEL,QAAC,kBAAAJ,cAAA,CAAAK,6BAAA,EAAA;AAAA,sBAAmBP,YAAA;AAAA,UACjB,QAAA;AAAA,SACH,CAAA;AAAA,OACF,CAAA,CAAA;AAAA,KAEJ;AAAA,GACF;AACF,EAAA;AAEA,SAAS,uBAAA,CAAwB,EAAE,OAAA,EAA4C,EAAA;AAC7E,EACE,uBAAAE,cAAA,CAACM,aAAA,EAAA;AAAA,IACC,OAAA;AAAA,IACA,UAAY,EAAA;AAAA,MACV,QAAA;AAAA,MACA,aAAA;AAAA,MACA,kBAAA;AAAA,KACF;AAAA,IACA,SAAU,EAAA,4BAAA;AAAA,GACZ,CAAA,CAAA;AAEJ,CAAA;AAKA,SAAS,QAAA,CAAS,EAAE,IAAA,EAAuC,EAAA;AACzD,EAAA,uBAAQN,cAAA,CAAAO,WAAA,EAAA;AAAA,IAAM,SAAS,IAAK,CAAA,IAAA;AAAA,IAAM,SAAU,EAAA,yBAAA;AAAA,GAA0B,CAAA,CAAA;AACxE,CAAA;AAKA,SAAS,aAAc,CAAA;AAAA,EACrB,IAAA;AAAA,EACA,WAAA;AACF,CAAuC,EAAA;AACrC,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAIC,eAAS,KAAK,CAAA,CAAA;AAC1C,EACE,uBAAAP,eAAA,CAACQ,YAAA,EAAA;AAAA,IACC,SAAU,EAAA,6CAAA;AAAA,IACV,IAAM,EAAA,MAAA;AAAA,IACN,YAAc,EAAA,SAAA;AAAA,IAEd,QAAA,EAAA;AAAA,sBAAAR,eAAA,CAACS,eAAA,EAAA;AAAA,QACC,SAAW,EAAAN,qBAAA;AAAA,UACT,wBAAA;AAAA,UACA,WAAe,IAAA,oBAAA;AAAA,SACjB;AAAA,QAEuH,QAAA,EAAA;AAAA,UAAA,WAAA;AAAA,0BAEtHJ,cAAA,CAAA,MAAA,EAAA;AAAA,YAAK,SAAU,EAAA,0CAAA;AAAA,YACd,yCAACW,6BAAiB,EAAA,EAAA,CAAA;AAAA,WACpB,CAAA;AAAA,SAAA;AAAA,OACF,CAAA;AAAA,sBAEAX,cAAA,CAACY,eAAA,EAAA;AAAA,QAAoB,SAAU,EAAA,wBAAA;AAAA,QAC7B,QAAC,kBAAAZ,cAAA,CAAAO,WAAA,EAAA;AAAA,UAAM,SAAS,IAAK,CAAA,IAAA;AAAA,SAAM,CAAA;AAAA,OAC7B,CAAA;AAAA,KAAA;AAAA,GACF,CAAA,CAAA;AAEJ,CAAA;AAKA,SAAS,kBAAA,CAAmB,EAAE,QAAA,EAA+B,EAAA;AAC3D,EAAA,uBAAQP,cAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,oCAAA;AAAA,IAAsC,QAAA;AAAA,GAAS,CAAA,CAAA;AACvE;;;;"}

package/dist/components/internal/AiChatAssistantMessage.js

@@ -1,67 +1,25 @@
-import { jsxs, jsx, 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 { CheckIcon } from '../../icons/Check.js';
-import { ChevronDownIcon } from '../../icons/ChevronDown.js';
+import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
+import { memo, forwardRef, useState } from 'react';
+import { ComponentsProvider } from '../../components.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 { Content as AiMessageContent } from '../../primitives/AiMessage/index.js';
+import { Root as CollapsibleRoot, Trigger as CollapsibleTrigger, Content as CollapsibleContent } from '../../primitives/Collapsible/index.js';
 import { classNames } from '../../utils/class-names.js';
-import { TooltipProvider } from '@radix-ui/react-tooltip';
+import { Prose } from './Prose.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",
@@ -69,53 +27,23 @@ const AiChatAssistantMessage = memo(
           });
         } else {
           children = /* @__PURE__ */ jsx(AssistantMessageContent, {
-            content: message.contentSoFar,
-            chatId: message.chatId
+            message
           });
         }
       } 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, {
+          message
         });
       } 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, {
+            message
           });
         } else {
           children = /* @__PURE__ */ jsxs(Fragment, {
             children: [
               /* @__PURE__ */ jsx(AssistantMessageContent, {
-                content: message.contentSoFar,
-                chatId: message.chatId
+                message
               }),
               /* @__PURE__ */ jsxs("div", {
                 className: "lb-ai-chat-message-error",
@@ -126,185 +54,81 @@ 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,
+      return /* @__PURE__ */ jsx("div", {
+        className: classNames(
+          "lb-ai-chat-message lb-ai-chat-assistant-message",
+          className
+        ),
+        ...props,
+        ref: forwardedRef,
+        children: /* @__PURE__ */ jsx(ComponentsProvider, {
+          components,
           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);
+function AssistantMessageContent({ message }) {
+  return /* @__PURE__ */ jsx(AiMessageContent, {
+    message,
+    components: {
+      TextPart,
+      ReasoningPart,
+      ToolInvocationPart
     },
-    className: "lb-ai-chat-message-action",
-    "aria-label": label,
-    icon: isCopied ? /* @__PURE__ */ jsx(CheckIcon, {}) : /* @__PURE__ */ jsx(CopyIcon, {})
+    className: "lb-ai-chat-message-content"
   });
 }
-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
-}) {
-  const isReasoning = content.some((part) => part.type === "reasoning") && content.every((part) => part.type === "reasoning");
-  return /* @__PURE__ */ jsx("div", {
-    className: "lb-ai-chat-message-content",
-    children: content.map((part, index) => {
-      switch (part.type) {
-        case "text": {
-          return /* @__PURE__ */ jsx(TextPart, {
-            text: part.text,
-            className: "lb-ai-chat-message-text"
-          }, index);
-        }
-        case "tool-call": {
-          return /* @__PURE__ */ jsx(ToolCallPart, {
-            chatId,
-            name: part.toolName,
-            args: part.args
-          }, index);
-        }
-        case "reasoning": {
-          return /* @__PURE__ */ jsx(ReasoningPart, {
-            text: part.text,
-            isPending: isReasoning
-          }, index);
-        }
-        default: {
-          return null;
-        }
-      }
-    })
-  });
-}
-const TextPart = forwardRef(
-  ({ text, ...props }, forwardedRef) => {
-    const tokens = useMemo(() => {
-      return new Lexer().lex(text);
-    }, [text]);
-    return /* @__PURE__ */ jsx("div", {
-      ref: forwardedRef,
-      ...props,
-      children: tokens.map((token, index) => {
-        return /* @__PURE__ */ jsx(MemoizedBlockTokenComp, {
-          token
-        }, index);
-      })
-    });
-  }
-);
-const MemoizedBlockTokenComp = memo(
-  function BlockTokenComp$1({ token }) {
-    return /* @__PURE__ */ jsx(BlockTokenComp, {
-      token
-    });
-  },
-  (prevProps, nextProps) => {
-    const prevToken = prevProps.token;
-    const nextToken = nextProps.token;
-    if (prevToken.raw.length !== nextToken.raw.length) {
-      return false;
-    }
-    if (prevToken.type !== nextToken.type) {
-      return false;
-    }
-    return prevToken.raw === nextToken.raw;
-  }
-);
-function ToolCallPart({
-  chatId,
-  name,
-  args
-}) {
-  const client = useClient();
-  const tool = useSignal(
-    client[kInternal].ai.signals.getToolDefinition\u03A3(chatId, name)
-  );
-  if (tool === void 0 || tool.render === void 0)
-    return null;
-  return /* @__PURE__ */ jsx("div", {
-    className: "lb-ai-chat-message-tool",
-    children: /* @__PURE__ */ jsx(tool.render, {
-      args
-    })
+function TextPart({ part }) {
+  return /* @__PURE__ */ jsx(Prose, {
+    content: part.text,
+    className: "lb-ai-chat-message-text"
   });
 }
 function ReasoningPart({
-  text,
-  isPending
+  part,
+  isStreaming
 }) {
   const [isOpen, setIsOpen] = useState(false);
-  return /* @__PURE__ */ jsxs(Root, {
-    className: "lb-ai-chat-message-collapsible lb-ai-chat-message-reasoning",
+  return /* @__PURE__ */ jsxs(CollapsibleRoot, {
+    className: "lb-collapsible lb-ai-chat-message-reasoning",
     open: isOpen,
     onOpenChange: setIsOpen,
     children: [
-      /* @__PURE__ */ jsxs(Trigger, {
+      /* @__PURE__ */ jsxs(CollapsibleTrigger, {
         className: classNames(
-          "lb-ai-chat-message-collapsible-trigger",
-          isPending && "lb-ai-chat-pending"
+          "lb-collapsible-trigger",
+          isStreaming && "lb-ai-chat-pending"
         ),
         children: [
           "Reasoning",
           /* @__PURE__ */ jsx("span", {
-            className: "lb-icon-container",
-            children: isOpen ? /* @__PURE__ */ jsx(ChevronDownIcon, {}) : /* @__PURE__ */ jsx(ChevronRightIcon, {})
+            className: "lb-collapsible-chevron lb-icon-container",
+            children: /* @__PURE__ */ jsx(ChevronRightIcon, {})
           })
         ]
       }),
-      /* @__PURE__ */ jsx(Content, {
-        className: "lb-ai-chat-message-collapsible-content",
-        children: text
+      /* @__PURE__ */ jsx(CollapsibleContent, {
+        className: "lb-collapsible-content",
+        children: /* @__PURE__ */ jsx(Prose, {
+          content: part.text
+        })
       })
     ]
   });
 }
+function ToolInvocationPart({ children }) {
+  return /* @__PURE__ */ jsx("div", {
+    className: "lb-ai-chat-message-tool-invocation",
+    children
+  });
+}
 
 export { AiChatAssistantMessage };
 //# sourceMappingURL=AiChatAssistantMessage.js.map

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 { UiAssistantMessage } from \"@liveblocks/core\";\nimport {\n  type ComponentProps,\n  forwardRef,\n  memo,\n  type PropsWithChildren,\n  type ReactNode,\n  useState,\n} from \"react\";\n\nimport { ComponentsProvider, type GlobalComponents } from \"../../components\";\nimport { ChevronRightIcon } from \"../../icons/ChevronRight\";\nimport { WarningIcon } from \"../../icons/Warning\";\nimport {\n  type AiChatMessageOverrides,\n  type GlobalOverrides,\n  useOverrides,\n} from \"../../overrides\";\nimport * as AiMessage from \"../../primitives/AiMessage\";\nimport type {\n  AiMessageContentReasoningPartProps,\n  AiMessageContentTextPartProps,\n} from \"../../primitives/AiMessage/types\";\nimport * as Collapsible from \"../../primitives/Collapsible\";\nimport { classNames } from \"../../utils/class-names\";\nimport { Prose } from \"./Prose\";\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 = <AssistantMessageContent message={message} />;\n        }\n      } else if (message.status === \"completed\") {\n        children = <AssistantMessageContent message={message} />;\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 = <AssistantMessageContent message={message} />;\n        } else {\n          children = (\n            <>\n              <AssistantMessageContent message={message} />\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          <ComponentsProvider components={components}>\n            {children}\n          </ComponentsProvider>\n        </div>\n      );\n    }\n  )\n);\n\nfunction AssistantMessageContent({ message }: { message: UiAssistantMessage }) {\n  return (\n    <AiMessage.Content\n      message={message}\n      components={{\n        TextPart,\n        ReasoningPart,\n        ToolInvocationPart,\n      }}\n      className=\"lb-ai-chat-message-content\"\n    />\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * TextPart\n * -----------------------------------------------------------------------------------------------*/\nfunction TextPart({ part }: AiMessageContentTextPartProps) {\n  return <Prose content={part.text} className=\"lb-ai-chat-message-text\" />;\n}\n\n/* -------------------------------------------------------------------------------------------------\n * ReasoningPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ReasoningPart({\n  part,\n  isStreaming,\n}: AiMessageContentReasoningPartProps) {\n  const [isOpen, setIsOpen] = useState(false);\n  return (\n    <Collapsible.Root\n      className=\"lb-collapsible lb-ai-chat-message-reasoning\"\n      open={isOpen}\n      onOpenChange={setIsOpen}\n    >\n      <Collapsible.Trigger\n        className={classNames(\n          \"lb-collapsible-trigger\",\n          isStreaming && \"lb-ai-chat-pending\"\n        )}\n      >\n        {/* TODO: If `isStreaming` is true, show \"Reasoning…\"/\"Thinking…\", otherwise show \"Reasoned/thought for x seconds\"? */}\n        Reasoning\n        <span className=\"lb-collapsible-chevron lb-icon-container\">\n          <ChevronRightIcon />\n        </span>\n      </Collapsible.Trigger>\n\n      <Collapsible.Content className=\"lb-collapsible-content\">\n        <Prose content={part.text} />\n      </Collapsible.Content>\n    </Collapsible.Root>\n  );\n}\n\n/* -------------------------------------------------------------------------------------------------\n * ToolInvocationPart\n * -----------------------------------------------------------------------------------------------*/\nfunction ToolInvocationPart({ children }: PropsWithChildren) {\n  return <div className=\"lb-ai-chat-message-tool-invocation\">{children}</div>;\n}\n"],"names":["AiMessage.Content","Collapsible.Root","Collapsible.Trigger","Collapsible.Content"],"mappings":";;;;;;;;;;;AA+CO,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,mBAAY,GAAA,CAAA,uBAAA,EAAA;AAAA,YAAwB,OAAA;AAAA,WAAkB,CAAA,CAAA;AAAA,SACxD;AAAA,OACF,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,WAAa,EAAA;AACzC,QAAA,QAAA,mBAAY,GAAA,CAAA,uBAAA,EAAA;AAAA,UAAwB,OAAA;AAAA,SAAkB,CAAA,CAAA;AAAA,OACxD,MAAA,IAAW,OAAQ,CAAA,MAAA,KAAW,QAAU,EAAA;AAEtC,QAAI,IAAA,OAAA,CAAQ,gBAAgB,iBAAmB,EAAA;AAC7C,UAAA,QAAA,mBAAY,GAAA,CAAA,uBAAA,EAAA;AAAA,YAAwB,OAAA;AAAA,WAAkB,CAAA,CAAA;AAAA,SACjD,MAAA;AACL,UACE,QAAA,mBAAA,IAAA,CAAA,QAAA,EAAA;AAAA,YACE,QAAA,EAAA;AAAA,8BAAC,GAAA,CAAA,uBAAA,EAAA;AAAA,gBAAwB,OAAA;AAAA,eAAkB,CAAA;AAAA,8BAE1C,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,QAEL,QAAC,kBAAA,GAAA,CAAA,kBAAA,EAAA;AAAA,UAAmB,UAAA;AAAA,UACjB,QAAA;AAAA,SACH,CAAA;AAAA,OACF,CAAA,CAAA;AAAA,KAEJ;AAAA,GACF;AACF,EAAA;AAEA,SAAS,uBAAA,CAAwB,EAAE,OAAA,EAA4C,EAAA;AAC7E,EACE,uBAAA,GAAA,CAACA,gBAAA,EAAA;AAAA,IACC,OAAA;AAAA,IACA,UAAY,EAAA;AAAA,MACV,QAAA;AAAA,MACA,aAAA;AAAA,MACA,kBAAA;AAAA,KACF;AAAA,IACA,SAAU,EAAA,4BAAA;AAAA,GACZ,CAAA,CAAA;AAEJ,CAAA;AAKA,SAAS,QAAA,CAAS,EAAE,IAAA,EAAuC,EAAA;AACzD,EAAA,uBAAQ,GAAA,CAAA,KAAA,EAAA;AAAA,IAAM,SAAS,IAAK,CAAA,IAAA;AAAA,IAAM,SAAU,EAAA,yBAAA;AAAA,GAA0B,CAAA,CAAA;AACxE,CAAA;AAKA,SAAS,aAAc,CAAA;AAAA,EACrB,IAAA;AAAA,EACA,WAAA;AACF,CAAuC,EAAA;AACrC,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AAC1C,EACE,uBAAA,IAAA,CAACC,eAAA,EAAA;AAAA,IACC,SAAU,EAAA,6CAAA;AAAA,IACV,IAAM,EAAA,MAAA;AAAA,IACN,YAAc,EAAA,SAAA;AAAA,IAEd,QAAA,EAAA;AAAA,sBAAA,IAAA,CAACC,kBAAA,EAAA;AAAA,QACC,SAAW,EAAA,UAAA;AAAA,UACT,wBAAA;AAAA,UACA,WAAe,IAAA,oBAAA;AAAA,SACjB;AAAA,QAEuH,QAAA,EAAA;AAAA,UAAA,WAAA;AAAA,0BAEtH,GAAA,CAAA,MAAA,EAAA;AAAA,YAAK,SAAU,EAAA,0CAAA;AAAA,YACd,8BAAC,gBAAiB,EAAA,EAAA,CAAA;AAAA,WACpB,CAAA;AAAA,SAAA;AAAA,OACF,CAAA;AAAA,sBAEA,GAAA,CAACC,kBAAA,EAAA;AAAA,QAAoB,SAAU,EAAA,wBAAA;AAAA,QAC7B,QAAC,kBAAA,GAAA,CAAA,KAAA,EAAA;AAAA,UAAM,SAAS,IAAK,CAAA,IAAA;AAAA,SAAM,CAAA;AAAA,OAC7B,CAAA;AAAA,KAAA;AAAA,GACF,CAAA,CAAA;AAEJ,CAAA;AAKA,SAAS,kBAAA,CAAmB,EAAE,QAAA,EAA+B,EAAA;AAC3D,EAAA,uBAAQ,GAAA,CAAA,KAAA,EAAA;AAAA,IAAI,SAAU,EAAA,oCAAA;AAAA,IAAsC,QAAA;AAAA,GAAS,CAAA,CAAA;AACvE;;;;"}

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
       ]
@@ -81,10 +94,9 @@ const AiChatComposer = react.forwardRef(
     return /* @__PURE__ */ jsxRuntime.jsx(TooltipPrimitive.TooltipProvider, {
       children: /* @__PURE__ */ jsxRuntime.jsx(index.AiChatComposerForm, {
         className: classNames.classNames(
-          "lb-ai-chat-composer lb-ai-chat-composer-form",
+          "lb-root 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",