@liveblocks/react-ui 3.9.1-tiptap1 → 3.9.2-tiptap1

package/dist/primitives/AiComposer/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../../../src/primitives/AiComposer/index.tsx"],"sourcesContent":["import {\n  type AiChatMessage,\n  kInternal,\n  Signal,\n  type WithNavigation,\n} from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useLayoutEffect, useSignal } from \"@liveblocks/react/_private\";\nimport { Slot } from \"@radix-ui/react-slot\";\nimport type { FocusEvent, FormEvent, KeyboardEvent, MouseEvent } from \"react\";\nimport {\n  forwardRef,\n  useCallback,\n  useImperativeHandle,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\nimport {\n  createEditor,\n  Editor as SlateEditor,\n  Transforms as SlateTransforms,\n} from \"slate\";\nimport { withHistory } from \"slate-history\";\nimport {\n  Editable,\n  ReactEditor,\n  type RenderPlaceholderProps,\n  Slate,\n  withReact,\n} from \"slate-react\";\n\nimport type { AiComposerBody } from \"../../types\";\nimport { requestSubmit } from \"../../utils/request-submit\";\nimport { useInitial } from \"../../utils/use-initial\";\nimport { withNormalize } from \"../slate/plugins/normalize\";\nimport { isEmpty } from \"../slate/utils/is-empty\";\nimport {\n  AiComposerContext,\n  AiComposerEditorContext,\n  useAiComposer,\n  useAiComposerEditorContext,\n} from \"./contexts\";\nimport type {\n  AiComposerEditorProps,\n  AiComposerFormProps,\n  AiComposerSubmitProps,\n} from \"./types\";\n\nconst AI_COMPOSER_SUBMIT_NAME = \"AiComposerSubmit\";\nconst AI_COMPOSER_ABORT_NAME = \"AiComposerAbort\";\nconst AI_COMPOSER_EDITOR_NAME = \"AiComposerEditor\";\nconst AI_COMPOSER_FORM_NAME = \"AiComposerForm\";\n\ntype UiChatMessage = WithNavigation<AiChatMessage>;\n\n/* -------------------------------------------------------------------------------------------------\n * Form\n * -----------------------------------------------------------------------------------------------*/\n\nconst emptyMessagesΣ = new Signal<UiChatMessage[]>([]);\n\nfunction getLastMessageId(messages: UiChatMessage[]) {\n  const lastMessage = messages[messages.length - 1];\n\n  if (lastMessage === undefined) {\n    return null;\n  }\n\n  return lastMessage.id;\n}\n\nfunction getAbortableMessageId(messages: UiChatMessage[]) {\n  return messages.find(\n    (message) =>\n      message.role === \"assistant\" &&\n      (message.status === \"generating\" || message.status === \"awaiting-tool\")\n  )?.id;\n}\n\n/**\n * Surrounds the AI composer's content and handles submissions.\n *\n * @example\n * <AiComposer.Form onComposerSubmit={({ text }) => {}}>\n *\t <AiComposer.Editor />\n *   <AiComposer.Submit />\n * </AiComposer.Form>\n */\nexport const AiComposerForm = forwardRef<HTMLFormElement, AiComposerFormProps>(\n  (\n    {\n      onComposerSubmit,\n      onSubmit,\n      disabled,\n      chatId,\n      branchId,\n      asChild,\n      ...props\n    },\n    forwardedRef\n  ) => {\n    const Component = asChild ? Slot : \"form\";\n    const client = useClient();\n    const formRef = useRef<HTMLFormElement | null>(null);\n    const editor = useInitial(() =>\n      withNormalize(withHistory(withReact(createEditor())))\n    );\n    const [isEditorEmpty, setEditorEmpty] = useState(true);\n    const [isSubmitting, setSubmitting] = useState(false);\n    const [isFocused, setFocused] = useState(false);\n    const messagesΣ = chatId\n      ? client[kInternal].ai.signals.getChatMessagesForBranchΣ(chatId, branchId)\n      : emptyMessagesΣ;\n    const lastMessageId = useSignal(messagesΣ, getLastMessageId);\n    const abortableMessageId = useSignal(messagesΣ, getAbortableMessageId);\n    const isAvailable = useSignal(\n      // Subscribe to connection status signal\n      client[kInternal].ai.signals.statusΣ,\n      // \"Disconnected\" means the AI service is not available\n      // as it represents a final error status.\n      (status) => status !== \"disconnected\"\n    );\n\n    const isDisabled = isSubmitting || disabled === true;\n\n    const canAbort = isAvailable && abortableMessageId !== undefined;\n    const canSubmit = isAvailable && !isEditorEmpty && !canAbort;\n\n    const clear = useCallback(() => {\n      SlateTransforms.delete(editor, {\n        at: {\n          anchor: SlateEditor.start(editor, []),\n          focus: SlateEditor.end(editor, []),\n        },\n      });\n    }, [editor]);\n\n    const select = useCallback(() => {\n      SlateTransforms.select(editor, SlateEditor.end(editor, []));\n    }, [editor]);\n\n    const focus = useCallback(\n      (resetSelection = true) => {\n        try {\n          if (!ReactEditor.isFocused(editor)) {\n            SlateTransforms.select(\n              editor,\n              resetSelection || !editor.selection\n                ? SlateEditor.end(editor, [])\n                : editor.selection\n            );\n            ReactEditor.focus(editor);\n          }\n        } catch {\n          // Slate's DOM-specific methods will throw if the editor's DOM\n          // node no longer exists. This action doesn't make sense on an\n          // unmounted editor so we can safely ignore it.\n        }\n      },\n      [editor]\n    );\n\n    const blur = useCallback(() => {\n      try {\n        ReactEditor.blur(editor);\n      } catch {\n        // Slate's DOM-specific methods will throw if the editor's DOM\n        // node no longer exists. This action doesn't make sense on an\n        // unmounted editor so we can safely ignore it.\n      }\n    }, [editor]);\n\n    const onSubmitEnd = useCallback(() => {\n      clear();\n      setSubmitting(false);\n    }, [clear]);\n\n    const handleSubmit = useCallback(\n      (event: FormEvent<HTMLFormElement>) => {\n        if (disabled) {\n          return;\n        }\n\n        // In some situations (e.g. pressing Enter while composing diacritics), it's possible\n        // for the form to be submitted as empty even though we already checked whether the\n        // editor was empty when handling the key press.\n        const isEditorEmpty = isEmpty(editor, editor.children);\n\n        // We even prevent the user's `onSubmit` handler from being called if the editor is empty.\n        if (isEditorEmpty) {\n          event.preventDefault();\n\n          return;\n        }\n\n        onSubmit?.(event);\n\n        if (onComposerSubmit === undefined || event.isDefaultPrevented()) {\n          event.preventDefault();\n          return;\n        }\n\n        // Extract the text content from the editor.\n        const content = editor.children\n          .map((block) => {\n            if (\"type\" in block && block.type === \"paragraph\") {\n              return block.children\n                .map((child) => {\n                  if (\"text\" in child) {\n                    return child.text;\n                  }\n                  return \"\";\n                })\n                .join(\"\");\n            }\n            return \"\";\n          })\n          .join(\"\\n\");\n\n        const promise = onComposerSubmit(\n          { text: content, lastMessageId },\n          event\n        );\n\n        event.preventDefault();\n\n        if (promise) {\n          setSubmitting(true);\n          promise.then(onSubmitEnd);\n        } else {\n          onSubmitEnd();\n        }\n      },\n      [disabled, editor, onSubmit, onComposerSubmit, onSubmitEnd, lastMessageId]\n    );\n\n    useLayoutEffect(() => {\n      setEditorEmpty(isEmpty(editor, editor.children));\n    }, [editor]);\n\n    const handleEditorValueChange = useCallback(() => {\n      setEditorEmpty(isEmpty(editor, editor.children));\n    }, [editor]);\n\n    const submit = useCallback(() => {\n      if (!canSubmit) {\n        return;\n      }\n\n      // We need to wait for the next frame in some cases like when composing diacritics,\n      // we want any native handling to be done first while still being handled on `keydown`.\n      requestAnimationFrame(() => {\n        if (formRef.current) {\n          requestSubmit(formRef.current);\n        }\n      });\n    }, [canSubmit]);\n\n    const abort = useCallback(() => {\n      if (!canAbort || !abortableMessageId) {\n        return;\n      }\n\n      client[kInternal].ai.abort(abortableMessageId);\n    }, [canAbort, abortableMessageId, client]);\n\n    useImperativeHandle<HTMLFormElement | null, HTMLFormElement | null>(\n      forwardedRef,\n      () => formRef.current,\n      []\n    );\n\n    return (\n      <AiComposerEditorContext.Provider\n        value={{\n          editor,\n          onEditorValueChange: handleEditorValueChange,\n          abortableMessageId,\n          setFocused,\n        }}\n      >\n        <AiComposerContext.Provider\n          value={{\n            isDisabled,\n            isEmpty: isEditorEmpty,\n            isFocused,\n            canSubmit,\n            canAbort,\n            submit,\n            abort,\n            clear,\n            focus,\n            blur,\n            select,\n          }}\n        >\n          <Component onSubmit={handleSubmit} {...props} ref={formRef} />\n        </AiComposerContext.Provider>\n      </AiComposerEditorContext.Provider>\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Editor\n * -----------------------------------------------------------------------------------------------*/\n\nfunction AiComposerEditorPlaceholder({\n  attributes,\n  children,\n}: RenderPlaceholderProps) {\n  const { opacity: _opacity, ...style } = attributes.style;\n\n  return (\n    <span {...attributes} style={style} data-placeholder=\"\">\n      {children}\n    </span>\n  );\n}\n\n/**\n * Displays the AI composer's editor.\n *\n * @example\n * <AiComposer.Editor placeholder=\"Write a message…\" />\n */\nconst AiComposerEditor = forwardRef<HTMLDivElement, AiComposerEditorProps>(\n  (\n    {\n      defaultValue = \"\",\n      onKeyDown,\n      onFocus,\n      onBlur,\n      disabled,\n      autoFocus,\n      dir,\n      ...props\n    },\n    forwardedRef\n  ) => {\n    const { editor, onEditorValueChange, setFocused } =\n      useAiComposerEditorContext();\n    const {\n      submit,\n      isDisabled: isComposerDisabled,\n      isFocused,\n      focus,\n      blur,\n      select,\n    } = useAiComposer();\n    const isDisabled = disabled || isComposerDisabled;\n\n    const handleKeyDown = useCallback(\n      (event: KeyboardEvent<HTMLDivElement>) => {\n        onKeyDown?.(event);\n        if (event.isDefaultPrevented()) return;\n\n        if (event.key === \"Enter\" && !event.shiftKey) {\n          event.preventDefault();\n          submit();\n        } else if (event.key === \"Enter\" && event.shiftKey) {\n          event.preventDefault();\n          editor.insertBreak();\n        } else if (event.key === \"Escape\") {\n          blur();\n        }\n      },\n      [editor, onKeyDown, submit, blur]\n    );\n\n    const handleFocus = useCallback(\n      (event: FocusEvent<HTMLDivElement>) => {\n        onFocus?.(event);\n\n        if (!event.isDefaultPrevented()) {\n          setFocused(true);\n        }\n      },\n      [onFocus, setFocused]\n    );\n\n    const handleBlur = useCallback(\n      (event: FocusEvent<HTMLDivElement>) => {\n        onBlur?.(event);\n\n        if (!event.isDefaultPrevented()) {\n          setFocused(false);\n        }\n      },\n      [onBlur, setFocused]\n    );\n\n    useImperativeHandle(\n      forwardedRef,\n      () => ReactEditor.toDOMNode(editor, editor) as HTMLDivElement,\n      [editor]\n    );\n\n    // Manually focus the editor when `autoFocus` is true\n    useLayoutEffect(() => {\n      if (autoFocus) {\n        focus();\n      }\n    }, [autoFocus, editor, focus]);\n\n    // Manually add a selection in the editor if the selection\n    // is still empty after being focused\n    useLayoutEffect(() => {\n      if (isFocused && editor.selection === null) {\n        select();\n      }\n    }, [editor, select, isFocused]);\n\n    const initialValue: AiComposerBody = useMemo(() => {\n      return defaultValue\n        .split(\"\\n\")\n        .map((text) => ({ type: \"paragraph\", children: [{ text }] }));\n    }, [defaultValue]);\n\n    return (\n      <Slate\n        editor={editor}\n        initialValue={initialValue}\n        onValueChange={onEditorValueChange}\n      >\n        <Editable\n          dir={dir}\n          enterKeyHint=\"send\"\n          autoCapitalize=\"sentences\"\n          aria-label=\"Composer editor\"\n          onKeyDown={handleKeyDown}\n          onFocus={handleFocus}\n          onBlur={handleBlur}\n          data-focused={isFocused || undefined}\n          data-disabled={isDisabled || undefined}\n          {...props}\n          readOnly={isDisabled}\n          disabled={isDisabled}\n          renderPlaceholder={AiComposerEditorPlaceholder}\n        />\n      </Slate>\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Submit\n * -----------------------------------------------------------------------------------------------*/\n\n/**\n * A button to submit the AI composer's content.\n *\n * @example\n * <AiComposer.Submit>Send</AiComposer.Submit>\n */\nexport const AiComposerSubmit = forwardRef<\n  HTMLButtonElement,\n  AiComposerSubmitProps\n>(({ disabled, asChild, ...props }, forwardedRef) => {\n  const Component = asChild ? Slot : \"button\";\n  const { isDisabled: isComposerDisabled, canSubmit } = useAiComposer();\n  const isDisabled = isComposerDisabled || disabled || !canSubmit;\n\n  return (\n    <Component\n      type=\"submit\"\n      {...props}\n      ref={forwardedRef}\n      disabled={isDisabled}\n    />\n  );\n});\n\n/* -------------------------------------------------------------------------------------------------\n * Abort\n * -----------------------------------------------------------------------------------------------*/\n\n/**\n * A button to abort a response related to the AI composer.\n *\n * @example\n * <AiComposer.Abort>Cancel</AiComposer.Abort>\n */\nexport const AiComposerAbort = forwardRef<\n  HTMLButtonElement,\n  AiComposerSubmitProps\n>(({ disabled, onClick, asChild, ...props }, forwardedRef) => {\n  const Component = asChild ? Slot : \"button\";\n  const { isDisabled: isComposerDisabled, canAbort, abort } = useAiComposer();\n  const isDisabled = isComposerDisabled || disabled || !canAbort;\n\n  const handleClick = useCallback(\n    (event: MouseEvent<HTMLButtonElement>) => {\n      onClick?.(event);\n\n      if (event.isDefaultPrevented()) {\n        return;\n      }\n\n      abort();\n    },\n    [abort, onClick]\n  );\n\n  return (\n    <Component\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={isDisabled}\n      onClick={handleClick}\n    />\n  );\n});\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiComposerEditor.displayName = AI_COMPOSER_EDITOR_NAME;\n  AiComposerForm.displayName = AI_COMPOSER_FORM_NAME;\n  AiComposerSubmit.displayName = AI_COMPOSER_SUBMIT_NAME;\n  AiComposerAbort.displayName = AI_COMPOSER_ABORT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiComposer.*\nexport {\n  AiComposerAbort as Abort,\n  AiComposerEditor as Editor,\n  AiComposerForm as Form,\n  AiComposerSubmit as Submit,\n};\n"],"names":["SlateTransforms","SlateEditor","isEditorEmpty"],"mappings":";;;;;;;;;;;;;;;AAiDA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAChC,MAAM,sBAAyB,GAAA,iBAAA,CAAA;AAC/B,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAChC,MAAM,qBAAwB,GAAA,gBAAA,CAAA;AAQ9B,MAAM,mBAAiB,GAAA,IAAI,MAAwB,CAAA,EAAE,CAAA,CAAA;AAErD,SAAS,iBAAiB,QAA2B,EAAA;AACnD,EAAM,MAAA,WAAA,GAAc,QAAS,CAAA,QAAA,CAAS,MAAS,GAAA,CAAA,CAAA,CAAA;AAE/C,EAAA,IAAI,gBAAgB,KAAW,CAAA,EAAA;AAC7B,IAAO,OAAA,IAAA,CAAA;AAAA,GACT;AAEA,EAAA,OAAO,WAAY,CAAA,EAAA,CAAA;AACrB,CAAA;AAEA,SAAS,sBAAsB,QAA2B,EAAA;AACxD,EAAA,OAAO,QAAS,CAAA,IAAA;AAAA,IACd,CAAC,YACC,OAAQ,CAAA,IAAA,KAAS,gBAChB,OAAQ,CAAA,MAAA,KAAW,YAAgB,IAAA,OAAA,CAAQ,MAAW,KAAA,eAAA,CAAA;AAAA,GACxD,EAAA,EAAA,CAAA;AACL,CAAA;AAWO,MAAM,cAAiB,GAAA,UAAA;AAAA,EAC5B,CACE;AAAA,IACE,gBAAA;AAAA,IACA,QAAA;AAAA,IACA,QAAA;AAAA,IACA,MAAA;AAAA,IACA,QAAA;AAAA,IACA,OAAA;AAAA,IACG,GAAA,KAAA;AAAA,KAEL,YACG,KAAA;AACH,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,MAAA,CAAA;AACnC,IAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AACzB,IAAM,MAAA,OAAA,GAAU,OAA+B,IAAI,CAAA,CAAA;AACnD,IAAA,MAAM,MAAS,GAAA,UAAA;AAAA,MAAW,MACxB,aAAc,CAAA,WAAA,CAAY,UAAU,YAAa,EAAC,CAAC,CAAC,CAAA;AAAA,KACtD,CAAA;AACA,IAAA,MAAM,CAAC,aAAA,EAAe,cAAc,CAAA,GAAI,SAAS,IAAI,CAAA,CAAA;AACrD,IAAA,MAAM,CAAC,YAAA,EAAc,aAAa,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AACpD,IAAA,MAAM,CAAC,SAAA,EAAW,UAAU,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AAC9C,IAAM,MAAA,cAAA,GAAY,SACd,MAAO,CAAA,SAAA,CAAA,CAAW,GAAG,OAAQ,CAAA,8BAAA,CAA0B,MAAQ,EAAA,QAAQ,CACvE,GAAA,mBAAA,CAAA;AACJ,IAAM,MAAA,aAAA,GAAgB,SAAU,CAAA,cAAA,EAAW,gBAAgB,CAAA,CAAA;AAC3D,IAAM,MAAA,kBAAA,GAAqB,SAAU,CAAA,cAAA,EAAW,qBAAqB,CAAA,CAAA;AACrE,IAAA,MAAM,WAAc,GAAA,SAAA;AAAA,MAElB,MAAA,CAAO,SAAW,CAAA,CAAA,EAAA,CAAG,OAAQ,CAAA,YAAA;AAAA,MAG7B,CAAC,WAAW,MAAW,KAAA,cAAA;AAAA,KACzB,CAAA;AAEA,IAAM,MAAA,UAAA,GAAa,gBAAgB,QAAa,KAAA,IAAA,CAAA;AAEhD,IAAM,MAAA,QAAA,GAAW,eAAe,kBAAuB,KAAA,KAAA,CAAA,CAAA;AACvD,IAAA,MAAM,SAAY,GAAA,WAAA,IAAe,CAAC,aAAA,IAAiB,CAAC,QAAA,CAAA;AAEpD,IAAM,MAAA,KAAA,GAAQ,YAAY,MAAM;AAC9B,MAAAA,UAAA,CAAgB,OAAO,MAAQ,EAAA;AAAA,QAC7B,EAAI,EAAA;AAAA,UACF,MAAQ,EAAAC,MAAA,CAAY,KAAM,CAAA,MAAA,EAAQ,EAAE,CAAA;AAAA,UACpC,KAAO,EAAAA,MAAA,CAAY,GAAI,CAAA,MAAA,EAAQ,EAAE,CAAA;AAAA,SACnC;AAAA,OACD,CAAA,CAAA;AAAA,KACH,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,MAAA,GAAS,YAAY,MAAM;AAC/B,MAAAD,UAAA,CAAgB,OAAO,MAAQ,EAAAC,MAAA,CAAY,IAAI,MAAQ,EAAA,EAAE,CAAC,CAAA,CAAA;AAAA,KAC5D,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAA,MAAM,KAAQ,GAAA,WAAA;AAAA,MACZ,CAAC,iBAAiB,IAAS,KAAA;AACzB,QAAI,IAAA;AACF,UAAA,IAAI,CAAC,WAAA,CAAY,SAAU,CAAA,MAAM,CAAG,EAAA;AAClC,YAAgBD,UAAA,CAAA,MAAA;AAAA,cACd,MAAA;AAAA,cACA,cAAA,IAAkB,CAAC,MAAA,CAAO,SACtB,GAAAC,MAAA,CAAY,IAAI,MAAQ,EAAA,EAAE,CAAA,GAC1B,MAAO,CAAA,SAAA;AAAA,aACb,CAAA;AACA,YAAA,WAAA,CAAY,MAAM,MAAM,CAAA,CAAA;AAAA,WAC1B;AAAA,SACA,CAAA,MAAA;AAAA,SAIF;AAAA,OACF;AAAA,MACA,CAAC,MAAM,CAAA;AAAA,KACT,CAAA;AAEA,IAAM,MAAA,IAAA,GAAO,YAAY,MAAM;AAC7B,MAAI,IAAA;AACF,QAAA,WAAA,CAAY,KAAK,MAAM,CAAA,CAAA;AAAA,OACvB,CAAA,MAAA;AAAA,OAIF;AAAA,KACF,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,WAAA,GAAc,YAAY,MAAM;AACpC,MAAM,KAAA,EAAA,CAAA;AACN,MAAA,aAAA,CAAc,KAAK,CAAA,CAAA;AAAA,KACrB,EAAG,CAAC,KAAK,CAAC,CAAA,CAAA;AAEV,IAAA,MAAM,YAAe,GAAA,WAAA;AAAA,MACnB,CAAC,KAAsC,KAAA;AACrC,QAAA,IAAI,QAAU,EAAA;AACZ,UAAA,OAAA;AAAA,SACF;AAKA,QAAA,MAAMC,cAAgB,GAAA,OAAA,CAAQ,MAAQ,EAAA,MAAA,CAAO,QAAQ,CAAA,CAAA;AAGrD,QAAA,IAAIA,cAAe,EAAA;AACjB,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AAErB,UAAA,OAAA;AAAA,SACF;AAEA,QAAA,QAAA,GAAW,KAAK,CAAA,CAAA;AAEhB,QAAA,IAAI,gBAAqB,KAAA,KAAA,CAAA,IAAa,KAAM,CAAA,kBAAA,EAAsB,EAAA;AAChE,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AACrB,UAAA,OAAA;AAAA,SACF;AAGA,QAAA,MAAM,OAAU,GAAA,MAAA,CAAO,QACpB,CAAA,GAAA,CAAI,CAAC,KAAU,KAAA;AACd,UAAA,IAAI,MAAU,IAAA,KAAA,IAAS,KAAM,CAAA,IAAA,KAAS,WAAa,EAAA;AACjD,YAAA,OAAO,KAAM,CAAA,QAAA,CACV,GAAI,CAAA,CAAC,KAAU,KAAA;AACd,cAAA,IAAI,UAAU,KAAO,EAAA;AACnB,gBAAA,OAAO,KAAM,CAAA,IAAA,CAAA;AAAA,eACf;AACA,cAAO,OAAA,EAAA,CAAA;AAAA,aACR,CACA,CAAA,IAAA,CAAK,EAAE,CAAA,CAAA;AAAA,WACZ;AACA,UAAO,OAAA,EAAA,CAAA;AAAA,SACR,CACA,CAAA,IAAA,CAAK,IAAI,CAAA,CAAA;AAEZ,QAAA,MAAM,OAAU,GAAA,gBAAA;AAAA,UACd,EAAE,IAAM,EAAA,OAAA,EAAS,aAAc,EAAA;AAAA,UAC/B,KAAA;AAAA,SACF,CAAA;AAEA,QAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AAErB,QAAA,IAAI,OAAS,EAAA;AACX,UAAA,aAAA,CAAc,IAAI,CAAA,CAAA;AAClB,UAAA,OAAA,CAAQ,KAAK,WAAW,CAAA,CAAA;AAAA,SACnB,MAAA;AACL,UAAY,WAAA,EAAA,CAAA;AAAA,SACd;AAAA,OACF;AAAA,MACA,CAAC,QAAU,EAAA,MAAA,EAAQ,QAAU,EAAA,gBAAA,EAAkB,aAAa,aAAa,CAAA;AAAA,KAC3E,CAAA;AAEA,IAAA,eAAA,CAAgB,MAAM;AACpB,MAAA,cAAA,CAAe,OAAQ,CAAA,MAAA,EAAQ,MAAO,CAAA,QAAQ,CAAC,CAAA,CAAA;AAAA,KACjD,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,uBAAA,GAA0B,YAAY,MAAM;AAChD,MAAA,cAAA,CAAe,OAAQ,CAAA,MAAA,EAAQ,MAAO,CAAA,QAAQ,CAAC,CAAA,CAAA;AAAA,KACjD,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,MAAA,GAAS,YAAY,MAAM;AAC/B,MAAA,IAAI,CAAC,SAAW,EAAA;AACd,QAAA,OAAA;AAAA,OACF;AAIA,MAAA,qBAAA,CAAsB,MAAM;AAC1B,QAAA,IAAI,QAAQ,OAAS,EAAA;AACnB,UAAA,aAAA,CAAc,QAAQ,OAAO,CAAA,CAAA;AAAA,SAC/B;AAAA,OACD,CAAA,CAAA;AAAA,KACH,EAAG,CAAC,SAAS,CAAC,CAAA,CAAA;AAEd,IAAM,MAAA,KAAA,GAAQ,YAAY,MAAM;AAC9B,MAAI,IAAA,CAAC,QAAY,IAAA,CAAC,kBAAoB,EAAA;AACpC,QAAA,OAAA;AAAA,OACF;AAEA,MAAO,MAAA,CAAA,SAAA,CAAA,CAAW,EAAG,CAAA,KAAA,CAAM,kBAAkB,CAAA,CAAA;AAAA,KAC5C,EAAA,CAAC,QAAU,EAAA,kBAAA,EAAoB,MAAM,CAAC,CAAA,CAAA;AAEzC,IAAA,mBAAA;AAAA,MACE,YAAA;AAAA,MACA,MAAM,OAAQ,CAAA,OAAA;AAAA,MACd,EAAC;AAAA,KACH,CAAA;AAEA,IACE,uBAAA,GAAA,CAAC,wBAAwB,QAAxB,EAAA;AAAA,MACC,KAAO,EAAA;AAAA,QACL,MAAA;AAAA,QACA,mBAAqB,EAAA,uBAAA;AAAA,QACrB,kBAAA;AAAA,QACA,UAAA;AAAA,OACF;AAAA,MAEA,QAAA,kBAAA,GAAA,CAAC,kBAAkB,QAAlB,EAAA;AAAA,QACC,KAAO,EAAA;AAAA,UACL,UAAA;AAAA,UACA,OAAS,EAAA,aAAA;AAAA,UACT,SAAA;AAAA,UACA,SAAA;AAAA,UACA,QAAA;AAAA,UACA,MAAA;AAAA,UACA,KAAA;AAAA,UACA,KAAA;AAAA,UACA,KAAA;AAAA,UACA,IAAA;AAAA,UACA,MAAA;AAAA,SACF;AAAA,QAEA,QAAC,kBAAA,GAAA,CAAA,SAAA,EAAA;AAAA,UAAU,QAAU,EAAA,YAAA;AAAA,UAAe,GAAG,KAAA;AAAA,UAAO,GAAK,EAAA,OAAA;AAAA,SAAS,CAAA;AAAA,OAC9D,CAAA;AAAA,KACF,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAMA,SAAS,2BAA4B,CAAA;AAAA,EACnC,UAAA;AAAA,EACA,QAAA;AACF,CAA2B,EAAA;AACzB,EAAA,MAAM,EAAE,OAAA,EAAS,QAAa,EAAA,GAAA,KAAA,KAAU,UAAW,CAAA,KAAA,CAAA;AAEnD,EAAA,uBACG,GAAA,CAAA,MAAA,EAAA;AAAA,IAAM,GAAG,UAAA;AAAA,IAAY,KAAA;AAAA,IAAc,kBAAiB,EAAA,EAAA;AAAA,IAClD,QAAA;AAAA,GACH,CAAA,CAAA;AAEJ,CAAA;AAQA,MAAM,gBAAmB,GAAA,UAAA;AAAA,EACvB,CACE;AAAA,IACE,YAAe,GAAA,EAAA;AAAA,IACf,SAAA;AAAA,IACA,OAAA;AAAA,IACA,MAAA;AAAA,IACA,QAAA;AAAA,IACA,SAAA;AAAA,IACA,GAAA;AAAA,IACG,GAAA,KAAA;AAAA,KAEL,YACG,KAAA;AACH,IAAA,MAAM,EAAE,MAAA,EAAQ,mBAAqB,EAAA,UAAA,KACnC,0BAA2B,EAAA,CAAA;AAC7B,IAAM,MAAA;AAAA,MACJ,MAAA;AAAA,MACA,UAAY,EAAA,kBAAA;AAAA,MACZ,SAAA;AAAA,MACA,KAAA;AAAA,MACA,IAAA;AAAA,MACA,MAAA;AAAA,QACE,aAAc,EAAA,CAAA;AAClB,IAAA,MAAM,aAAa,QAAY,IAAA,kBAAA,CAAA;AAE/B,IAAA,MAAM,aAAgB,GAAA,WAAA;AAAA,MACpB,CAAC,KAAyC,KAAA;AACxC,QAAA,SAAA,GAAY,KAAK,CAAA,CAAA;AACjB,QAAA,IAAI,MAAM,kBAAmB,EAAA;AAAG,UAAA,OAAA;AAEhC,QAAA,IAAI,KAAM,CAAA,GAAA,KAAQ,OAAW,IAAA,CAAC,MAAM,QAAU,EAAA;AAC5C,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AACrB,UAAO,MAAA,EAAA,CAAA;AAAA,SACE,MAAA,IAAA,KAAA,CAAM,GAAQ,KAAA,OAAA,IAAW,MAAM,QAAU,EAAA;AAClD,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AACrB,UAAA,MAAA,CAAO,WAAY,EAAA,CAAA;AAAA,SACrB,MAAA,IAAW,KAAM,CAAA,GAAA,KAAQ,QAAU,EAAA;AACjC,UAAK,IAAA,EAAA,CAAA;AAAA,SACP;AAAA,OACF;AAAA,MACA,CAAC,MAAA,EAAQ,SAAW,EAAA,MAAA,EAAQ,IAAI,CAAA;AAAA,KAClC,CAAA;AAEA,IAAA,MAAM,WAAc,GAAA,WAAA;AAAA,MAClB,CAAC,KAAsC,KAAA;AACrC,QAAA,OAAA,GAAU,KAAK,CAAA,CAAA;AAEf,QAAI,IAAA,CAAC,KAAM,CAAA,kBAAA,EAAsB,EAAA;AAC/B,UAAA,UAAA,CAAW,IAAI,CAAA,CAAA;AAAA,SACjB;AAAA,OACF;AAAA,MACA,CAAC,SAAS,UAAU,CAAA;AAAA,KACtB,CAAA;AAEA,IAAA,MAAM,UAAa,GAAA,WAAA;AAAA,MACjB,CAAC,KAAsC,KAAA;AACrC,QAAA,MAAA,GAAS,KAAK,CAAA,CAAA;AAEd,QAAI,IAAA,CAAC,KAAM,CAAA,kBAAA,EAAsB,EAAA;AAC/B,UAAA,UAAA,CAAW,KAAK,CAAA,CAAA;AAAA,SAClB;AAAA,OACF;AAAA,MACA,CAAC,QAAQ,UAAU,CAAA;AAAA,KACrB,CAAA;AAEA,IAAA,mBAAA;AAAA,MACE,YAAA;AAAA,MACA,MAAM,WAAA,CAAY,SAAU,CAAA,MAAA,EAAQ,MAAM,CAAA;AAAA,MAC1C,CAAC,MAAM,CAAA;AAAA,KACT,CAAA;AAGA,IAAA,eAAA,CAAgB,MAAM;AACpB,MAAA,IAAI,SAAW,EAAA;AACb,QAAM,KAAA,EAAA,CAAA;AAAA,OACR;AAAA,KACC,EAAA,CAAC,SAAW,EAAA,MAAA,EAAQ,KAAK,CAAC,CAAA,CAAA;AAI7B,IAAA,eAAA,CAAgB,MAAM;AACpB,MAAI,IAAA,SAAA,IAAa,MAAO,CAAA,SAAA,KAAc,IAAM,EAAA;AAC1C,QAAO,MAAA,EAAA,CAAA;AAAA,OACT;AAAA,KACC,EAAA,CAAC,MAAQ,EAAA,MAAA,EAAQ,SAAS,CAAC,CAAA,CAAA;AAE9B,IAAM,MAAA,YAAA,GAA+B,QAAQ,MAAM;AACjD,MAAA,OAAO,aACJ,KAAM,CAAA,IAAI,CACV,CAAA,GAAA,CAAI,CAAC,IAAU,MAAA,EAAE,IAAM,EAAA,WAAA,EAAa,UAAU,CAAC,EAAE,IAAK,EAAC,GAAI,CAAA,CAAA,CAAA;AAAA,KAChE,EAAG,CAAC,YAAY,CAAC,CAAA,CAAA;AAEjB,IAAA,uBACG,GAAA,CAAA,KAAA,EAAA;AAAA,MACC,MAAA;AAAA,MACA,YAAA;AAAA,MACA,aAAe,EAAA,mBAAA;AAAA,MAEf,QAAC,kBAAA,GAAA,CAAA,QAAA,EAAA;AAAA,QACC,GAAA;AAAA,QACA,YAAa,EAAA,MAAA;AAAA,QACb,cAAe,EAAA,WAAA;AAAA,QACf,YAAW,EAAA,iBAAA;AAAA,QACX,SAAW,EAAA,aAAA;AAAA,QACX,OAAS,EAAA,WAAA;AAAA,QACT,MAAQ,EAAA,UAAA;AAAA,QACR,gBAAc,SAAa,IAAA,KAAA,CAAA;AAAA,QAC3B,iBAAe,UAAc,IAAA,KAAA,CAAA;AAAA,QAC5B,GAAG,KAAA;AAAA,QACJ,QAAU,EAAA,UAAA;AAAA,QACV,QAAU,EAAA,UAAA;AAAA,QACV,iBAAmB,EAAA,2BAAA;AAAA,OACrB,CAAA;AAAA,KACF,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAYa,MAAA,gBAAA,GAAmB,WAG9B,CAAC,EAAE,UAAU,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AACnD,EAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,QAAA,CAAA;AACnC,EAAA,MAAM,EAAE,UAAA,EAAY,kBAAoB,EAAA,SAAA,KAAc,aAAc,EAAA,CAAA;AACpE,EAAM,MAAA,UAAA,GAAa,kBAAsB,IAAA,QAAA,IAAY,CAAC,SAAA,CAAA;AAEtD,EAAA,uBACG,GAAA,CAAA,SAAA,EAAA;AAAA,IACC,IAAK,EAAA,QAAA;AAAA,IACJ,GAAG,KAAA;AAAA,IACJ,GAAK,EAAA,YAAA;AAAA,IACL,QAAU,EAAA,UAAA;AAAA,GACZ,CAAA,CAAA;AAEJ,CAAC,EAAA;AAYY,MAAA,eAAA,GAAkB,WAG7B,CAAC,EAAE,UAAU,OAAS,EAAA,OAAA,EAAA,GAAY,KAAM,EAAA,EAAG,YAAiB,KAAA;AAC5D,EAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,QAAA,CAAA;AACnC,EAAA,MAAM,EAAE,UAAY,EAAA,kBAAA,EAAoB,QAAU,EAAA,KAAA,KAAU,aAAc,EAAA,CAAA;AAC1E,EAAM,MAAA,UAAA,GAAa,kBAAsB,IAAA,QAAA,IAAY,CAAC,QAAA,CAAA;AAEtD,EAAA,MAAM,WAAc,GAAA,WAAA;AAAA,IAClB,CAAC,KAAyC,KAAA;AACxC,MAAA,OAAA,GAAU,KAAK,CAAA,CAAA;AAEf,MAAI,IAAA,KAAA,CAAM,oBAAsB,EAAA;AAC9B,QAAA,OAAA;AAAA,OACF;AAEA,MAAM,KAAA,EAAA,CAAA;AAAA,KACR;AAAA,IACA,CAAC,OAAO,OAAO,CAAA;AAAA,GACjB,CAAA;AAEA,EAAA,uBACG,GAAA,CAAA,SAAA,EAAA;AAAA,IACC,IAAK,EAAA,QAAA;AAAA,IACJ,GAAG,KAAA;AAAA,IACJ,GAAK,EAAA,YAAA;AAAA,IACL,QAAU,EAAA,UAAA;AAAA,IACV,OAAS,EAAA,WAAA;AAAA,GACX,CAAA,CAAA;AAEJ,CAAC,EAAA;AAED,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AAC/B,EAAA,cAAA,CAAe,WAAc,GAAA,qBAAA,CAAA;AAC7B,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AAC/B,EAAA,eAAA,CAAgB,WAAc,GAAA,sBAAA,CAAA;AAChC;;;;"}
+{"version":3,"file":"index.js","sources":["../../../src/primitives/AiComposer/index.tsx"],"sourcesContent":["import {\n  type AiChatMessage,\n  kInternal,\n  Signal,\n  type WithNavigation,\n} from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useLayoutEffect, useSignal } from \"@liveblocks/react/_private\";\nimport { Slot } from \"@radix-ui/react-slot\";\nimport type { FocusEvent, FormEvent, KeyboardEvent, MouseEvent } from \"react\";\nimport {\n  forwardRef,\n  useCallback,\n  useImperativeHandle,\n  useMemo,\n  useRef,\n  useState,\n} from \"react\";\nimport {\n  createEditor,\n  Editor as SlateEditor,\n  Transforms as SlateTransforms,\n} from \"slate\";\nimport { withHistory } from \"slate-history\";\nimport {\n  Editable,\n  ReactEditor,\n  type RenderPlaceholderProps,\n  Slate,\n  withReact,\n} from \"slate-react\";\n\nimport type { AiComposerBody } from \"../../types\";\nimport { requestSubmit } from \"../../utils/request-submit\";\nimport { useInitial } from \"../../utils/use-initial\";\nimport { withNormalize } from \"../slate/plugins/normalize\";\nimport { isEmpty } from \"../slate/utils/is-empty\";\nimport {\n  AiComposerContext,\n  AiComposerEditorContext,\n  useAiComposer,\n  useAiComposerEditorContext,\n} from \"./contexts\";\nimport type {\n  AiComposerEditorProps,\n  AiComposerFormProps,\n  AiComposerSubmitProps,\n} from \"./types\";\n\nconst AI_COMPOSER_SUBMIT_NAME = \"AiComposerSubmit\";\nconst AI_COMPOSER_ABORT_NAME = \"AiComposerAbort\";\nconst AI_COMPOSER_EDITOR_NAME = \"AiComposerEditor\";\nconst AI_COMPOSER_FORM_NAME = \"AiComposerForm\";\n\ntype UiChatMessage = WithNavigation<AiChatMessage>;\n\n/* -------------------------------------------------------------------------------------------------\n * Form\n * -----------------------------------------------------------------------------------------------*/\n\nconst emptyMessagesΣ = new Signal<UiChatMessage[]>([]);\n\nfunction getLastMessageId(messages: UiChatMessage[]) {\n  const lastMessage = messages[messages.length - 1];\n\n  if (lastMessage === undefined) {\n    return null;\n  }\n\n  return lastMessage.id;\n}\n\nfunction getAbortableMessageId(messages: UiChatMessage[]) {\n  return messages.find(\n    (message) =>\n      message.role === \"assistant\" &&\n      (message.status === \"generating\" || message.status === \"awaiting-tool\")\n  )?.id;\n}\n\n/**\n * Surrounds the AI composer's content and handles submissions.\n *\n * @example\n * <AiComposer.Form onComposerSubmit={({ text }) => {}}>\n *\t <AiComposer.Editor />\n *   <AiComposer.Submit />\n * </AiComposer.Form>\n */\nexport const AiComposerForm = forwardRef<HTMLFormElement, AiComposerFormProps>(\n  (\n    {\n      onComposerSubmit,\n      onSubmit,\n      disabled,\n      chatId,\n      branchId,\n      asChild,\n      ...props\n    },\n    forwardedRef\n  ) => {\n    const Component = asChild ? Slot : \"form\";\n    const client = useClient();\n    const formRef = useRef<HTMLFormElement | null>(null);\n    const editor = useInitial(() =>\n      withNormalize(withHistory(withReact(createEditor())))\n    );\n    const [isEditorEmpty, setEditorEmpty] = useState(true);\n    const [isSubmitting, setSubmitting] = useState(false);\n    const [isFocused, setFocused] = useState(false);\n    const messagesΣ = chatId\n      ? client[kInternal].ai.signals.getChatMessagesForBranchΣ(chatId, branchId)\n      : emptyMessagesΣ;\n    const lastMessageId = useSignal(messagesΣ, getLastMessageId);\n    const abortableMessageId = useSignal(messagesΣ, getAbortableMessageId);\n    const isAvailable = useSignal(\n      // Subscribe to connection status signal\n      client[kInternal].ai.signals.statusΣ,\n      // \"Disconnected\" means the AI service is not available\n      // as it represents a final error status.\n      (status) => status !== \"disconnected\"\n    );\n\n    const isDisabled = isSubmitting || disabled === true;\n\n    const canAbort = isAvailable && abortableMessageId !== undefined;\n    const canSubmit = isAvailable && !isEditorEmpty && !canAbort;\n\n    const clear = useCallback(() => {\n      SlateTransforms.delete(editor, {\n        at: {\n          anchor: SlateEditor.start(editor, []),\n          focus: SlateEditor.end(editor, []),\n        },\n      });\n    }, [editor]);\n\n    const select = useCallback(() => {\n      SlateTransforms.select(editor, SlateEditor.end(editor, []));\n    }, [editor]);\n\n    const focus = useCallback(\n      (resetSelection = true) => {\n        try {\n          if (!ReactEditor.isFocused(editor)) {\n            SlateTransforms.select(\n              editor,\n              resetSelection || !editor.selection\n                ? SlateEditor.end(editor, [])\n                : editor.selection\n            );\n            ReactEditor.focus(editor);\n          }\n        } catch {\n          // Slate's DOM-specific methods will throw if the editor's DOM\n          // node no longer exists. This action doesn't make sense on an\n          // unmounted editor so we can safely ignore it.\n        }\n      },\n      [editor]\n    );\n\n    const blur = useCallback(() => {\n      try {\n        ReactEditor.blur(editor);\n      } catch {\n        // Slate's DOM-specific methods will throw if the editor's DOM\n        // node no longer exists. This action doesn't make sense on an\n        // unmounted editor so we can safely ignore it.\n      }\n    }, [editor]);\n\n    const onSubmitEnd = useCallback(() => {\n      clear();\n      setSubmitting(false);\n    }, [clear]);\n\n    const handleSubmit = useCallback(\n      (event: FormEvent<HTMLFormElement>) => {\n        if (disabled) {\n          return;\n        }\n\n        // In some situations (e.g. pressing Enter while composing diacritics), it's possible\n        // for the form to be submitted as empty even though we already checked whether the\n        // editor was empty when handling the key press.\n        const isEditorEmpty = isEmpty(editor, editor.children);\n\n        // We even prevent the user's `onSubmit` handler from being called if the editor is empty.\n        if (isEditorEmpty) {\n          event.preventDefault();\n\n          return;\n        }\n\n        onSubmit?.(event);\n\n        if (onComposerSubmit === undefined || event.isDefaultPrevented()) {\n          event.preventDefault();\n          return;\n        }\n\n        // Extract the text content from the editor.\n        const content = editor.children\n          .map((block) => {\n            if (\"type\" in block && block.type === \"paragraph\") {\n              return block.children\n                .map((child) => {\n                  if (\"text\" in child) {\n                    return child.text;\n                  }\n                  return \"\";\n                })\n                .join(\"\");\n            }\n            return \"\";\n          })\n          .join(\"\\n\");\n\n        const promise = onComposerSubmit(\n          { text: content, lastMessageId },\n          event\n        );\n\n        event.preventDefault();\n\n        if (promise) {\n          setSubmitting(true);\n          promise.then(onSubmitEnd);\n        } else {\n          onSubmitEnd();\n        }\n      },\n      [disabled, editor, onSubmit, onComposerSubmit, onSubmitEnd, lastMessageId]\n    );\n\n    useLayoutEffect(() => {\n      setEditorEmpty(isEmpty(editor, editor.children));\n    }, [editor]);\n\n    const handleEditorValueChange = useCallback(() => {\n      setEditorEmpty(isEmpty(editor, editor.children));\n    }, [editor]);\n\n    const submit = useCallback(() => {\n      if (!canSubmit) {\n        return;\n      }\n\n      // We need to wait for the next frame in some cases like when composing diacritics,\n      // we want any native handling to be done first while still being handled on `keydown`.\n      requestAnimationFrame(() => {\n        if (formRef.current) {\n          requestSubmit(formRef.current);\n        }\n      });\n    }, [canSubmit]);\n\n    const abort = useCallback(() => {\n      if (!canAbort || !abortableMessageId) {\n        return;\n      }\n\n      client[kInternal].ai.abort(abortableMessageId);\n    }, [canAbort, abortableMessageId, client]);\n\n    useImperativeHandle<HTMLFormElement | null, HTMLFormElement | null>(\n      forwardedRef,\n      () => formRef.current,\n      []\n    );\n\n    return (\n      <AiComposerEditorContext.Provider\n        value={{\n          editor,\n          onEditorValueChange: handleEditorValueChange,\n          abortableMessageId,\n          setFocused,\n        }}\n      >\n        <AiComposerContext.Provider\n          value={{\n            isDisabled,\n            isEmpty: isEditorEmpty,\n            isFocused,\n            canSubmit,\n            canAbort,\n            submit,\n            abort,\n            clear,\n            focus,\n            blur,\n            select,\n          }}\n        >\n          <Component onSubmit={handleSubmit} {...props} ref={formRef} />\n        </AiComposerContext.Provider>\n      </AiComposerEditorContext.Provider>\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Editor\n * -----------------------------------------------------------------------------------------------*/\n\nfunction AiComposerEditorPlaceholder({\n  attributes,\n  children,\n}: RenderPlaceholderProps) {\n  const { opacity: _opacity, ...style } = attributes.style;\n\n  return (\n    <span {...attributes} style={style} data-placeholder=\"\">\n      {children}\n    </span>\n  );\n}\n\n/**\n * Displays the AI composer's editor.\n *\n * @example\n * <AiComposer.Editor placeholder=\"Write a message…\" />\n */\nconst AiComposerEditor = forwardRef<HTMLDivElement, AiComposerEditorProps>(\n  (\n    {\n      defaultValue = \"\",\n      onKeyDown,\n      onFocus,\n      onBlur,\n      disabled,\n      autoFocus,\n      dir,\n      ...props\n    },\n    forwardedRef\n  ) => {\n    const { editor, onEditorValueChange, setFocused } =\n      useAiComposerEditorContext();\n    const {\n      submit,\n      isDisabled: isComposerDisabled,\n      isFocused,\n      focus,\n      blur,\n      select,\n    } = useAiComposer();\n    const isDisabled = disabled || isComposerDisabled;\n\n    const handleKeyDown = useCallback(\n      (event: KeyboardEvent<HTMLDivElement>) => {\n        onKeyDown?.(event);\n        if (event.isDefaultPrevented()) return;\n\n        if (event.key === \"Enter\" && !event.shiftKey) {\n          event.preventDefault();\n          submit();\n        } else if (event.key === \"Enter\" && event.shiftKey) {\n          event.preventDefault();\n          editor.insertBreak();\n        } else if (event.key === \"Escape\") {\n          blur();\n        }\n      },\n      [editor, onKeyDown, submit, blur]\n    );\n\n    const handleFocus = useCallback(\n      (event: FocusEvent<HTMLDivElement>) => {\n        onFocus?.(event);\n\n        if (!event.isDefaultPrevented()) {\n          setFocused(true);\n        }\n      },\n      [onFocus, setFocused]\n    );\n\n    const handleBlur = useCallback(\n      (event: FocusEvent<HTMLDivElement>) => {\n        onBlur?.(event);\n\n        if (!event.isDefaultPrevented()) {\n          setFocused(false);\n        }\n      },\n      [onBlur, setFocused]\n    );\n\n    useImperativeHandle(\n      forwardedRef,\n      () => ReactEditor.toDOMNode(editor, editor) as HTMLDivElement,\n      [editor]\n    );\n\n    // Manually focus the editor when `autoFocus` is true\n    useLayoutEffect(() => {\n      if (autoFocus) {\n        focus();\n      }\n    }, [autoFocus, editor, focus]);\n\n    // Manually add a selection in the editor if the selection\n    // is still empty after being focused\n    useLayoutEffect(() => {\n      if (isFocused && editor.selection === null) {\n        select();\n      }\n    }, [editor, select, isFocused]);\n\n    const initialValue: AiComposerBody = useMemo(() => {\n      return defaultValue\n        .split(\"\\n\")\n        .map((text) => ({ type: \"paragraph\", children: [{ text }] }));\n    }, [defaultValue]);\n\n    return (\n      <Slate\n        editor={editor}\n        initialValue={initialValue}\n        onValueChange={onEditorValueChange}\n      >\n        <Editable\n          dir={dir}\n          enterKeyHint=\"send\"\n          autoCapitalize=\"sentences\"\n          aria-label=\"Composer editor\"\n          onKeyDown={handleKeyDown}\n          onFocus={handleFocus}\n          onBlur={handleBlur}\n          data-focused={isFocused || undefined}\n          data-disabled={isDisabled || undefined}\n          {...props}\n          readOnly={isDisabled}\n          disabled={isDisabled}\n          renderPlaceholder={AiComposerEditorPlaceholder}\n        />\n      </Slate>\n    );\n  }\n);\n\n/* -------------------------------------------------------------------------------------------------\n * Submit\n * -----------------------------------------------------------------------------------------------*/\n\n/**\n * A button to submit the AI composer's content.\n *\n * @example\n * <AiComposer.Submit>Send</AiComposer.Submit>\n */\nexport const AiComposerSubmit = forwardRef<\n  HTMLButtonElement,\n  AiComposerSubmitProps\n>(({ disabled, asChild, ...props }, forwardedRef) => {\n  const Component = asChild ? Slot : \"button\";\n  const { isDisabled: isComposerDisabled, canSubmit } = useAiComposer();\n  const isDisabled = isComposerDisabled || disabled || !canSubmit;\n\n  return (\n    <Component\n      type=\"submit\"\n      {...props}\n      ref={forwardedRef}\n      disabled={isDisabled}\n    />\n  );\n});\n\n/* -------------------------------------------------------------------------------------------------\n * Abort\n * -----------------------------------------------------------------------------------------------*/\n\n/**\n * A button to abort a response related to the AI composer.\n *\n * @example\n * <AiComposer.Abort>Cancel</AiComposer.Abort>\n */\nexport const AiComposerAbort = forwardRef<\n  HTMLButtonElement,\n  AiComposerSubmitProps\n>(({ disabled, onClick, asChild, ...props }, forwardedRef) => {\n  const Component = asChild ? Slot : \"button\";\n  const { isDisabled: isComposerDisabled, canAbort, abort } = useAiComposer();\n  const isDisabled = isComposerDisabled || disabled || !canAbort;\n\n  const handleClick = useCallback(\n    (event: MouseEvent<HTMLButtonElement>) => {\n      onClick?.(event);\n\n      if (event.isDefaultPrevented()) {\n        return;\n      }\n\n      abort();\n    },\n    [abort, onClick]\n  );\n\n  return (\n    <Component\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={isDisabled}\n      onClick={handleClick}\n    />\n  );\n});\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiComposerEditor.displayName = AI_COMPOSER_EDITOR_NAME;\n  AiComposerForm.displayName = AI_COMPOSER_FORM_NAME;\n  AiComposerSubmit.displayName = AI_COMPOSER_SUBMIT_NAME;\n  AiComposerAbort.displayName = AI_COMPOSER_ABORT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiComposer.*\nexport {\n  AiComposerAbort as Abort,\n  AiComposerEditor as Editor,\n  AiComposerForm as Form,\n  AiComposerSubmit as Submit,\n};\n"],"names":["SlateTransforms","SlateEditor","isEditorEmpty"],"mappings":";;;;;;;;;;;;;;;AAiDA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAChC,MAAM,sBAAyB,GAAA,iBAAA,CAAA;AAC/B,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAChC,MAAM,qBAAwB,GAAA,gBAAA,CAAA;AAQ9B,MAAM,mBAAiB,GAAA,IAAI,MAAwB,CAAA,EAAE,CAAA,CAAA;AAErD,SAAS,iBAAiB,QAA2B,EAAA;AACnD,EAAA,MAAM,WAAc,GAAA,QAAA,CAAS,QAAS,CAAA,MAAA,GAAS,CAAC,CAAA,CAAA;AAEhD,EAAA,IAAI,gBAAgB,KAAW,CAAA,EAAA;AAC7B,IAAO,OAAA,IAAA,CAAA;AAAA,GACT;AAEA,EAAA,OAAO,WAAY,CAAA,EAAA,CAAA;AACrB,CAAA;AAEA,SAAS,sBAAsB,QAA2B,EAAA;AACxD,EAAA,OAAO,QAAS,CAAA,IAAA;AAAA,IACd,CAAC,YACC,OAAQ,CAAA,IAAA,KAAS,gBAChB,OAAQ,CAAA,MAAA,KAAW,YAAgB,IAAA,OAAA,CAAQ,MAAW,KAAA,eAAA,CAAA;AAAA,GACxD,EAAA,EAAA,CAAA;AACL,CAAA;AAWO,MAAM,cAAiB,GAAA,UAAA;AAAA,EAC5B,CACE;AAAA,IACE,gBAAA;AAAA,IACA,QAAA;AAAA,IACA,QAAA;AAAA,IACA,MAAA;AAAA,IACA,QAAA;AAAA,IACA,OAAA;AAAA,IACA,GAAG,KAAA;AAAA,KAEL,YACG,KAAA;AACH,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,MAAA,CAAA;AACnC,IAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AACzB,IAAM,MAAA,OAAA,GAAU,OAA+B,IAAI,CAAA,CAAA;AACnD,IAAA,MAAM,MAAS,GAAA,UAAA;AAAA,MAAW,MACxB,aAAc,CAAA,WAAA,CAAY,UAAU,YAAa,EAAC,CAAC,CAAC,CAAA;AAAA,KACtD,CAAA;AACA,IAAA,MAAM,CAAC,aAAA,EAAe,cAAc,CAAA,GAAI,SAAS,IAAI,CAAA,CAAA;AACrD,IAAA,MAAM,CAAC,YAAA,EAAc,aAAa,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AACpD,IAAA,MAAM,CAAC,SAAA,EAAW,UAAU,CAAA,GAAI,SAAS,KAAK,CAAA,CAAA;AAC9C,IAAM,MAAA,cAAA,GAAY,MACd,GAAA,MAAA,CAAO,SAAS,CAAA,CAAE,GAAG,OAAQ,CAAA,8BAAA,CAA0B,MAAQ,EAAA,QAAQ,CACvE,GAAA,mBAAA,CAAA;AACJ,IAAM,MAAA,aAAA,GAAgB,SAAU,CAAA,cAAA,EAAW,gBAAgB,CAAA,CAAA;AAC3D,IAAM,MAAA,kBAAA,GAAqB,SAAU,CAAA,cAAA,EAAW,qBAAqB,CAAA,CAAA;AACrE,IAAA,MAAM,WAAc,GAAA,SAAA;AAAA;AAAA,MAElB,MAAO,CAAA,SAAS,CAAE,CAAA,EAAA,CAAG,OAAQ,CAAA,YAAA;AAAA;AAAA;AAAA,MAG7B,CAAC,WAAW,MAAW,KAAA,cAAA;AAAA,KACzB,CAAA;AAEA,IAAM,MAAA,UAAA,GAAa,gBAAgB,QAAa,KAAA,IAAA,CAAA;AAEhD,IAAM,MAAA,QAAA,GAAW,eAAe,kBAAuB,KAAA,KAAA,CAAA,CAAA;AACvD,IAAA,MAAM,SAAY,GAAA,WAAA,IAAe,CAAC,aAAA,IAAiB,CAAC,QAAA,CAAA;AAEpD,IAAM,MAAA,KAAA,GAAQ,YAAY,MAAM;AAC9B,MAAAA,UAAA,CAAgB,OAAO,MAAQ,EAAA;AAAA,QAC7B,EAAI,EAAA;AAAA,UACF,MAAQ,EAAAC,MAAA,CAAY,KAAM,CAAA,MAAA,EAAQ,EAAE,CAAA;AAAA,UACpC,KAAO,EAAAA,MAAA,CAAY,GAAI,CAAA,MAAA,EAAQ,EAAE,CAAA;AAAA,SACnC;AAAA,OACD,CAAA,CAAA;AAAA,KACH,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,MAAA,GAAS,YAAY,MAAM;AAC/B,MAAAD,UAAA,CAAgB,OAAO,MAAQ,EAAAC,MAAA,CAAY,IAAI,MAAQ,EAAA,EAAE,CAAC,CAAA,CAAA;AAAA,KAC5D,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAA,MAAM,KAAQ,GAAA,WAAA;AAAA,MACZ,CAAC,iBAAiB,IAAS,KAAA;AACzB,QAAI,IAAA;AACF,UAAA,IAAI,CAAC,WAAA,CAAY,SAAU,CAAA,MAAM,CAAG,EAAA;AAClC,YAAgBD,UAAA,CAAA,MAAA;AAAA,cACd,MAAA;AAAA,cACA,cAAA,IAAkB,CAAC,MAAA,CAAO,SACtB,GAAAC,MAAA,CAAY,IAAI,MAAQ,EAAA,EAAE,CAAA,GAC1B,MAAO,CAAA,SAAA;AAAA,aACb,CAAA;AACA,YAAA,WAAA,CAAY,MAAM,MAAM,CAAA,CAAA;AAAA,WAC1B;AAAA,SACM,CAAA,MAAA;AAAA,SAIR;AAAA,OACF;AAAA,MACA,CAAC,MAAM,CAAA;AAAA,KACT,CAAA;AAEA,IAAM,MAAA,IAAA,GAAO,YAAY,MAAM;AAC7B,MAAI,IAAA;AACF,QAAA,WAAA,CAAY,KAAK,MAAM,CAAA,CAAA;AAAA,OACjB,CAAA,MAAA;AAAA,OAIR;AAAA,KACF,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,WAAA,GAAc,YAAY,MAAM;AACpC,MAAM,KAAA,EAAA,CAAA;AACN,MAAA,aAAA,CAAc,KAAK,CAAA,CAAA;AAAA,KACrB,EAAG,CAAC,KAAK,CAAC,CAAA,CAAA;AAEV,IAAA,MAAM,YAAe,GAAA,WAAA;AAAA,MACnB,CAAC,KAAsC,KAAA;AACrC,QAAA,IAAI,QAAU,EAAA;AACZ,UAAA,OAAA;AAAA,SACF;AAKA,QAAA,MAAMC,cAAgB,GAAA,OAAA,CAAQ,MAAQ,EAAA,MAAA,CAAO,QAAQ,CAAA,CAAA;AAGrD,QAAA,IAAIA,cAAe,EAAA;AACjB,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AAErB,UAAA,OAAA;AAAA,SACF;AAEA,QAAA,QAAA,GAAW,KAAK,CAAA,CAAA;AAEhB,QAAA,IAAI,gBAAqB,KAAA,KAAA,CAAA,IAAa,KAAM,CAAA,kBAAA,EAAsB,EAAA;AAChE,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AACrB,UAAA,OAAA;AAAA,SACF;AAGA,QAAA,MAAM,OAAU,GAAA,MAAA,CAAO,QACpB,CAAA,GAAA,CAAI,CAAC,KAAU,KAAA;AACd,UAAA,IAAI,MAAU,IAAA,KAAA,IAAS,KAAM,CAAA,IAAA,KAAS,WAAa,EAAA;AACjD,YAAA,OAAO,KAAM,CAAA,QAAA,CACV,GAAI,CAAA,CAAC,KAAU,KAAA;AACd,cAAA,IAAI,UAAU,KAAO,EAAA;AACnB,gBAAA,OAAO,KAAM,CAAA,IAAA,CAAA;AAAA,eACf;AACA,cAAO,OAAA,EAAA,CAAA;AAAA,aACR,CACA,CAAA,IAAA,CAAK,EAAE,CAAA,CAAA;AAAA,WACZ;AACA,UAAO,OAAA,EAAA,CAAA;AAAA,SACR,CACA,CAAA,IAAA,CAAK,IAAI,CAAA,CAAA;AAEZ,QAAA,MAAM,OAAU,GAAA,gBAAA;AAAA,UACd,EAAE,IAAM,EAAA,OAAA,EAAS,aAAc,EAAA;AAAA,UAC/B,KAAA;AAAA,SACF,CAAA;AAEA,QAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AAErB,QAAA,IAAI,OAAS,EAAA;AACX,UAAA,aAAA,CAAc,IAAI,CAAA,CAAA;AAClB,UAAA,OAAA,CAAQ,KAAK,WAAW,CAAA,CAAA;AAAA,SACnB,MAAA;AACL,UAAY,WAAA,EAAA,CAAA;AAAA,SACd;AAAA,OACF;AAAA,MACA,CAAC,QAAU,EAAA,MAAA,EAAQ,QAAU,EAAA,gBAAA,EAAkB,aAAa,aAAa,CAAA;AAAA,KAC3E,CAAA;AAEA,IAAA,eAAA,CAAgB,MAAM;AACpB,MAAA,cAAA,CAAe,OAAQ,CAAA,MAAA,EAAQ,MAAO,CAAA,QAAQ,CAAC,CAAA,CAAA;AAAA,KACjD,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,uBAAA,GAA0B,YAAY,MAAM;AAChD,MAAA,cAAA,CAAe,OAAQ,CAAA,MAAA,EAAQ,MAAO,CAAA,QAAQ,CAAC,CAAA,CAAA;AAAA,KACjD,EAAG,CAAC,MAAM,CAAC,CAAA,CAAA;AAEX,IAAM,MAAA,MAAA,GAAS,YAAY,MAAM;AAC/B,MAAA,IAAI,CAAC,SAAW,EAAA;AACd,QAAA,OAAA;AAAA,OACF;AAIA,MAAA,qBAAA,CAAsB,MAAM;AAC1B,QAAA,IAAI,QAAQ,OAAS,EAAA;AACnB,UAAA,aAAA,CAAc,QAAQ,OAAO,CAAA,CAAA;AAAA,SAC/B;AAAA,OACD,CAAA,CAAA;AAAA,KACH,EAAG,CAAC,SAAS,CAAC,CAAA,CAAA;AAEd,IAAM,MAAA,KAAA,GAAQ,YAAY,MAAM;AAC9B,MAAI,IAAA,CAAC,QAAY,IAAA,CAAC,kBAAoB,EAAA;AACpC,QAAA,OAAA;AAAA,OACF;AAEA,MAAA,MAAA,CAAO,SAAS,CAAA,CAAE,EAAG,CAAA,KAAA,CAAM,kBAAkB,CAAA,CAAA;AAAA,KAC5C,EAAA,CAAC,QAAU,EAAA,kBAAA,EAAoB,MAAM,CAAC,CAAA,CAAA;AAEzC,IAAA,mBAAA;AAAA,MACE,YAAA;AAAA,MACA,MAAM,OAAQ,CAAA,OAAA;AAAA,MACd,EAAC;AAAA,KACH,CAAA;AAEA,IACE,uBAAA,GAAA;AAAA,MAAC,uBAAwB,CAAA,QAAA;AAAA,MAAxB;AAAA,QACC,KAAO,EAAA;AAAA,UACL,MAAA;AAAA,UACA,mBAAqB,EAAA,uBAAA;AAAA,UACrB,kBAAA;AAAA,UACA,UAAA;AAAA,SACF;AAAA,QAEA,QAAA,kBAAA,GAAA;AAAA,UAAC,iBAAkB,CAAA,QAAA;AAAA,UAAlB;AAAA,YACC,KAAO,EAAA;AAAA,cACL,UAAA;AAAA,cACA,OAAS,EAAA,aAAA;AAAA,cACT,SAAA;AAAA,cACA,SAAA;AAAA,cACA,QAAA;AAAA,cACA,MAAA;AAAA,cACA,KAAA;AAAA,cACA,KAAA;AAAA,cACA,KAAA;AAAA,cACA,IAAA;AAAA,cACA,MAAA;AAAA,aACF;AAAA,YAEA,8BAAC,SAAU,EAAA,EAAA,QAAA,EAAU,cAAe,GAAG,KAAA,EAAO,KAAK,OAAS,EAAA,CAAA;AAAA,WAAA;AAAA,SAC9D;AAAA,OAAA;AAAA,KACF,CAAA;AAAA,GAEJ;AACF,EAAA;AAMA,SAAS,2BAA4B,CAAA;AAAA,EACnC,UAAA;AAAA,EACA,QAAA;AACF,CAA2B,EAAA;AACzB,EAAA,MAAM,EAAE,OAAS,EAAA,QAAA,EAAU,GAAG,KAAA,KAAU,UAAW,CAAA,KAAA,CAAA;AAEnD,EAAA,2BACG,MAAM,EAAA,EAAA,GAAG,YAAY,KAAc,EAAA,kBAAA,EAAiB,IAClD,QACH,EAAA,CAAA,CAAA;AAEJ,CAAA;AAQA,MAAM,gBAAmB,GAAA,UAAA;AAAA,EACvB,CACE;AAAA,IACE,YAAe,GAAA,EAAA;AAAA,IACf,SAAA;AAAA,IACA,OAAA;AAAA,IACA,MAAA;AAAA,IACA,QAAA;AAAA,IACA,SAAA;AAAA,IACA,GAAA;AAAA,IACA,GAAG,KAAA;AAAA,KAEL,YACG,KAAA;AACH,IAAA,MAAM,EAAE,MAAA,EAAQ,mBAAqB,EAAA,UAAA,KACnC,0BAA2B,EAAA,CAAA;AAC7B,IAAM,MAAA;AAAA,MACJ,MAAA;AAAA,MACA,UAAY,EAAA,kBAAA;AAAA,MACZ,SAAA;AAAA,MACA,KAAA;AAAA,MACA,IAAA;AAAA,MACA,MAAA;AAAA,QACE,aAAc,EAAA,CAAA;AAClB,IAAA,MAAM,aAAa,QAAY,IAAA,kBAAA,CAAA;AAE/B,IAAA,MAAM,aAAgB,GAAA,WAAA;AAAA,MACpB,CAAC,KAAyC,KAAA;AACxC,QAAA,SAAA,GAAY,KAAK,CAAA,CAAA;AACjB,QAAA,IAAI,MAAM,kBAAmB,EAAA;AAAG,UAAA,OAAA;AAEhC,QAAA,IAAI,KAAM,CAAA,GAAA,KAAQ,OAAW,IAAA,CAAC,MAAM,QAAU,EAAA;AAC5C,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AACrB,UAAO,MAAA,EAAA,CAAA;AAAA,SACE,MAAA,IAAA,KAAA,CAAM,GAAQ,KAAA,OAAA,IAAW,MAAM,QAAU,EAAA;AAClD,UAAA,KAAA,CAAM,cAAe,EAAA,CAAA;AACrB,UAAA,MAAA,CAAO,WAAY,EAAA,CAAA;AAAA,SACrB,MAAA,IAAW,KAAM,CAAA,GAAA,KAAQ,QAAU,EAAA;AACjC,UAAK,IAAA,EAAA,CAAA;AAAA,SACP;AAAA,OACF;AAAA,MACA,CAAC,MAAA,EAAQ,SAAW,EAAA,MAAA,EAAQ,IAAI,CAAA;AAAA,KAClC,CAAA;AAEA,IAAA,MAAM,WAAc,GAAA,WAAA;AAAA,MAClB,CAAC,KAAsC,KAAA;AACrC,QAAA,OAAA,GAAU,KAAK,CAAA,CAAA;AAEf,QAAI,IAAA,CAAC,KAAM,CAAA,kBAAA,EAAsB,EAAA;AAC/B,UAAA,UAAA,CAAW,IAAI,CAAA,CAAA;AAAA,SACjB;AAAA,OACF;AAAA,MACA,CAAC,SAAS,UAAU,CAAA;AAAA,KACtB,CAAA;AAEA,IAAA,MAAM,UAAa,GAAA,WAAA;AAAA,MACjB,CAAC,KAAsC,KAAA;AACrC,QAAA,MAAA,GAAS,KAAK,CAAA,CAAA;AAEd,QAAI,IAAA,CAAC,KAAM,CAAA,kBAAA,EAAsB,EAAA;AAC/B,UAAA,UAAA,CAAW,KAAK,CAAA,CAAA;AAAA,SAClB;AAAA,OACF;AAAA,MACA,CAAC,QAAQ,UAAU,CAAA;AAAA,KACrB,CAAA;AAEA,IAAA,mBAAA;AAAA,MACE,YAAA;AAAA,MACA,MAAM,WAAA,CAAY,SAAU,CAAA,MAAA,EAAQ,MAAM,CAAA;AAAA,MAC1C,CAAC,MAAM,CAAA;AAAA,KACT,CAAA;AAGA,IAAA,eAAA,CAAgB,MAAM;AACpB,MAAA,IAAI,SAAW,EAAA;AACb,QAAM,KAAA,EAAA,CAAA;AAAA,OACR;AAAA,KACC,EAAA,CAAC,SAAW,EAAA,MAAA,EAAQ,KAAK,CAAC,CAAA,CAAA;AAI7B,IAAA,eAAA,CAAgB,MAAM;AACpB,MAAI,IAAA,SAAA,IAAa,MAAO,CAAA,SAAA,KAAc,IAAM,EAAA;AAC1C,QAAO,MAAA,EAAA,CAAA;AAAA,OACT;AAAA,KACC,EAAA,CAAC,MAAQ,EAAA,MAAA,EAAQ,SAAS,CAAC,CAAA,CAAA;AAE9B,IAAM,MAAA,YAAA,GAA+B,QAAQ,MAAM;AACjD,MAAA,OAAO,aACJ,KAAM,CAAA,IAAI,CACV,CAAA,GAAA,CAAI,CAAC,IAAU,MAAA,EAAE,IAAM,EAAA,WAAA,EAAa,UAAU,CAAC,EAAE,IAAK,EAAC,GAAI,CAAA,CAAA,CAAA;AAAA,KAChE,EAAG,CAAC,YAAY,CAAC,CAAA,CAAA;AAEjB,IACE,uBAAA,GAAA;AAAA,MAAC,KAAA;AAAA,MAAA;AAAA,QACC,MAAA;AAAA,QACA,YAAA;AAAA,QACA,aAAe,EAAA,mBAAA;AAAA,QAEf,QAAA,kBAAA,GAAA;AAAA,UAAC,QAAA;AAAA,UAAA;AAAA,YACC,GAAA;AAAA,YACA,YAAa,EAAA,MAAA;AAAA,YACb,cAAe,EAAA,WAAA;AAAA,YACf,YAAW,EAAA,iBAAA;AAAA,YACX,SAAW,EAAA,aAAA;AAAA,YACX,OAAS,EAAA,WAAA;AAAA,YACT,MAAQ,EAAA,UAAA;AAAA,YACR,gBAAc,SAAa,IAAA,KAAA,CAAA;AAAA,YAC3B,iBAAe,UAAc,IAAA,KAAA,CAAA;AAAA,YAC5B,GAAG,KAAA;AAAA,YACJ,QAAU,EAAA,UAAA;AAAA,YACV,QAAU,EAAA,UAAA;AAAA,YACV,iBAAmB,EAAA,2BAAA;AAAA,WAAA;AAAA,SACrB;AAAA,OAAA;AAAA,KACF,CAAA;AAAA,GAEJ;AACF,EAAA;AAYa,MAAA,gBAAA,GAAmB,WAG9B,CAAC,EAAE,UAAU,OAAS,EAAA,GAAG,KAAM,EAAA,EAAG,YAAiB,KAAA;AACnD,EAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,QAAA,CAAA;AACnC,EAAA,MAAM,EAAE,UAAA,EAAY,kBAAoB,EAAA,SAAA,KAAc,aAAc,EAAA,CAAA;AACpE,EAAM,MAAA,UAAA,GAAa,kBAAsB,IAAA,QAAA,IAAY,CAAC,SAAA,CAAA;AAEtD,EACE,uBAAA,GAAA;AAAA,IAAC,SAAA;AAAA,IAAA;AAAA,MACC,IAAK,EAAA,QAAA;AAAA,MACJ,GAAG,KAAA;AAAA,MACJ,GAAK,EAAA,YAAA;AAAA,MACL,QAAU,EAAA,UAAA;AAAA,KAAA;AAAA,GACZ,CAAA;AAEJ,CAAC,EAAA;AAYY,MAAA,eAAA,GAAkB,UAG7B,CAAA,CAAC,EAAE,QAAA,EAAU,SAAS,OAAS,EAAA,GAAG,KAAM,EAAA,EAAG,YAAiB,KAAA;AAC5D,EAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,QAAA,CAAA;AACnC,EAAA,MAAM,EAAE,UAAY,EAAA,kBAAA,EAAoB,QAAU,EAAA,KAAA,KAAU,aAAc,EAAA,CAAA;AAC1E,EAAM,MAAA,UAAA,GAAa,kBAAsB,IAAA,QAAA,IAAY,CAAC,QAAA,CAAA;AAEtD,EAAA,MAAM,WAAc,GAAA,WAAA;AAAA,IAClB,CAAC,KAAyC,KAAA;AACxC,MAAA,OAAA,GAAU,KAAK,CAAA,CAAA;AAEf,MAAI,IAAA,KAAA,CAAM,oBAAsB,EAAA;AAC9B,QAAA,OAAA;AAAA,OACF;AAEA,MAAM,KAAA,EAAA,CAAA;AAAA,KACR;AAAA,IACA,CAAC,OAAO,OAAO,CAAA;AAAA,GACjB,CAAA;AAEA,EACE,uBAAA,GAAA;AAAA,IAAC,SAAA;AAAA,IAAA;AAAA,MACC,IAAK,EAAA,QAAA;AAAA,MACJ,GAAG,KAAA;AAAA,MACJ,GAAK,EAAA,YAAA;AAAA,MACL,QAAU,EAAA,UAAA;AAAA,MACV,OAAS,EAAA,WAAA;AAAA,KAAA;AAAA,GACX,CAAA;AAEJ,CAAC,EAAA;AAED,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AAC/B,EAAA,cAAA,CAAe,WAAc,GAAA,qBAAA,CAAA;AAC7B,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AAC/B,EAAA,eAAA,CAAgB,WAAc,GAAA,sBAAA,CAAA;AAChC;;;;"}

package/dist/primitives/AiMessage/index.cjs

@@ -10,24 +10,14 @@ var toolInvocation = require('./tool-invocation.cjs');
 const AI_MESSAGE_CONTENT_NAME = "AiMessageContent";
 const defaultMessageContentComponents = {
   TextPart: ({ part }) => {
-    return /* @__PURE__ */ jsxRuntime.jsx(Markdown.Markdown, {
-      content: part.text
-    });
+    return /* @__PURE__ */ jsxRuntime.jsx(Markdown.Markdown, { content: part.text });
   },
   ReasoningPart: ({ part }) => {
-    return /* @__PURE__ */ jsxRuntime.jsx(Markdown.Markdown, {
-      content: part.text
-    });
+    return /* @__PURE__ */ jsxRuntime.jsx(Markdown.Markdown, { content: part.text });
   },
   RetrievalPart: () => null,
   ToolInvocationPart: ({ part, message }) => {
-    return /* @__PURE__ */ jsxRuntime.jsx(ErrorBoundary.ErrorBoundary, {
-      fallback: null,
-      children: /* @__PURE__ */ jsxRuntime.jsx(toolInvocation.AiMessageToolInvocation, {
-        part,
-        message
-      })
-    });
+    return /* @__PURE__ */ jsxRuntime.jsx(ErrorBoundary.ErrorBoundary, { fallback: null, children: /* @__PURE__ */ jsxRuntime.jsx(toolInvocation.AiMessageToolInvocation, { part, message }) });
   },
   SourcesPart: () => null
 };
@@ -47,46 +37,35 @@ const AiMessageContent = react.forwardRef(
     const content = message.content ?? message.contentSoFar;
     const numParts = content.length;
     const isGenerating = message.role === "assistant" && message.status === "generating";
-    return /* @__PURE__ */ jsxRuntime.jsx(Component, {
-      ...props,
-      ref: forwardedRef,
-      children: content.map((part, index) => {
-        const isStreaming = isGenerating && index === numParts - 1;
-        const extra = { index, isStreaming };
-        switch (part.type) {
-          case "text":
-            return /* @__PURE__ */ jsxRuntime.jsx(TextPart, {
-              part,
-              ...extra
-            }, index);
-          case "reasoning":
-            return /* @__PURE__ */ jsxRuntime.jsx(ReasoningPart, {
-              part,
-              ...extra
-            }, index);
-          case "retrieval":
-            return /* @__PURE__ */ jsxRuntime.jsx(RetrievalPart, {
-              part,
-              ...extra
-            }, index);
-          case "tool-invocation":
-            return /* @__PURE__ */ jsxRuntime.jsx(ToolInvocationPart, {
+    return /* @__PURE__ */ jsxRuntime.jsx(Component, { ...props, ref: forwardedRef, children: content.map((part, index) => {
+      const isStreaming = isGenerating && index === numParts - 1;
+      const extra = { index, isStreaming };
+      switch (part.type) {
+        case "text":
+          return /* @__PURE__ */ jsxRuntime.jsx(TextPart, { part, ...extra }, index);
+        case "reasoning":
+          return /* @__PURE__ */ jsxRuntime.jsx(ReasoningPart, { part, ...extra }, index);
+        case "retrieval":
+          return /* @__PURE__ */ jsxRuntime.jsx(RetrievalPart, { part, ...extra }, index);
+        case "tool-invocation":
+          return /* @__PURE__ */ jsxRuntime.jsx(
+            ToolInvocationPart,
+            {
               part,
               ...extra,
               message
-            }, index);
-          case "sources":
-            if (message.role === "assistant" && message.status !== "completed" && message.status !== "failed") {
-              return null;
-            }
-            return /* @__PURE__ */ jsxRuntime.jsx(SourcesPart, {
-              part
-            }, index);
-          default:
+            },
+            index
+          );
+        case "sources":
+          if (message.role === "assistant" && message.status !== "completed" && message.status !== "failed") {
             return null;
-        }
-      })
-    });
+          }
+          return /* @__PURE__ */ jsxRuntime.jsx(SourcesPart, { part }, index);
+        default:
+          return null;
+      }
+    }) });
   }
 );
 if (process.env.NODE_ENV !== "production") {

package/dist/primitives/AiMessage/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","sources":["../../../src/primitives/AiMessage/index.tsx"],"sourcesContent":["import type { AiAssistantMessage } from \"@liveblocks/core\";\nimport { Slot } from \"@radix-ui/react-slot\";\nimport { forwardRef, useMemo } from \"react\";\n\nimport { ErrorBoundary } from \"../../utils/ErrorBoundary\";\nimport { Markdown } from \"../Markdown\";\nimport { AiMessageToolInvocation } from \"./tool-invocation\";\nimport type {\n  AiMessageContentComponents,\n  AiMessageContentProps,\n} from \"./types\";\n\nconst AI_MESSAGE_CONTENT_NAME = \"AiMessageContent\";\n\nconst defaultMessageContentComponents: AiMessageContentComponents = {\n  TextPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ReasoningPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  RetrievalPart: () => null,\n  ToolInvocationPart: ({ part, message }) => {\n    return (\n      <ErrorBoundary fallback={null}>\n        <AiMessageToolInvocation part={part} message={message} />\n      </ErrorBoundary>\n    );\n  },\n  SourcesPart: () => null,\n};\n\n/**\n * --------------------------------------------------------------------------\n * @private The API for this component is not yet stable.\n * --------------------------------------------------------------------------\n *\n * Primitive to help display an user or assistant message’s content, which is\n * an array of parts.\n *\n * @example\n * <AiMessage.Content message={message} components={{ TextPart }} />\n */\nconst AiMessageContent = forwardRef<HTMLDivElement, AiMessageContentProps>(\n  ({ message, components, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const {\n      ReasoningPart,\n      RetrievalPart,\n      TextPart,\n      ToolInvocationPart,\n      SourcesPart,\n    } = useMemo(\n      () => ({ ...defaultMessageContentComponents, ...components }),\n      [components]\n    );\n\n    const content = message.content ?? message.contentSoFar;\n    const numParts = content.length;\n    const isGenerating =\n      message.role === \"assistant\" && message.status === \"generating\";\n\n    return (\n      <Component {...props} ref={forwardedRef}>\n        {content.map((part, index) => {\n          // A part is considered to be still \"streaming in\" if it's the last\n          // part in the content array, and the message is in \"generating\"\n          // state.\n          const isStreaming = isGenerating && index === numParts - 1;\n          const extra = { index, isStreaming };\n          switch (part.type) {\n            case \"text\":\n              return <TextPart key={index} part={part} {...extra} />;\n\n            case \"reasoning\":\n              return <ReasoningPart key={index} part={part} {...extra} />;\n\n            case \"retrieval\":\n              return <RetrievalPart key={index} part={part} {...extra} />;\n\n            case \"tool-invocation\":\n              return (\n                <ToolInvocationPart\n                  key={index}\n                  part={part}\n                  {...extra}\n                  // For tool invocations, we know for sure it's an AiAssistantMessage\n                  message={message as AiAssistantMessage}\n                />\n              );\n\n            case \"sources\":\n              // Sources are only shown after the message is either completed or failed.\n              if (\n                message.role === \"assistant\" &&\n                message.status !== \"completed\" &&\n                message.status !== \"failed\"\n              ) {\n                return null;\n              }\n\n              return <SourcesPart key={index} part={part} />;\n\n            default:\n              return null;\n          }\n        })}\n      </Component>\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiMessage.*\nexport { AiMessageContent as Content };\n"],"names":["jsx","Markdown","ErrorBoundary","AiMessageToolInvocation","forwardRef","Slot","useMemo"],"mappings":";;;;;;;;;AAYA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAEhC,MAAM,+BAA8D,GAAA;AAAA,EAClE,QAAU,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AACtB,IAAA,uBAAQA,cAAA,CAAAC,iBAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,aAAe,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AAC3B,IAAA,uBAAQD,cAAA,CAAAC,iBAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,eAAe,MAAM,IAAA;AAAA,EACrB,kBAAoB,EAAA,CAAC,EAAE,IAAA,EAAM,SAAc,KAAA;AACzC,IAAA,uBACGD,cAAA,CAAAE,2BAAA,EAAA;AAAA,MAAc,QAAU,EAAA,IAAA;AAAA,MACvB,QAAC,kBAAAF,cAAA,CAAAG,sCAAA,EAAA;AAAA,QAAwB,IAAA;AAAA,QAAY,OAAA;AAAA,OAAkB,CAAA;AAAA,KACzD,CAAA,CAAA;AAAA,GAEJ;AAAA,EACA,aAAa,MAAM,IAAA;AACrB,CAAA,CAAA;AAaA,MAAM,gBAAmB,GAAAC,gBAAA;AAAA,EACvB,CAAC,EAAE,OAAA,EAAS,YAAY,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AAC5D,IAAM,MAAA,SAAA,GAAY,UAAUC,cAAO,GAAA,KAAA,CAAA;AACnC,IAAM,MAAA;AAAA,MACJ,aAAA;AAAA,MACA,aAAA;AAAA,MACA,QAAA;AAAA,MACA,kBAAA;AAAA,MACA,WAAA;AAAA,KACE,GAAAC,aAAA;AAAA,MACF,OAAO,EAAE,GAAG,+BAAA,EAAiC,GAAG,UAAW,EAAA,CAAA;AAAA,MAC3D,CAAC,UAAU,CAAA;AAAA,KACb,CAAA;AAEA,IAAM,MAAA,OAAA,GAAU,OAAQ,CAAA,OAAA,IAAW,OAAQ,CAAA,YAAA,CAAA;AAC3C,IAAA,MAAM,WAAW,OAAQ,CAAA,MAAA,CAAA;AACzB,IAAA,MAAM,YACJ,GAAA,OAAA,CAAQ,IAAS,KAAA,WAAA,IAAe,QAAQ,MAAW,KAAA,YAAA,CAAA;AAErD,IAAA,uBACGN,cAAA,CAAA,SAAA,EAAA;AAAA,MAAW,GAAG,KAAA;AAAA,MAAO,GAAK,EAAA,YAAA;AAAA,MACxB,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAI5B,QAAM,MAAA,WAAA,GAAc,YAAgB,IAAA,KAAA,KAAU,QAAW,GAAA,CAAA,CAAA;AACzD,QAAM,MAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,WAAY,EAAA,CAAA;AACnC,QAAA,QAAQ,KAAK,IAAM;AAAA,UACjB,KAAK,MAAA;AACH,YAAA,uBAAQA,cAAA,CAAA,QAAA,EAAA;AAAA,cAAqB,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAEtD,KAAK,WAAA;AACH,YAAA,uBAAQA,cAAA,CAAA,aAAA,EAAA;AAAA,cAA0B,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAE3D,KAAK,WAAA;AACH,YAAA,uBAAQA,cAAA,CAAA,aAAA,EAAA;AAAA,cAA0B,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAE3D,KAAK,iBAAA;AACH,YAAA,uBACGA,cAAA,CAAA,kBAAA,EAAA;AAAA,cAEC,IAAA;AAAA,cACC,GAAG,KAAA;AAAA,cAEJ,OAAA;AAAA,aAAA,EAJK,KAKP,CAAA,CAAA;AAAA,UAGJ,KAAK,SAAA;AAEH,YACE,IAAA,OAAA,CAAQ,SAAS,WACjB,IAAA,OAAA,CAAQ,WAAW,WACnB,IAAA,OAAA,CAAQ,WAAW,QACnB,EAAA;AACA,cAAO,OAAA,IAAA,CAAA;AAAA,aACT;AAEA,YAAA,uBAAQA,cAAA,CAAA,WAAA,EAAA;AAAA,cAAwB,IAAA;AAAA,aAAA,EAAP,KAAmB,CAAA,CAAA;AAAA,UAE9C;AACE,YAAO,OAAA,IAAA,CAAA;AAAA,SACX;AAAA,OACD,CAAA;AAAA,KACH,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AACjC;;;;"}
+{"version":3,"file":"index.cjs","sources":["../../../src/primitives/AiMessage/index.tsx"],"sourcesContent":["import type { AiAssistantMessage } from \"@liveblocks/core\";\nimport { Slot } from \"@radix-ui/react-slot\";\nimport { forwardRef, useMemo } from \"react\";\n\nimport { ErrorBoundary } from \"../../utils/ErrorBoundary\";\nimport { Markdown } from \"../Markdown\";\nimport { AiMessageToolInvocation } from \"./tool-invocation\";\nimport type {\n  AiMessageContentComponents,\n  AiMessageContentProps,\n} from \"./types\";\n\nconst AI_MESSAGE_CONTENT_NAME = \"AiMessageContent\";\n\nconst defaultMessageContentComponents: AiMessageContentComponents = {\n  TextPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ReasoningPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  RetrievalPart: () => null,\n  ToolInvocationPart: ({ part, message }) => {\n    return (\n      <ErrorBoundary fallback={null}>\n        <AiMessageToolInvocation part={part} message={message} />\n      </ErrorBoundary>\n    );\n  },\n  SourcesPart: () => null,\n};\n\n/**\n * --------------------------------------------------------------------------\n * @private The API for this component is not yet stable.\n * --------------------------------------------------------------------------\n *\n * Primitive to help display an user or assistant message’s content, which is\n * an array of parts.\n *\n * @example\n * <AiMessage.Content message={message} components={{ TextPart }} />\n */\nconst AiMessageContent = forwardRef<HTMLDivElement, AiMessageContentProps>(\n  ({ message, components, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const {\n      ReasoningPart,\n      RetrievalPart,\n      TextPart,\n      ToolInvocationPart,\n      SourcesPart,\n    } = useMemo(\n      () => ({ ...defaultMessageContentComponents, ...components }),\n      [components]\n    );\n\n    const content = message.content ?? message.contentSoFar;\n    const numParts = content.length;\n    const isGenerating =\n      message.role === \"assistant\" && message.status === \"generating\";\n\n    return (\n      <Component {...props} ref={forwardedRef}>\n        {content.map((part, index) => {\n          // A part is considered to be still \"streaming in\" if it's the last\n          // part in the content array, and the message is in \"generating\"\n          // state.\n          const isStreaming = isGenerating && index === numParts - 1;\n          const extra = { index, isStreaming };\n          switch (part.type) {\n            case \"text\":\n              return <TextPart key={index} part={part} {...extra} />;\n\n            case \"reasoning\":\n              return <ReasoningPart key={index} part={part} {...extra} />;\n\n            case \"retrieval\":\n              return <RetrievalPart key={index} part={part} {...extra} />;\n\n            case \"tool-invocation\":\n              return (\n                <ToolInvocationPart\n                  key={index}\n                  part={part}\n                  {...extra}\n                  // For tool invocations, we know for sure it's an AiAssistantMessage\n                  message={message as AiAssistantMessage}\n                />\n              );\n\n            case \"sources\":\n              // Sources are only shown after the message is either completed or failed.\n              if (\n                message.role === \"assistant\" &&\n                message.status !== \"completed\" &&\n                message.status !== \"failed\"\n              ) {\n                return null;\n              }\n\n              return <SourcesPart key={index} part={part} />;\n\n            default:\n              return null;\n          }\n        })}\n      </Component>\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiMessage.*\nexport { AiMessageContent as Content };\n"],"names":["jsx","Markdown","ErrorBoundary","AiMessageToolInvocation","forwardRef","Slot","useMemo"],"mappings":";;;;;;;;;AAYA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAEhC,MAAM,+BAA8D,GAAA;AAAA,EAClE,QAAU,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AACtB,IAAA,uBAAQA,cAAA,CAAAC,iBAAA,EAAA,EAAS,OAAS,EAAA,IAAA,CAAK,IAAM,EAAA,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,aAAe,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AAC3B,IAAA,uBAAQD,cAAA,CAAAC,iBAAA,EAAA,EAAS,OAAS,EAAA,IAAA,CAAK,IAAM,EAAA,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,eAAe,MAAM,IAAA;AAAA,EACrB,kBAAoB,EAAA,CAAC,EAAE,IAAA,EAAM,SAAc,KAAA;AACzC,IACE,uBAAAD,cAAA,CAACE,+BAAc,QAAU,EAAA,IAAA,EACvB,yCAACC,sCAAwB,EAAA,EAAA,IAAA,EAAY,SAAkB,CACzD,EAAA,CAAA,CAAA;AAAA,GAEJ;AAAA,EACA,aAAa,MAAM,IAAA;AACrB,CAAA,CAAA;AAaA,MAAM,gBAAmB,GAAAC,gBAAA;AAAA,EACvB,CAAC,EAAE,OAAS,EAAA,UAAA,EAAY,SAAS,GAAG,KAAA,IAAS,YAAiB,KAAA;AAC5D,IAAM,MAAA,SAAA,GAAY,UAAUC,cAAO,GAAA,KAAA,CAAA;AACnC,IAAM,MAAA;AAAA,MACJ,aAAA;AAAA,MACA,aAAA;AAAA,MACA,QAAA;AAAA,MACA,kBAAA;AAAA,MACA,WAAA;AAAA,KACE,GAAAC,aAAA;AAAA,MACF,OAAO,EAAE,GAAG,+BAAA,EAAiC,GAAG,UAAW,EAAA,CAAA;AAAA,MAC3D,CAAC,UAAU,CAAA;AAAA,KACb,CAAA;AAEA,IAAM,MAAA,OAAA,GAAU,OAAQ,CAAA,OAAA,IAAW,OAAQ,CAAA,YAAA,CAAA;AAC3C,IAAA,MAAM,WAAW,OAAQ,CAAA,MAAA,CAAA;AACzB,IAAA,MAAM,YACJ,GAAA,OAAA,CAAQ,IAAS,KAAA,WAAA,IAAe,QAAQ,MAAW,KAAA,YAAA,CAAA;AAErD,IACE,uBAAAN,cAAA,CAAC,SAAW,EAAA,EAAA,GAAG,KAAO,EAAA,GAAA,EAAK,cACxB,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAI5B,MAAM,MAAA,WAAA,GAAc,YAAgB,IAAA,KAAA,KAAU,QAAW,GAAA,CAAA,CAAA;AACzD,MAAM,MAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,WAAY,EAAA,CAAA;AACnC,MAAA,QAAQ,KAAK,IAAM;AAAA,QACjB,KAAK,MAAA;AACH,UAAA,uBAAQA,cAAA,CAAA,QAAA,EAAA,EAAqB,IAAa,EAAA,GAAG,SAAvB,KAA8B,CAAA,CAAA;AAAA,QAEtD,KAAK,WAAA;AACH,UAAA,uBAAQA,cAAA,CAAA,aAAA,EAAA,EAA0B,IAAa,EAAA,GAAG,SAAvB,KAA8B,CAAA,CAAA;AAAA,QAE3D,KAAK,WAAA;AACH,UAAA,uBAAQA,cAAA,CAAA,aAAA,EAAA,EAA0B,IAAa,EAAA,GAAG,SAAvB,KAA8B,CAAA,CAAA;AAAA,QAE3D,KAAK,iBAAA;AACH,UACE,uBAAAA,cAAA;AAAA,YAAC,kBAAA;AAAA,YAAA;AAAA,cAEC,IAAA;AAAA,cACC,GAAG,KAAA;AAAA,cAEJ,OAAA;AAAA,aAAA;AAAA,YAJK,KAAA;AAAA,WAKP,CAAA;AAAA,QAGJ,KAAK,SAAA;AAEH,UACE,IAAA,OAAA,CAAQ,SAAS,WACjB,IAAA,OAAA,CAAQ,WAAW,WACnB,IAAA,OAAA,CAAQ,WAAW,QACnB,EAAA;AACA,YAAO,OAAA,IAAA,CAAA;AAAA,WACT;AAEA,UAAO,uBAAAA,cAAA,CAAC,WAAwB,EAAA,EAAA,IAAA,EAAA,EAAP,KAAmB,CAAA,CAAA;AAAA,QAE9C;AACE,UAAO,OAAA,IAAA,CAAA;AAAA,OACX;AAAA,KACD,CACH,EAAA,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AACjC;;;;"}

package/dist/primitives/AiMessage/index.js

@@ -8,24 +8,14 @@ import { AiMessageToolInvocation } from './tool-invocation.js';
 const AI_MESSAGE_CONTENT_NAME = "AiMessageContent";
 const defaultMessageContentComponents = {
   TextPart: ({ part }) => {
-    return /* @__PURE__ */ jsx(Markdown, {
-      content: part.text
-    });
+    return /* @__PURE__ */ jsx(Markdown, { content: part.text });
   },
   ReasoningPart: ({ part }) => {
-    return /* @__PURE__ */ jsx(Markdown, {
-      content: part.text
-    });
+    return /* @__PURE__ */ jsx(Markdown, { content: part.text });
   },
   RetrievalPart: () => null,
   ToolInvocationPart: ({ part, message }) => {
-    return /* @__PURE__ */ jsx(ErrorBoundary, {
-      fallback: null,
-      children: /* @__PURE__ */ jsx(AiMessageToolInvocation, {
-        part,
-        message
-      })
-    });
+    return /* @__PURE__ */ jsx(ErrorBoundary, { fallback: null, children: /* @__PURE__ */ jsx(AiMessageToolInvocation, { part, message }) });
   },
   SourcesPart: () => null
 };
@@ -45,46 +35,35 @@ const AiMessageContent = forwardRef(
     const content = message.content ?? message.contentSoFar;
     const numParts = content.length;
     const isGenerating = message.role === "assistant" && message.status === "generating";
-    return /* @__PURE__ */ jsx(Component, {
-      ...props,
-      ref: forwardedRef,
-      children: content.map((part, index) => {
-        const isStreaming = isGenerating && index === numParts - 1;
-        const extra = { index, isStreaming };
-        switch (part.type) {
-          case "text":
-            return /* @__PURE__ */ jsx(TextPart, {
-              part,
-              ...extra
-            }, index);
-          case "reasoning":
-            return /* @__PURE__ */ jsx(ReasoningPart, {
-              part,
-              ...extra
-            }, index);
-          case "retrieval":
-            return /* @__PURE__ */ jsx(RetrievalPart, {
-              part,
-              ...extra
-            }, index);
-          case "tool-invocation":
-            return /* @__PURE__ */ jsx(ToolInvocationPart, {
+    return /* @__PURE__ */ jsx(Component, { ...props, ref: forwardedRef, children: content.map((part, index) => {
+      const isStreaming = isGenerating && index === numParts - 1;
+      const extra = { index, isStreaming };
+      switch (part.type) {
+        case "text":
+          return /* @__PURE__ */ jsx(TextPart, { part, ...extra }, index);
+        case "reasoning":
+          return /* @__PURE__ */ jsx(ReasoningPart, { part, ...extra }, index);
+        case "retrieval":
+          return /* @__PURE__ */ jsx(RetrievalPart, { part, ...extra }, index);
+        case "tool-invocation":
+          return /* @__PURE__ */ jsx(
+            ToolInvocationPart,
+            {
               part,
               ...extra,
               message
-            }, index);
-          case "sources":
-            if (message.role === "assistant" && message.status !== "completed" && message.status !== "failed") {
-              return null;
-            }
-            return /* @__PURE__ */ jsx(SourcesPart, {
-              part
-            }, index);
-          default:
+            },
+            index
+          );
+        case "sources":
+          if (message.role === "assistant" && message.status !== "completed" && message.status !== "failed") {
             return null;
-        }
-      })
-    });
+          }
+          return /* @__PURE__ */ jsx(SourcesPart, { part }, index);
+        default:
+          return null;
+      }
+    }) });
   }
 );
 if (process.env.NODE_ENV !== "production") {

package/dist/primitives/AiMessage/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../../../src/primitives/AiMessage/index.tsx"],"sourcesContent":["import type { AiAssistantMessage } from \"@liveblocks/core\";\nimport { Slot } from \"@radix-ui/react-slot\";\nimport { forwardRef, useMemo } from \"react\";\n\nimport { ErrorBoundary } from \"../../utils/ErrorBoundary\";\nimport { Markdown } from \"../Markdown\";\nimport { AiMessageToolInvocation } from \"./tool-invocation\";\nimport type {\n  AiMessageContentComponents,\n  AiMessageContentProps,\n} from \"./types\";\n\nconst AI_MESSAGE_CONTENT_NAME = \"AiMessageContent\";\n\nconst defaultMessageContentComponents: AiMessageContentComponents = {\n  TextPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ReasoningPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  RetrievalPart: () => null,\n  ToolInvocationPart: ({ part, message }) => {\n    return (\n      <ErrorBoundary fallback={null}>\n        <AiMessageToolInvocation part={part} message={message} />\n      </ErrorBoundary>\n    );\n  },\n  SourcesPart: () => null,\n};\n\n/**\n * --------------------------------------------------------------------------\n * @private The API for this component is not yet stable.\n * --------------------------------------------------------------------------\n *\n * Primitive to help display an user or assistant message’s content, which is\n * an array of parts.\n *\n * @example\n * <AiMessage.Content message={message} components={{ TextPart }} />\n */\nconst AiMessageContent = forwardRef<HTMLDivElement, AiMessageContentProps>(\n  ({ message, components, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const {\n      ReasoningPart,\n      RetrievalPart,\n      TextPart,\n      ToolInvocationPart,\n      SourcesPart,\n    } = useMemo(\n      () => ({ ...defaultMessageContentComponents, ...components }),\n      [components]\n    );\n\n    const content = message.content ?? message.contentSoFar;\n    const numParts = content.length;\n    const isGenerating =\n      message.role === \"assistant\" && message.status === \"generating\";\n\n    return (\n      <Component {...props} ref={forwardedRef}>\n        {content.map((part, index) => {\n          // A part is considered to be still \"streaming in\" if it's the last\n          // part in the content array, and the message is in \"generating\"\n          // state.\n          const isStreaming = isGenerating && index === numParts - 1;\n          const extra = { index, isStreaming };\n          switch (part.type) {\n            case \"text\":\n              return <TextPart key={index} part={part} {...extra} />;\n\n            case \"reasoning\":\n              return <ReasoningPart key={index} part={part} {...extra} />;\n\n            case \"retrieval\":\n              return <RetrievalPart key={index} part={part} {...extra} />;\n\n            case \"tool-invocation\":\n              return (\n                <ToolInvocationPart\n                  key={index}\n                  part={part}\n                  {...extra}\n                  // For tool invocations, we know for sure it's an AiAssistantMessage\n                  message={message as AiAssistantMessage}\n                />\n              );\n\n            case \"sources\":\n              // Sources are only shown after the message is either completed or failed.\n              if (\n                message.role === \"assistant\" &&\n                message.status !== \"completed\" &&\n                message.status !== \"failed\"\n              ) {\n                return null;\n              }\n\n              return <SourcesPart key={index} part={part} />;\n\n            default:\n              return null;\n          }\n        })}\n      </Component>\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiMessage.*\nexport { AiMessageContent as Content };\n"],"names":[],"mappings":";;;;;;;AAYA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAEhC,MAAM,+BAA8D,GAAA;AAAA,EAClE,QAAU,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AACtB,IAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,aAAe,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AAC3B,IAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA;AAAA,MAAS,SAAS,IAAK,CAAA,IAAA;AAAA,KAAM,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,eAAe,MAAM,IAAA;AAAA,EACrB,kBAAoB,EAAA,CAAC,EAAE,IAAA,EAAM,SAAc,KAAA;AACzC,IAAA,uBACG,GAAA,CAAA,aAAA,EAAA;AAAA,MAAc,QAAU,EAAA,IAAA;AAAA,MACvB,QAAC,kBAAA,GAAA,CAAA,uBAAA,EAAA;AAAA,QAAwB,IAAA;AAAA,QAAY,OAAA;AAAA,OAAkB,CAAA;AAAA,KACzD,CAAA,CAAA;AAAA,GAEJ;AAAA,EACA,aAAa,MAAM,IAAA;AACrB,CAAA,CAAA;AAaA,MAAM,gBAAmB,GAAA,UAAA;AAAA,EACvB,CAAC,EAAE,OAAA,EAAS,YAAY,OAAY,EAAA,GAAA,KAAA,IAAS,YAAiB,KAAA;AAC5D,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,KAAA,CAAA;AACnC,IAAM,MAAA;AAAA,MACJ,aAAA;AAAA,MACA,aAAA;AAAA,MACA,QAAA;AAAA,MACA,kBAAA;AAAA,MACA,WAAA;AAAA,KACE,GAAA,OAAA;AAAA,MACF,OAAO,EAAE,GAAG,+BAAA,EAAiC,GAAG,UAAW,EAAA,CAAA;AAAA,MAC3D,CAAC,UAAU,CAAA;AAAA,KACb,CAAA;AAEA,IAAM,MAAA,OAAA,GAAU,OAAQ,CAAA,OAAA,IAAW,OAAQ,CAAA,YAAA,CAAA;AAC3C,IAAA,MAAM,WAAW,OAAQ,CAAA,MAAA,CAAA;AACzB,IAAA,MAAM,YACJ,GAAA,OAAA,CAAQ,IAAS,KAAA,WAAA,IAAe,QAAQ,MAAW,KAAA,YAAA,CAAA;AAErD,IAAA,uBACG,GAAA,CAAA,SAAA,EAAA;AAAA,MAAW,GAAG,KAAA;AAAA,MAAO,GAAK,EAAA,YAAA;AAAA,MACxB,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAI5B,QAAM,MAAA,WAAA,GAAc,YAAgB,IAAA,KAAA,KAAU,QAAW,GAAA,CAAA,CAAA;AACzD,QAAM,MAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,WAAY,EAAA,CAAA;AACnC,QAAA,QAAQ,KAAK,IAAM;AAAA,UACjB,KAAK,MAAA;AACH,YAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA;AAAA,cAAqB,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAEtD,KAAK,WAAA;AACH,YAAA,uBAAQ,GAAA,CAAA,aAAA,EAAA;AAAA,cAA0B,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAE3D,KAAK,WAAA;AACH,YAAA,uBAAQ,GAAA,CAAA,aAAA,EAAA;AAAA,cAA0B,IAAA;AAAA,cAAa,GAAG,KAAA;AAAA,aAAA,EAAvB,KAA8B,CAAA,CAAA;AAAA,UAE3D,KAAK,iBAAA;AACH,YAAA,uBACG,GAAA,CAAA,kBAAA,EAAA;AAAA,cAEC,IAAA;AAAA,cACC,GAAG,KAAA;AAAA,cAEJ,OAAA;AAAA,aAAA,EAJK,KAKP,CAAA,CAAA;AAAA,UAGJ,KAAK,SAAA;AAEH,YACE,IAAA,OAAA,CAAQ,SAAS,WACjB,IAAA,OAAA,CAAQ,WAAW,WACnB,IAAA,OAAA,CAAQ,WAAW,QACnB,EAAA;AACA,cAAO,OAAA,IAAA,CAAA;AAAA,aACT;AAEA,YAAA,uBAAQ,GAAA,CAAA,WAAA,EAAA;AAAA,cAAwB,IAAA;AAAA,aAAA,EAAP,KAAmB,CAAA,CAAA;AAAA,UAE9C;AACE,YAAO,OAAA,IAAA,CAAA;AAAA,SACX;AAAA,OACD,CAAA;AAAA,KACH,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AACjC;;;;"}
+{"version":3,"file":"index.js","sources":["../../../src/primitives/AiMessage/index.tsx"],"sourcesContent":["import type { AiAssistantMessage } from \"@liveblocks/core\";\nimport { Slot } from \"@radix-ui/react-slot\";\nimport { forwardRef, useMemo } from \"react\";\n\nimport { ErrorBoundary } from \"../../utils/ErrorBoundary\";\nimport { Markdown } from \"../Markdown\";\nimport { AiMessageToolInvocation } from \"./tool-invocation\";\nimport type {\n  AiMessageContentComponents,\n  AiMessageContentProps,\n} from \"./types\";\n\nconst AI_MESSAGE_CONTENT_NAME = \"AiMessageContent\";\n\nconst defaultMessageContentComponents: AiMessageContentComponents = {\n  TextPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  ReasoningPart: ({ part }) => {\n    return <Markdown content={part.text} />;\n  },\n  RetrievalPart: () => null,\n  ToolInvocationPart: ({ part, message }) => {\n    return (\n      <ErrorBoundary fallback={null}>\n        <AiMessageToolInvocation part={part} message={message} />\n      </ErrorBoundary>\n    );\n  },\n  SourcesPart: () => null,\n};\n\n/**\n * --------------------------------------------------------------------------\n * @private The API for this component is not yet stable.\n * --------------------------------------------------------------------------\n *\n * Primitive to help display an user or assistant message’s content, which is\n * an array of parts.\n *\n * @example\n * <AiMessage.Content message={message} components={{ TextPart }} />\n */\nconst AiMessageContent = forwardRef<HTMLDivElement, AiMessageContentProps>(\n  ({ message, components, asChild, ...props }, forwardedRef) => {\n    const Component = asChild ? Slot : \"div\";\n    const {\n      ReasoningPart,\n      RetrievalPart,\n      TextPart,\n      ToolInvocationPart,\n      SourcesPart,\n    } = useMemo(\n      () => ({ ...defaultMessageContentComponents, ...components }),\n      [components]\n    );\n\n    const content = message.content ?? message.contentSoFar;\n    const numParts = content.length;\n    const isGenerating =\n      message.role === \"assistant\" && message.status === \"generating\";\n\n    return (\n      <Component {...props} ref={forwardedRef}>\n        {content.map((part, index) => {\n          // A part is considered to be still \"streaming in\" if it's the last\n          // part in the content array, and the message is in \"generating\"\n          // state.\n          const isStreaming = isGenerating && index === numParts - 1;\n          const extra = { index, isStreaming };\n          switch (part.type) {\n            case \"text\":\n              return <TextPart key={index} part={part} {...extra} />;\n\n            case \"reasoning\":\n              return <ReasoningPart key={index} part={part} {...extra} />;\n\n            case \"retrieval\":\n              return <RetrievalPart key={index} part={part} {...extra} />;\n\n            case \"tool-invocation\":\n              return (\n                <ToolInvocationPart\n                  key={index}\n                  part={part}\n                  {...extra}\n                  // For tool invocations, we know for sure it's an AiAssistantMessage\n                  message={message as AiAssistantMessage}\n                />\n              );\n\n            case \"sources\":\n              // Sources are only shown after the message is either completed or failed.\n              if (\n                message.role === \"assistant\" &&\n                message.status !== \"completed\" &&\n                message.status !== \"failed\"\n              ) {\n                return null;\n              }\n\n              return <SourcesPart key={index} part={part} />;\n\n            default:\n              return null;\n          }\n        })}\n      </Component>\n    );\n  }\n);\n\nif (process.env.NODE_ENV !== \"production\") {\n  AiMessageContent.displayName = AI_MESSAGE_CONTENT_NAME;\n}\n\n// NOTE: Every export from this file will be available publicly as AiMessage.*\nexport { AiMessageContent as Content };\n"],"names":[],"mappings":";;;;;;;AAYA,MAAM,uBAA0B,GAAA,kBAAA,CAAA;AAEhC,MAAM,+BAA8D,GAAA;AAAA,EAClE,QAAU,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AACtB,IAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA,EAAS,OAAS,EAAA,IAAA,CAAK,IAAM,EAAA,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,aAAe,EAAA,CAAC,EAAE,IAAA,EAAW,KAAA;AAC3B,IAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA,EAAS,OAAS,EAAA,IAAA,CAAK,IAAM,EAAA,CAAA,CAAA;AAAA,GACvC;AAAA,EACA,eAAe,MAAM,IAAA;AAAA,EACrB,kBAAoB,EAAA,CAAC,EAAE,IAAA,EAAM,SAAc,KAAA;AACzC,IACE,uBAAA,GAAA,CAAC,iBAAc,QAAU,EAAA,IAAA,EACvB,8BAAC,uBAAwB,EAAA,EAAA,IAAA,EAAY,SAAkB,CACzD,EAAA,CAAA,CAAA;AAAA,GAEJ;AAAA,EACA,aAAa,MAAM,IAAA;AACrB,CAAA,CAAA;AAaA,MAAM,gBAAmB,GAAA,UAAA;AAAA,EACvB,CAAC,EAAE,OAAS,EAAA,UAAA,EAAY,SAAS,GAAG,KAAA,IAAS,YAAiB,KAAA;AAC5D,IAAM,MAAA,SAAA,GAAY,UAAU,IAAO,GAAA,KAAA,CAAA;AACnC,IAAM,MAAA;AAAA,MACJ,aAAA;AAAA,MACA,aAAA;AAAA,MACA,QAAA;AAAA,MACA,kBAAA;AAAA,MACA,WAAA;AAAA,KACE,GAAA,OAAA;AAAA,MACF,OAAO,EAAE,GAAG,+BAAA,EAAiC,GAAG,UAAW,EAAA,CAAA;AAAA,MAC3D,CAAC,UAAU,CAAA;AAAA,KACb,CAAA;AAEA,IAAM,MAAA,OAAA,GAAU,OAAQ,CAAA,OAAA,IAAW,OAAQ,CAAA,YAAA,CAAA;AAC3C,IAAA,MAAM,WAAW,OAAQ,CAAA,MAAA,CAAA;AACzB,IAAA,MAAM,YACJ,GAAA,OAAA,CAAQ,IAAS,KAAA,WAAA,IAAe,QAAQ,MAAW,KAAA,YAAA,CAAA;AAErD,IACE,uBAAA,GAAA,CAAC,SAAW,EAAA,EAAA,GAAG,KAAO,EAAA,GAAA,EAAK,cACxB,QAAQ,EAAA,OAAA,CAAA,GAAA,CAAI,CAAC,IAAA,EAAM,KAAU,KAAA;AAI5B,MAAM,MAAA,WAAA,GAAc,YAAgB,IAAA,KAAA,KAAU,QAAW,GAAA,CAAA,CAAA;AACzD,MAAM,MAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,WAAY,EAAA,CAAA;AACnC,MAAA,QAAQ,KAAK,IAAM;AAAA,QACjB,KAAK,MAAA;AACH,UAAA,uBAAQ,GAAA,CAAA,QAAA,EAAA,EAAqB,IAAa,EAAA,GAAG,SAAvB,KAA8B,CAAA,CAAA;AAAA,QAEtD,KAAK,WAAA;AACH,UAAA,uBAAQ,GAAA,CAAA,aAAA,EAAA,EAA0B,IAAa,EAAA,GAAG,SAAvB,KAA8B,CAAA,CAAA;AAAA,QAE3D,KAAK,WAAA;AACH,UAAA,uBAAQ,GAAA,CAAA,aAAA,EAAA,EAA0B,IAAa,EAAA,GAAG,SAAvB,KAA8B,CAAA,CAAA;AAAA,QAE3D,KAAK,iBAAA;AACH,UACE,uBAAA,GAAA;AAAA,YAAC,kBAAA;AAAA,YAAA;AAAA,cAEC,IAAA;AAAA,cACC,GAAG,KAAA;AAAA,cAEJ,OAAA;AAAA,aAAA;AAAA,YAJK,KAAA;AAAA,WAKP,CAAA;AAAA,QAGJ,KAAK,SAAA;AAEH,UACE,IAAA,OAAA,CAAQ,SAAS,WACjB,IAAA,OAAA,CAAQ,WAAW,WACnB,IAAA,OAAA,CAAQ,WAAW,QACnB,EAAA;AACA,YAAO,OAAA,IAAA,CAAA;AAAA,WACT;AAEA,UAAO,uBAAA,GAAA,CAAC,WAAwB,EAAA,EAAA,IAAA,EAAA,EAAP,KAAmB,CAAA,CAAA;AAAA,QAE9C;AACE,UAAO,OAAA,IAAA,CAAA;AAAA,OACX;AAAA,KACD,CACH,EAAA,CAAA,CAAA;AAAA,GAEJ;AACF,EAAA;AAEA,IAAI,OAAA,CAAQ,GAAI,CAAA,QAAA,KAAa,YAAc,EAAA;AACzC,EAAA,gBAAA,CAAiB,WAAc,GAAA,uBAAA,CAAA;AACjC;;;;"}

package/dist/primitives/AiMessage/tool-invocation.cjs

@@ -62,13 +62,13 @@ function AiMessageToolInvocation({
   }, [part, respond, tool?.execute, message.status, partialArgs]);
   if (tool?.render === void 0)
     return null;
-  return /* @__PURE__ */ jsxRuntime.jsx(contexts.AiToolInvocationContext.Provider, {
-    value: props,
-    children: /* @__PURE__ */ jsxRuntime.jsx(StableRenderFn, {
+  return /* @__PURE__ */ jsxRuntime.jsx(contexts.AiToolInvocationContext.Provider, { value: props, children: /* @__PURE__ */ jsxRuntime.jsx(
+    StableRenderFn,
+    {
       renderFn: tool.render,
       props
-    })
-  });
+    }
+  ) });
 }
 
 exports.AiMessageToolInvocation = AiMessageToolInvocation;

package/dist/primitives/AiMessage/tool-invocation.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"tool-invocation.cjs","sources":["../../../src/primitives/AiMessage/tool-invocation.tsx"],"sourcesContent":["import type {\n  AiAssistantMessage,\n  AiToolInvocationPart,\n  AiToolInvocationProps,\n  JsonObject,\n  ToolResultResponse,\n} from \"@liveblocks/core\";\nimport { kInternal } from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { type FunctionComponent, useCallback, useMemo } from \"react\";\n\nimport { AiToolInvocationContext } from \"./contexts\";\n\ntype OpaqueAiToolInvocationProps = AiToolInvocationProps<\n  JsonObject,\n  JsonObject\n>;\n\nfunction StableRenderFn(props: {\n  renderFn: FunctionComponent<OpaqueAiToolInvocationProps>;\n  props: OpaqueAiToolInvocationProps;\n}) {\n  return props.renderFn(props.props);\n}\n\n/**\n * @internal\n *\n * This could become publicly exposed as <AiMessage.ToolInvocation /> in the future,\n * but because namespace exports can't be marked `@internal`, we're keeping it in its\n * own file for now.\n */\nexport function AiMessageToolInvocation({\n  message,\n  part,\n}: {\n  message: AiAssistantMessage;\n  part: AiToolInvocationPart;\n}) {\n  const client = useClient();\n  const ai = client[kInternal].ai;\n  const tool = useSignal(ai.signals.getToolΣ(part.name, message.chatId));\n\n  const respond = useCallback(\n    (result: ToolResultResponse | undefined) => {\n      if (message.status !== \"awaiting-tool\") {\n        // console.log(\"Ignoring respond(): message not awaiting tool result\");\n      } else if (part.stage === \"receiving\") {\n        // console.log(\n        //   `Ignoring respond(): tool '${part.name}' (${part.invocationId}) is still receiving`\n        // );\n      } else if (part.stage === \"executed\") {\n        console.log(\n          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`\n        );\n      } else {\n        ai.setToolResult(\n          message.chatId,\n          message.id,\n          part.invocationId,\n          result ?? { data: {} },\n          { copilotId: message.copilotId }\n        );\n      }\n    },\n    [\n      ai,\n      message.chatId,\n      message.id,\n      message.status,\n      part.invocationId,\n      part.name,\n      part.stage,\n      message.copilotId,\n    ]\n  );\n\n  const partialArgs = part?.partialArgs;\n  const props = useMemo(() => {\n    // NOTE: Not really used, but necessary to trick useMemo into re-evaluating\n    // when it changes. Without this trick, tool call streaming won't\n    // re-render. The reason this is needed is that `part` gets mutated\n    // in-place by the delta handling, rather than part being replaced by a new\n    // object on every chunk.\n    partialArgs;\n\n    const { type: _, ...rest } = part;\n    return {\n      ...rest,\n      respond,\n      types: undefined as never,\n      [kInternal]: {\n        execute: tool?.execute,\n        messageStatus: message.status,\n      },\n    } satisfies OpaqueAiToolInvocationProps;\n  }, [part, respond, tool?.execute, message.status, partialArgs]);\n\n  if (tool?.render === undefined) return null;\n  return (\n    <AiToolInvocationContext.Provider value={props}>\n      <StableRenderFn\n        renderFn={tool.render as FunctionComponent<OpaqueAiToolInvocationProps>}\n        props={props}\n      />\n    </AiToolInvocationContext.Provider>\n  );\n}\n"],"names":["useClient","kInternal","useSignal","useCallback","useMemo","jsx","AiToolInvocationContext"],"mappings":";;;;;;;;;AAmBA,SAAS,eAAe,KAGrB,EAAA;AACD,EAAO,OAAA,KAAA,CAAM,QAAS,CAAA,KAAA,CAAM,KAAK,CAAA,CAAA;AACnC,CAAA;AASO,SAAS,uBAAwB,CAAA;AAAA,EACtC,OAAA;AAAA,EACA,IAAA;AACF,CAGG,EAAA;AACD,EAAA,MAAM,SAASA,eAAU,EAAA,CAAA;AACzB,EAAM,MAAA,EAAA,GAAK,OAAOC,cAAW,CAAA,CAAA,EAAA,CAAA;AAC7B,EAAM,MAAA,IAAA,GAAOC,mBAAU,EAAG,CAAA,OAAA,CAAQ,cAAS,IAAK,CAAA,IAAA,EAAM,OAAQ,CAAA,MAAM,CAAC,CAAA,CAAA;AAErE,EAAA,MAAM,OAAU,GAAAC,mBAAA;AAAA,IACd,CAAC,MAA2C,KAAA;AAC1C,MAAI,IAAA,OAAA,CAAQ,WAAW,eAAiB,EAAA;AAAA,OAExC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,WAAa,EAAA;AAAA,OAIvC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,UAAY,EAAA;AACpC,QAAQ,OAAA,CAAA,GAAA;AAAA,UACN,CAAA,0BAAA,EAA6B,IAAK,CAAA,IAAA,CAAA,GAAA,EAAU,IAAK,CAAA,YAAA,CAAA,sBAAA,CAAA;AAAA,SACnD,CAAA;AAAA,OACK,MAAA;AACL,QAAG,EAAA,CAAA,aAAA;AAAA,UACD,OAAQ,CAAA,MAAA;AAAA,UACR,OAAQ,CAAA,EAAA;AAAA,UACR,IAAK,CAAA,YAAA;AAAA,UACL,MAAU,IAAA,EAAE,IAAM,EAAA,EAAG,EAAA;AAAA,UACrB,EAAE,SAAW,EAAA,OAAA,CAAQ,SAAU,EAAA;AAAA,SACjC,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA;AAAA,MACE,EAAA;AAAA,MACA,OAAQ,CAAA,MAAA;AAAA,MACR,OAAQ,CAAA,EAAA;AAAA,MACR,OAAQ,CAAA,MAAA;AAAA,MACR,IAAK,CAAA,YAAA;AAAA,MACL,IAAK,CAAA,IAAA;AAAA,MACL,IAAK,CAAA,KAAA;AAAA,MACL,OAAQ,CAAA,SAAA;AAAA,KACV;AAAA,GACF,CAAA;AAEA,EAAA,MAAM,cAAc,IAAM,EAAA,WAAA,CAAA;AAC1B,EAAM,MAAA,KAAA,GAAQC,gBAAQ,MAAM;AAM1B,IAAA,WAAA,CAAA;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,CAAM,EAAA,GAAA,IAAA,EAAS,GAAA,IAAA,CAAA;AAC7B,IAAO,OAAA;AAAA,MACL,GAAG,IAAA;AAAA,MACH,OAAA;AAAA,MACA,KAAO,EAAA,KAAA,CAAA;AAAA,MACP,CAACH,cAAY,GAAA;AAAA,QACX,SAAS,IAAM,EAAA,OAAA;AAAA,QACf,eAAe,OAAQ,CAAA,MAAA;AAAA,OACzB;AAAA,KACF,CAAA;AAAA,GACF,EAAG,CAAC,IAAM,EAAA,OAAA,EAAS,MAAM,OAAS,EAAA,OAAA,CAAQ,MAAQ,EAAA,WAAW,CAAC,CAAA,CAAA;AAE9D,EAAA,IAAI,MAAM,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AACvC,EACE,uBAAAI,cAAA,CAACC,iCAAwB,QAAxB,EAAA;AAAA,IAAiC,KAAO,EAAA,KAAA;AAAA,IACvC,QAAC,kBAAAD,cAAA,CAAA,cAAA,EAAA;AAAA,MACC,UAAU,IAAK,CAAA,MAAA;AAAA,MACf,KAAA;AAAA,KACF,CAAA;AAAA,GACF,CAAA,CAAA;AAEJ;;;;"}
+{"version":3,"file":"tool-invocation.cjs","sources":["../../../src/primitives/AiMessage/tool-invocation.tsx"],"sourcesContent":["import type {\n  AiAssistantMessage,\n  AiToolInvocationPart,\n  AiToolInvocationProps,\n  JsonObject,\n  ToolResultResponse,\n} from \"@liveblocks/core\";\nimport { kInternal } from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { type FunctionComponent, useCallback, useMemo } from \"react\";\n\nimport { AiToolInvocationContext } from \"./contexts\";\n\ntype OpaqueAiToolInvocationProps = AiToolInvocationProps<\n  JsonObject,\n  JsonObject\n>;\n\nfunction StableRenderFn(props: {\n  renderFn: FunctionComponent<OpaqueAiToolInvocationProps>;\n  props: OpaqueAiToolInvocationProps;\n}) {\n  return props.renderFn(props.props);\n}\n\n/**\n * @internal\n *\n * This could become publicly exposed as <AiMessage.ToolInvocation /> in the future,\n * but because namespace exports can't be marked `@internal`, we're keeping it in its\n * own file for now.\n */\nexport function AiMessageToolInvocation({\n  message,\n  part,\n}: {\n  message: AiAssistantMessage;\n  part: AiToolInvocationPart;\n}) {\n  const client = useClient();\n  const ai = client[kInternal].ai;\n  const tool = useSignal(ai.signals.getToolΣ(part.name, message.chatId));\n\n  const respond = useCallback(\n    (result: ToolResultResponse | undefined) => {\n      if (message.status !== \"awaiting-tool\") {\n        // console.log(\"Ignoring respond(): message not awaiting tool result\");\n      } else if (part.stage === \"receiving\") {\n        // console.log(\n        //   `Ignoring respond(): tool '${part.name}' (${part.invocationId}) is still receiving`\n        // );\n      } else if (part.stage === \"executed\") {\n        console.log(\n          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`\n        );\n      } else {\n        ai.setToolResult(\n          message.chatId,\n          message.id,\n          part.invocationId,\n          result ?? { data: {} },\n          { copilotId: message.copilotId }\n        );\n      }\n    },\n    [\n      ai,\n      message.chatId,\n      message.id,\n      message.status,\n      part.invocationId,\n      part.name,\n      part.stage,\n      message.copilotId,\n    ]\n  );\n\n  const partialArgs = part?.partialArgs;\n  const props = useMemo(() => {\n    // NOTE: Not really used, but necessary to trick useMemo into re-evaluating\n    // when it changes. Without this trick, tool call streaming won't\n    // re-render. The reason this is needed is that `part` gets mutated\n    // in-place by the delta handling, rather than part being replaced by a new\n    // object on every chunk.\n    partialArgs;\n\n    const { type: _, ...rest } = part;\n    return {\n      ...rest,\n      respond,\n      types: undefined as never,\n      [kInternal]: {\n        execute: tool?.execute,\n        messageStatus: message.status,\n      },\n    } satisfies OpaqueAiToolInvocationProps;\n  }, [part, respond, tool?.execute, message.status, partialArgs]);\n\n  if (tool?.render === undefined) return null;\n  return (\n    <AiToolInvocationContext.Provider value={props}>\n      <StableRenderFn\n        renderFn={tool.render as FunctionComponent<OpaqueAiToolInvocationProps>}\n        props={props}\n      />\n    </AiToolInvocationContext.Provider>\n  );\n}\n"],"names":["useClient","kInternal","useSignal","useCallback","useMemo","jsx","AiToolInvocationContext"],"mappings":";;;;;;;;;AAmBA,SAAS,eAAe,KAGrB,EAAA;AACD,EAAO,OAAA,KAAA,CAAM,QAAS,CAAA,KAAA,CAAM,KAAK,CAAA,CAAA;AACnC,CAAA;AASO,SAAS,uBAAwB,CAAA;AAAA,EACtC,OAAA;AAAA,EACA,IAAA;AACF,CAGG,EAAA;AACD,EAAA,MAAM,SAASA,eAAU,EAAA,CAAA;AACzB,EAAM,MAAA,EAAA,GAAK,MAAO,CAAAC,cAAS,CAAE,CAAA,EAAA,CAAA;AAC7B,EAAM,MAAA,IAAA,GAAOC,mBAAU,EAAG,CAAA,OAAA,CAAQ,cAAS,IAAK,CAAA,IAAA,EAAM,OAAQ,CAAA,MAAM,CAAC,CAAA,CAAA;AAErE,EAAA,MAAM,OAAU,GAAAC,mBAAA;AAAA,IACd,CAAC,MAA2C,KAAA;AAC1C,MAAI,IAAA,OAAA,CAAQ,WAAW,eAAiB,EAAA;AAAA,OAExC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,WAAa,EAAA;AAAA,OAIvC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,UAAY,EAAA;AACpC,QAAQ,OAAA,CAAA,GAAA;AAAA,UACN,CAA6B,0BAAA,EAAA,IAAA,CAAK,IAAI,CAAA,GAAA,EAAM,KAAK,YAAY,CAAA,sBAAA,CAAA;AAAA,SAC/D,CAAA;AAAA,OACK,MAAA;AACL,QAAG,EAAA,CAAA,aAAA;AAAA,UACD,OAAQ,CAAA,MAAA;AAAA,UACR,OAAQ,CAAA,EAAA;AAAA,UACR,IAAK,CAAA,YAAA;AAAA,UACL,MAAU,IAAA,EAAE,IAAM,EAAA,EAAG,EAAA;AAAA,UACrB,EAAE,SAAW,EAAA,OAAA,CAAQ,SAAU,EAAA;AAAA,SACjC,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA;AAAA,MACE,EAAA;AAAA,MACA,OAAQ,CAAA,MAAA;AAAA,MACR,OAAQ,CAAA,EAAA;AAAA,MACR,OAAQ,CAAA,MAAA;AAAA,MACR,IAAK,CAAA,YAAA;AAAA,MACL,IAAK,CAAA,IAAA;AAAA,MACL,IAAK,CAAA,KAAA;AAAA,MACL,OAAQ,CAAA,SAAA;AAAA,KACV;AAAA,GACF,CAAA;AAEA,EAAA,MAAM,cAAc,IAAM,EAAA,WAAA,CAAA;AAC1B,EAAM,MAAA,KAAA,GAAQC,gBAAQ,MAAM;AAM1B,IAAA,WAAA,CAAA;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,CAAG,EAAA,GAAG,MAAS,GAAA,IAAA,CAAA;AAC7B,IAAO,OAAA;AAAA,MACL,GAAG,IAAA;AAAA,MACH,OAAA;AAAA,MACA,KAAO,EAAA,KAAA,CAAA;AAAA,MACP,CAACH,cAAS,GAAG;AAAA,QACX,SAAS,IAAM,EAAA,OAAA;AAAA,QACf,eAAe,OAAQ,CAAA,MAAA;AAAA,OACzB;AAAA,KACF,CAAA;AAAA,GACF,EAAG,CAAC,IAAM,EAAA,OAAA,EAAS,MAAM,OAAS,EAAA,OAAA,CAAQ,MAAQ,EAAA,WAAW,CAAC,CAAA,CAAA;AAE9D,EAAA,IAAI,MAAM,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AACvC,EAAA,uBACGI,cAAA,CAAAC,gCAAA,CAAwB,QAAxB,EAAA,EAAiC,OAAO,KACvC,EAAA,QAAA,kBAAAD,cAAA;AAAA,IAAC,cAAA;AAAA,IAAA;AAAA,MACC,UAAU,IAAK,CAAA,MAAA;AAAA,MACf,KAAA;AAAA,KAAA;AAAA,GAEJ,EAAA,CAAA,CAAA;AAEJ;;;;"}

package/dist/primitives/AiMessage/tool-invocation.js

@@ -60,13 +60,13 @@ function AiMessageToolInvocation({
   }, [part, respond, tool?.execute, message.status, partialArgs]);
   if (tool?.render === void 0)
     return null;
-  return /* @__PURE__ */ jsx(AiToolInvocationContext.Provider, {
-    value: props,
-    children: /* @__PURE__ */ jsx(StableRenderFn, {
+  return /* @__PURE__ */ jsx(AiToolInvocationContext.Provider, { value: props, children: /* @__PURE__ */ jsx(
+    StableRenderFn,
+    {
       renderFn: tool.render,
       props
-    })
-  });
+    }
+  ) });
 }
 
 export { AiMessageToolInvocation };

package/dist/primitives/AiMessage/tool-invocation.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-invocation.js","sources":["../../../src/primitives/AiMessage/tool-invocation.tsx"],"sourcesContent":["import type {\n  AiAssistantMessage,\n  AiToolInvocationPart,\n  AiToolInvocationProps,\n  JsonObject,\n  ToolResultResponse,\n} from \"@liveblocks/core\";\nimport { kInternal } from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { type FunctionComponent, useCallback, useMemo } from \"react\";\n\nimport { AiToolInvocationContext } from \"./contexts\";\n\ntype OpaqueAiToolInvocationProps = AiToolInvocationProps<\n  JsonObject,\n  JsonObject\n>;\n\nfunction StableRenderFn(props: {\n  renderFn: FunctionComponent<OpaqueAiToolInvocationProps>;\n  props: OpaqueAiToolInvocationProps;\n}) {\n  return props.renderFn(props.props);\n}\n\n/**\n * @internal\n *\n * This could become publicly exposed as <AiMessage.ToolInvocation /> in the future,\n * but because namespace exports can't be marked `@internal`, we're keeping it in its\n * own file for now.\n */\nexport function AiMessageToolInvocation({\n  message,\n  part,\n}: {\n  message: AiAssistantMessage;\n  part: AiToolInvocationPart;\n}) {\n  const client = useClient();\n  const ai = client[kInternal].ai;\n  const tool = useSignal(ai.signals.getToolΣ(part.name, message.chatId));\n\n  const respond = useCallback(\n    (result: ToolResultResponse | undefined) => {\n      if (message.status !== \"awaiting-tool\") {\n        // console.log(\"Ignoring respond(): message not awaiting tool result\");\n      } else if (part.stage === \"receiving\") {\n        // console.log(\n        //   `Ignoring respond(): tool '${part.name}' (${part.invocationId}) is still receiving`\n        // );\n      } else if (part.stage === \"executed\") {\n        console.log(\n          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`\n        );\n      } else {\n        ai.setToolResult(\n          message.chatId,\n          message.id,\n          part.invocationId,\n          result ?? { data: {} },\n          { copilotId: message.copilotId }\n        );\n      }\n    },\n    [\n      ai,\n      message.chatId,\n      message.id,\n      message.status,\n      part.invocationId,\n      part.name,\n      part.stage,\n      message.copilotId,\n    ]\n  );\n\n  const partialArgs = part?.partialArgs;\n  const props = useMemo(() => {\n    // NOTE: Not really used, but necessary to trick useMemo into re-evaluating\n    // when it changes. Without this trick, tool call streaming won't\n    // re-render. The reason this is needed is that `part` gets mutated\n    // in-place by the delta handling, rather than part being replaced by a new\n    // object on every chunk.\n    partialArgs;\n\n    const { type: _, ...rest } = part;\n    return {\n      ...rest,\n      respond,\n      types: undefined as never,\n      [kInternal]: {\n        execute: tool?.execute,\n        messageStatus: message.status,\n      },\n    } satisfies OpaqueAiToolInvocationProps;\n  }, [part, respond, tool?.execute, message.status, partialArgs]);\n\n  if (tool?.render === undefined) return null;\n  return (\n    <AiToolInvocationContext.Provider value={props}>\n      <StableRenderFn\n        renderFn={tool.render as FunctionComponent<OpaqueAiToolInvocationProps>}\n        props={props}\n      />\n    </AiToolInvocationContext.Provider>\n  );\n}\n"],"names":[],"mappings":";;;;;;;AAmBA,SAAS,eAAe,KAGrB,EAAA;AACD,EAAO,OAAA,KAAA,CAAM,QAAS,CAAA,KAAA,CAAM,KAAK,CAAA,CAAA;AACnC,CAAA;AASO,SAAS,uBAAwB,CAAA;AAAA,EACtC,OAAA;AAAA,EACA,IAAA;AACF,CAGG,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,cAAS,IAAK,CAAA,IAAA,EAAM,OAAQ,CAAA,MAAM,CAAC,CAAA,CAAA;AAErE,EAAA,MAAM,OAAU,GAAA,WAAA;AAAA,IACd,CAAC,MAA2C,KAAA;AAC1C,MAAI,IAAA,OAAA,CAAQ,WAAW,eAAiB,EAAA;AAAA,OAExC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,WAAa,EAAA;AAAA,OAIvC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,UAAY,EAAA;AACpC,QAAQ,OAAA,CAAA,GAAA;AAAA,UACN,CAAA,0BAAA,EAA6B,IAAK,CAAA,IAAA,CAAA,GAAA,EAAU,IAAK,CAAA,YAAA,CAAA,sBAAA,CAAA;AAAA,SACnD,CAAA;AAAA,OACK,MAAA;AACL,QAAG,EAAA,CAAA,aAAA;AAAA,UACD,OAAQ,CAAA,MAAA;AAAA,UACR,OAAQ,CAAA,EAAA;AAAA,UACR,IAAK,CAAA,YAAA;AAAA,UACL,MAAU,IAAA,EAAE,IAAM,EAAA,EAAG,EAAA;AAAA,UACrB,EAAE,SAAW,EAAA,OAAA,CAAQ,SAAU,EAAA;AAAA,SACjC,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA;AAAA,MACE,EAAA;AAAA,MACA,OAAQ,CAAA,MAAA;AAAA,MACR,OAAQ,CAAA,EAAA;AAAA,MACR,OAAQ,CAAA,MAAA;AAAA,MACR,IAAK,CAAA,YAAA;AAAA,MACL,IAAK,CAAA,IAAA;AAAA,MACL,IAAK,CAAA,KAAA;AAAA,MACL,OAAQ,CAAA,SAAA;AAAA,KACV;AAAA,GACF,CAAA;AAEA,EAAA,MAAM,cAAc,IAAM,EAAA,WAAA,CAAA;AAC1B,EAAM,MAAA,KAAA,GAAQ,QAAQ,MAAM;AAM1B,IAAA,WAAA,CAAA;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,CAAM,EAAA,GAAA,IAAA,EAAS,GAAA,IAAA,CAAA;AAC7B,IAAO,OAAA;AAAA,MACL,GAAG,IAAA;AAAA,MACH,OAAA;AAAA,MACA,KAAO,EAAA,KAAA,CAAA;AAAA,MACP,CAAC,SAAY,GAAA;AAAA,QACX,SAAS,IAAM,EAAA,OAAA;AAAA,QACf,eAAe,OAAQ,CAAA,MAAA;AAAA,OACzB;AAAA,KACF,CAAA;AAAA,GACF,EAAG,CAAC,IAAM,EAAA,OAAA,EAAS,MAAM,OAAS,EAAA,OAAA,CAAQ,MAAQ,EAAA,WAAW,CAAC,CAAA,CAAA;AAE9D,EAAA,IAAI,MAAM,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AACvC,EACE,uBAAA,GAAA,CAAC,wBAAwB,QAAxB,EAAA;AAAA,IAAiC,KAAO,EAAA,KAAA;AAAA,IACvC,QAAC,kBAAA,GAAA,CAAA,cAAA,EAAA;AAAA,MACC,UAAU,IAAK,CAAA,MAAA;AAAA,MACf,KAAA;AAAA,KACF,CAAA;AAAA,GACF,CAAA,CAAA;AAEJ;;;;"}
+{"version":3,"file":"tool-invocation.js","sources":["../../../src/primitives/AiMessage/tool-invocation.tsx"],"sourcesContent":["import type {\n  AiAssistantMessage,\n  AiToolInvocationPart,\n  AiToolInvocationProps,\n  JsonObject,\n  ToolResultResponse,\n} from \"@liveblocks/core\";\nimport { kInternal } from \"@liveblocks/core\";\nimport { useClient } from \"@liveblocks/react\";\nimport { useSignal } from \"@liveblocks/react/_private\";\nimport { type FunctionComponent, useCallback, useMemo } from \"react\";\n\nimport { AiToolInvocationContext } from \"./contexts\";\n\ntype OpaqueAiToolInvocationProps = AiToolInvocationProps<\n  JsonObject,\n  JsonObject\n>;\n\nfunction StableRenderFn(props: {\n  renderFn: FunctionComponent<OpaqueAiToolInvocationProps>;\n  props: OpaqueAiToolInvocationProps;\n}) {\n  return props.renderFn(props.props);\n}\n\n/**\n * @internal\n *\n * This could become publicly exposed as <AiMessage.ToolInvocation /> in the future,\n * but because namespace exports can't be marked `@internal`, we're keeping it in its\n * own file for now.\n */\nexport function AiMessageToolInvocation({\n  message,\n  part,\n}: {\n  message: AiAssistantMessage;\n  part: AiToolInvocationPart;\n}) {\n  const client = useClient();\n  const ai = client[kInternal].ai;\n  const tool = useSignal(ai.signals.getToolΣ(part.name, message.chatId));\n\n  const respond = useCallback(\n    (result: ToolResultResponse | undefined) => {\n      if (message.status !== \"awaiting-tool\") {\n        // console.log(\"Ignoring respond(): message not awaiting tool result\");\n      } else if (part.stage === \"receiving\") {\n        // console.log(\n        //   `Ignoring respond(): tool '${part.name}' (${part.invocationId}) is still receiving`\n        // );\n      } else if (part.stage === \"executed\") {\n        console.log(\n          `Ignoring respond(): tool '${part.name}' (${part.invocationId}) has already executed`\n        );\n      } else {\n        ai.setToolResult(\n          message.chatId,\n          message.id,\n          part.invocationId,\n          result ?? { data: {} },\n          { copilotId: message.copilotId }\n        );\n      }\n    },\n    [\n      ai,\n      message.chatId,\n      message.id,\n      message.status,\n      part.invocationId,\n      part.name,\n      part.stage,\n      message.copilotId,\n    ]\n  );\n\n  const partialArgs = part?.partialArgs;\n  const props = useMemo(() => {\n    // NOTE: Not really used, but necessary to trick useMemo into re-evaluating\n    // when it changes. Without this trick, tool call streaming won't\n    // re-render. The reason this is needed is that `part` gets mutated\n    // in-place by the delta handling, rather than part being replaced by a new\n    // object on every chunk.\n    partialArgs;\n\n    const { type: _, ...rest } = part;\n    return {\n      ...rest,\n      respond,\n      types: undefined as never,\n      [kInternal]: {\n        execute: tool?.execute,\n        messageStatus: message.status,\n      },\n    } satisfies OpaqueAiToolInvocationProps;\n  }, [part, respond, tool?.execute, message.status, partialArgs]);\n\n  if (tool?.render === undefined) return null;\n  return (\n    <AiToolInvocationContext.Provider value={props}>\n      <StableRenderFn\n        renderFn={tool.render as FunctionComponent<OpaqueAiToolInvocationProps>}\n        props={props}\n      />\n    </AiToolInvocationContext.Provider>\n  );\n}\n"],"names":[],"mappings":";;;;;;;AAmBA,SAAS,eAAe,KAGrB,EAAA;AACD,EAAO,OAAA,KAAA,CAAM,QAAS,CAAA,KAAA,CAAM,KAAK,CAAA,CAAA;AACnC,CAAA;AASO,SAAS,uBAAwB,CAAA;AAAA,EACtC,OAAA;AAAA,EACA,IAAA;AACF,CAGG,EAAA;AACD,EAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AACzB,EAAM,MAAA,EAAA,GAAK,MAAO,CAAA,SAAS,CAAE,CAAA,EAAA,CAAA;AAC7B,EAAM,MAAA,IAAA,GAAO,UAAU,EAAG,CAAA,OAAA,CAAQ,cAAS,IAAK,CAAA,IAAA,EAAM,OAAQ,CAAA,MAAM,CAAC,CAAA,CAAA;AAErE,EAAA,MAAM,OAAU,GAAA,WAAA;AAAA,IACd,CAAC,MAA2C,KAAA;AAC1C,MAAI,IAAA,OAAA,CAAQ,WAAW,eAAiB,EAAA;AAAA,OAExC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,WAAa,EAAA;AAAA,OAIvC,MAAA,IAAW,IAAK,CAAA,KAAA,KAAU,UAAY,EAAA;AACpC,QAAQ,OAAA,CAAA,GAAA;AAAA,UACN,CAA6B,0BAAA,EAAA,IAAA,CAAK,IAAI,CAAA,GAAA,EAAM,KAAK,YAAY,CAAA,sBAAA,CAAA;AAAA,SAC/D,CAAA;AAAA,OACK,MAAA;AACL,QAAG,EAAA,CAAA,aAAA;AAAA,UACD,OAAQ,CAAA,MAAA;AAAA,UACR,OAAQ,CAAA,EAAA;AAAA,UACR,IAAK,CAAA,YAAA;AAAA,UACL,MAAU,IAAA,EAAE,IAAM,EAAA,EAAG,EAAA;AAAA,UACrB,EAAE,SAAW,EAAA,OAAA,CAAQ,SAAU,EAAA;AAAA,SACjC,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA;AAAA,MACE,EAAA;AAAA,MACA,OAAQ,CAAA,MAAA;AAAA,MACR,OAAQ,CAAA,EAAA;AAAA,MACR,OAAQ,CAAA,MAAA;AAAA,MACR,IAAK,CAAA,YAAA;AAAA,MACL,IAAK,CAAA,IAAA;AAAA,MACL,IAAK,CAAA,KAAA;AAAA,MACL,OAAQ,CAAA,SAAA;AAAA,KACV;AAAA,GACF,CAAA;AAEA,EAAA,MAAM,cAAc,IAAM,EAAA,WAAA,CAAA;AAC1B,EAAM,MAAA,KAAA,GAAQ,QAAQ,MAAM;AAM1B,IAAA,WAAA,CAAA;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,CAAG,EAAA,GAAG,MAAS,GAAA,IAAA,CAAA;AAC7B,IAAO,OAAA;AAAA,MACL,GAAG,IAAA;AAAA,MACH,OAAA;AAAA,MACA,KAAO,EAAA,KAAA,CAAA;AAAA,MACP,CAAC,SAAS,GAAG;AAAA,QACX,SAAS,IAAM,EAAA,OAAA;AAAA,QACf,eAAe,OAAQ,CAAA,MAAA;AAAA,OACzB;AAAA,KACF,CAAA;AAAA,GACF,EAAG,CAAC,IAAM,EAAA,OAAA,EAAS,MAAM,OAAS,EAAA,OAAA,CAAQ,MAAQ,EAAA,WAAW,CAAC,CAAA,CAAA;AAE9D,EAAA,IAAI,MAAM,MAAW,KAAA,KAAA,CAAA;AAAW,IAAO,OAAA,IAAA,CAAA;AACvC,EAAA,uBACG,GAAA,CAAA,uBAAA,CAAwB,QAAxB,EAAA,EAAiC,OAAO,KACvC,EAAA,QAAA,kBAAA,GAAA;AAAA,IAAC,cAAA;AAAA,IAAA;AAAA,MACC,UAAU,IAAK,CAAA,MAAA;AAAA,MACf,KAAA;AAAA,KAAA;AAAA,GAEJ,EAAA,CAAA,CAAA;AAEJ;;;;"}

package/dist/primitives/Collapsible/index.cjs

@@ -25,15 +25,21 @@ const CollapsibleRoot = react.forwardRef(
     );
     const Component = asChild ? reactSlot.Slot : "div";
     const id = react.useId();
-    return /* @__PURE__ */ jsxRuntime.jsx(CollapsibleContext.Provider, {
-      value: { open: isOpen, onOpenChange, disabled, contentId: id },
-      children: /* @__PURE__ */ jsxRuntime.jsx(Component, {
-        ...props,
-        ref: forwardedRef,
-        "data-state": isOpen ? "open" : "closed",
-        "data-disabled": disabled ? "" : void 0
-      })
-    });
+    return /* @__PURE__ */ jsxRuntime.jsx(
+      CollapsibleContext.Provider,
+      {
+        value: { open: isOpen, onOpenChange, disabled, contentId: id },
+        children: /* @__PURE__ */ jsxRuntime.jsx(
+          Component,
+          {
+            ...props,
+            ref: forwardedRef,
+            "data-state": isOpen ? "open" : "closed",
+            "data-disabled": disabled ? "" : void 0
+          }
+        )
+      }
+    );
   }
 );
 const CollapsibleTrigger = react.forwardRef(
@@ -44,24 +50,27 @@ const CollapsibleTrigger = react.forwardRef(
       throw new Error("Collapsible.Root is missing from the React tree.");
     }
     const { open, disabled, contentId, onOpenChange } = context;
-    return /* @__PURE__ */ jsxRuntime.jsx(Component, {
-      ...props,
-      ref: forwardedRef,
-      type: "button",
-      "aria-controls": contentId,
-      "aria-expanded": open || false,
-      "data-state": open ? "open" : "closed",
-      "data-disabled": disabled ? "" : void 0,
-      disabled,
-      onClick: (event) => {
-        onClick?.(event);
-        if (event.defaultPrevented)
-          return;
-        if (disabled)
-          return;
-        onOpenChange(!open);
+    return /* @__PURE__ */ jsxRuntime.jsx(
+      Component,
+      {
+        ...props,
+        ref: forwardedRef,
+        type: "button",
+        "aria-controls": contentId,
+        "aria-expanded": open || false,
+        "data-state": open ? "open" : "closed",
+        "data-disabled": disabled ? "" : void 0,
+        disabled,
+        onClick: (event) => {
+          onClick?.(event);
+          if (event.defaultPrevented)
+            return;
+          if (disabled)
+            return;
+          onOpenChange(!open);
+        }
       }
-    });
+    );
   }
 );
 const CollapsibleContent = react.forwardRef(
@@ -105,14 +114,17 @@ const CollapsibleContent = react.forwardRef(
     react.useImperativeHandle(forwardedRef, () => {
       return divRef.current;
     }, []);
-    return /* @__PURE__ */ jsxRuntime.jsx(Component, {
-      ...props,
-      ref: divRef,
-      "data-state": open ? "open" : "closed",
-      "data-disabled": disabled ? "" : void 0,
-      id: contentId,
-      hidden: !open
-    });
+    return /* @__PURE__ */ jsxRuntime.jsx(
+      Component,
+      {
+        ...props,
+        ref: divRef,
+        "data-state": open ? "open" : "closed",
+        "data-disabled": disabled ? "" : void 0,
+        id: contentId,
+        hidden: !open
+      }
+    );
   }
 );
 if (process.env.NODE_ENV !== "production") {