laif-ds 0.1.98 → 0.2.0

package/dist/components/ui/chat.js

@@ -1,228 +1,259 @@
 "use client";
-import { jsxs as C, jsx as t, Fragment as j } from "react/jsx-runtime";
-import { forwardRef as g, useRef as L, useCallback as N, useState as z } from "react";
-import { cn as B } from "../../lib/utils.js";
-import { useAutoScroll as D } from "../../hooks/use-auto-scroll.js";
-import { Button as x } from "./button.js";
-import { CopyButton as T } from "./copy-button.js";
-import { MessageInput as P } from "./message-input.js";
-import { MessageList as U } from "./message-list.js";
-import { PromptSuggestions as E } from "./prompt-suggestions.js";
-import O from "../../node_modules/lucide-react/dist/esm/icons/thumbs-up.js";
-import q from "../../node_modules/lucide-react/dist/esm/icons/thumbs-down.js";
-import H from "../../node_modules/lucide-react/dist/esm/icons/arrow-down.js";
-function J({
-  messages: e,
-  handleSubmit: n,
+import { jsxs as N, Fragment as T, jsx as t } from "react/jsx-runtime";
+import { forwardRef as g, useRef as U, useCallback as I, useState as E } from "react";
+import { cn as M } from "../../lib/utils.js";
+import { useAutoScroll as O } from "../../hooks/use-auto-scroll.js";
+import { Button as w } from "./button.js";
+import { CopyButton as k } from "./copy-button.js";
+import { MessageInput as q } from "./message-input.js";
+import { MessageList as H } from "./message-list.js";
+import { PromptSuggestions as J } from "./prompt-suggestions.js";
+import S from "../../node_modules/lucide-react/dist/esm/icons/save.js";
+import K from "../../node_modules/lucide-react/dist/esm/icons/thumbs-up.js";
+import Q from "../../node_modules/lucide-react/dist/esm/icons/thumbs-down.js";
+import V from "../../node_modules/lucide-react/dist/esm/icons/arrow-down.js";
+function W({
+  messages: o,
+  handleSubmit: r,
   input: i,
   handleInputChange: u,
-  stop: s,
+  stop: a,
   isGenerating: c,
-  append: d,
-  suggestions: f,
-  className: w,
-  onRateResponse: p,
-  setMessages: v,
-  transcribeAudio: M
+  append: f,
+  suggestions: m,
+  className: v,
+  onRateResponse: b,
+  setMessages: C,
+  transcribeAudio: F,
+  allowAttachments: _ = !1,
+  onEdit: j,
+  onMessageSave: d
 }) {
-  const k = e.at(-1), A = e.length === 0, y = k?.role === "user", b = L(e);
-  b.current = e;
-  const F = N(() => {
-    if (s?.(), !v) return;
-    const r = [...b.current], l = r.findLast(
-      (a) => a.role === "assistant"
+  const L = o.at(-1), B = o.length === 0, y = L?.role === "user", x = U(o);
+  x.current = o;
+  const D = I(() => {
+    if (a?.(), !C) return;
+    const e = [...x.current], l = e.findLast(
+      (s) => s.role === "assistant"
     );
     if (!l) return;
-    let m = !1, h = { ...l };
+    let h = !1, p = { ...l };
     if (l.toolInvocations) {
-      const a = l.toolInvocations.map(
-        (o) => o.state === "call" ? (m = !0, {
-          ...o,
+      const s = l.toolInvocations.map(
+        (n) => n.state === "call" ? (h = !0, {
+          ...n,
           state: "result",
           result: {
             content: "Tool execution was cancelled",
             __cancelled: !0
             // Special marker to indicate cancellation
           }
-        }) : o
+        }) : n
       );
-      m && (h = {
-        ...h,
-        toolInvocations: a
+      h && (p = {
+        ...p,
+        toolInvocations: s
       });
     }
     if (l.parts && l.parts.length > 0) {
-      const a = l.parts.map((o) => o.type === "tool-invocation" && o.toolInvocation && o.toolInvocation.state === "call" ? (m = !0, {
-        ...o,
+      const s = l.parts.map((n) => n.type === "tool-invocation" && n.toolInvocation && n.toolInvocation.state === "call" ? (h = !0, {
+        ...n,
         toolInvocation: {
-          ...o.toolInvocation,
+          ...n.toolInvocation,
           state: "result",
           result: {
             content: "Tool execution was cancelled",
             __cancelled: !0
           }
         }
-      }) : o);
-      m && (h = {
-        ...h,
-        parts: a
+      }) : n);
+      h && (p = {
+        ...p,
+        parts: s
       });
     }
-    if (m) {
-      const a = r.findIndex(
-        (o) => o.id === l.id
+    if (h) {
+      const s = e.findIndex(
+        (n) => n.id === l.id
       );
-      a !== -1 && (r[a] = h, v(r));
+      s !== -1 && (e[s] = p, C(e));
     }
-  }, [s, v, b]), _ = N(
-    (r) => ({
-      actions: p ? /* @__PURE__ */ C(j, { children: [
+  }, [a, C, x]), P = I(
+    (e) => ({
+      actions: b ? /* @__PURE__ */ N(T, { children: [
         /* @__PURE__ */ t("div", { className: "border-d-border border-r pr-1", children: /* @__PURE__ */ t(
-          T,
+          k,
           {
-            content: r.content,
+            content: e.content,
             copyMessage: "Copied response to clipboard!"
           }
         ) }),
+        d && /* @__PURE__ */ t("div", { className: "border-d-border border-r pr-1", children: /* @__PURE__ */ t(
+          w,
+          {
+            size: "icon",
+            variant: "ghost",
+            className: "h-6 w-6",
+            onClick: () => d(e.id, e.content),
+            title: "Save message",
+            children: /* @__PURE__ */ t(S, { className: "h-4 w-4" })
+          }
+        ) }),
         /* @__PURE__ */ t(
-          x,
+          w,
           {
             size: "icon",
             variant: "ghost",
             className: "h-6 w-6",
-            onClick: () => p(r.id, "thumbs-up"),
-            children: /* @__PURE__ */ t(O, { className: "h-4 w-4" })
+            onClick: () => b(e.id, "thumbs-up"),
+            children: /* @__PURE__ */ t(K, { className: "h-4 w-4" })
           }
         ),
         /* @__PURE__ */ t(
-          x,
+          w,
           {
             size: "icon",
             variant: "ghost",
             className: "h-6 w-6",
-            onClick: () => p(r.id, "thumbs-down"),
-            children: /* @__PURE__ */ t(q, { className: "h-4 w-4" })
+            onClick: () => b(e.id, "thumbs-down"),
+            children: /* @__PURE__ */ t(Q, { className: "h-4 w-4" })
           }
         )
-      ] }) : /* @__PURE__ */ t(
-        T,
-        {
-          content: r.content,
-          copyMessage: "Copied response to clipboard!"
-        }
-      )
+      ] }) : /* @__PURE__ */ N(T, { children: [
+        /* @__PURE__ */ t(
+          k,
+          {
+            content: e.content,
+            copyMessage: "Copied response to clipboard!"
+          }
+        ),
+        d && /* @__PURE__ */ t("div", { className: "border-d-border border-l pl-1", children: /* @__PURE__ */ t(
+          w,
+          {
+            size: "icon",
+            variant: "ghost",
+            className: "h-6 w-6",
+            onClick: () => d(e.id, e.content),
+            title: "Save message",
+            children: /* @__PURE__ */ t(S, { className: "h-4 w-4" })
+          }
+        ) })
+      ] }),
+      onMessageSave: d
     }),
-    [p]
+    [b, d]
   );
-  return /* @__PURE__ */ C(S, { className: w, children: [
-    A && d && f ? /* @__PURE__ */ t(
-      E,
+  return /* @__PURE__ */ N(z, { className: v, children: [
+    B && f && m ? /* @__PURE__ */ t(
+      J,
       {
         label: "Try these prompts ✨",
-        append: d,
-        suggestions: f
+        append: f,
+        suggestions: m
       }
     ) : null,
-    e.length > 0 ? /* @__PURE__ */ t(K, { messages: e, children: /* @__PURE__ */ t(
-      U,
+    o.length > 0 ? /* @__PURE__ */ t(X, { messages: o, children: /* @__PURE__ */ t(
+      H,
       {
-        messages: e,
+        messages: o,
         isTyping: y,
-        messageOptions: _
+        messageOptions: P,
+        onEdit: j,
+        onMessageSave: d
       }
     ) }) : null,
     /* @__PURE__ */ t(
-      I,
+      A,
       {
         className: "flex-shrink-0",
         isPending: c || y,
-        handleSubmit: n,
-        children: ({ files: r, setFiles: l }) => /* @__PURE__ */ t(
-          P,
+        handleSubmit: r,
+        children: ({ files: e, setFiles: l }) => /* @__PURE__ */ t(
+          q,
           {
             value: i,
             onChange: u,
-            allowAttachments: !0,
-            files: r,
+            allowAttachments: _,
+            files: e,
             setFiles: l,
-            stop: F,
+            stop: D,
             isGenerating: c,
-            transcribeAudio: M
+            transcribeAudio: F
           }
         )
       }
     )
   ] });
 }
-J.displayName = "Chat";
-function K({
-  messages: e,
-  children: n
+W.displayName = "Chat";
+function X({
+  messages: o,
+  children: r
 }) {
   const {
     containerRef: i,
     scrollToBottom: u,
-    handleScroll: s,
+    handleScroll: a,
     shouldAutoScroll: c,
-    handleTouchStart: d
-  } = D([e]);
-  return /* @__PURE__ */ C(
+    handleTouchStart: f
+  } = O([o]);
+  return /* @__PURE__ */ N(
     "div",
     {
       className: "relative flex-1 overflow-y-auto pb-4",
       ref: i,
-      onScroll: s,
-      onTouchStart: d,
+      onScroll: a,
+      onTouchStart: f,
       children: [
-        /* @__PURE__ */ t("div", { className: "max-w-full", children: n }),
+        /* @__PURE__ */ t("div", { className: "max-w-full", children: r }),
         !c && /* @__PURE__ */ t("div", { className: "pointer-events-none absolute right-0 bottom-0 left-0 flex items-end justify-end", children: /* @__PURE__ */ t("div", { className: "sticky bottom-0 left-0 flex w-full justify-end", children: /* @__PURE__ */ t(
-          x,
+          w,
           {
             onClick: u,
             className: "animate-in fade-in-0 slide-in-from-bottom-1 pointer-events-auto h-8 w-8 rounded-full ease-in-out",
             size: "icon",
             variant: "ghost",
-            children: /* @__PURE__ */ t(H, { className: "h-4 w-4" })
+            children: /* @__PURE__ */ t(V, { className: "h-4 w-4" })
           }
         ) }) })
       ]
     }
   );
 }
-const S = g(({ className: e, ...n }, i) => /* @__PURE__ */ t(
+const z = g(({ className: o, ...r }, i) => /* @__PURE__ */ t(
   "div",
   {
     ref: i,
-    className: B("flex h-full max-h-full w-full flex-col", e),
-    ...n
+    className: M("flex h-full max-h-full w-full flex-col", o),
+    ...r
   }
 ));
-S.displayName = "ChatContainer";
-const I = g(
-  ({ children: e, handleSubmit: n, className: i }, u) => {
-    const [s, c] = z(null);
-    return /* @__PURE__ */ t("form", { ref: u, onSubmit: (f) => {
-      if (!s) {
-        n(f);
+z.displayName = "ChatContainer";
+const A = g(
+  ({ children: o, handleSubmit: r, className: i }, u) => {
+    const [a, c] = E(null);
+    return /* @__PURE__ */ t("form", { ref: u, onSubmit: (m) => {
+      if (!a) {
+        r(m);
         return;
       }
-      const w = Q(s);
-      n(f, { experimental_attachments: w }), c(null);
-    }, className: i, children: e({ files: s, setFiles: c }) });
+      const v = Y(a);
+      r(m, { experimental_attachments: v }), c(null);
+    }, className: i, children: o({ files: a, setFiles: c }) });
   }
 );
-I.displayName = "ChatForm";
-function Q(e) {
+A.displayName = "ChatForm";
+function Y(o) {
   if (typeof window > "u")
     return {};
-  const n = new DataTransfer();
-  for (const i of Array.from(e))
-    n.items.add(i);
-  return n.files;
+  const r = new DataTransfer();
+  for (const i of Array.from(o))
+    r.items.add(i);
+  return r.files;
 }
 export {
-  J as Chat,
-  S as ChatContainer,
-  I as ChatForm,
-  K as ChatMessages
+  W as Chat,
+  z as ChatContainer,
+  A as ChatForm,
+  X as ChatMessages
 };

package/dist/components/ui/dialog.js

@@ -1,34 +1,34 @@
 "use client";
 import { jsx as a, jsxs as n } from "react/jsx-runtime";
-import { Root as l, Trigger as d, Content as c, Close as s, Title as g, Description as f, Portal as u, Overlay as m } from "../../node_modules/@radix-ui/react-dialog/dist/index.js";
+import { Root as c, Trigger as g, Content as f, Close as l, Title as u, Description as m, Portal as p, Overlay as x } from "../../node_modules/@radix-ui/react-dialog/dist/index.js";
 import { cn as o } from "../../lib/utils.js";
-import p from "../../node_modules/lucide-react/dist/esm/icons/x.js";
-function w({
+import D from "../../node_modules/lucide-react/dist/esm/icons/x.js";
+function C({
   ...t
 }) {
-  return /* @__PURE__ */ a(l, { "data-slot": "dialog", ...t });
+  return /* @__PURE__ */ a(c, { "data-slot": "dialog", ...t });
 }
-function z({
+function k({
   ...t
 }) {
-  return /* @__PURE__ */ a(d, { "data-slot": "dialog-trigger", ...t });
+  return /* @__PURE__ */ a(g, { "data-slot": "dialog-trigger", ...t });
 }
-function x({
+function v({
   ...t
 }) {
-  return /* @__PURE__ */ a(u, { "data-slot": "dialog-portal", ...t });
+  return /* @__PURE__ */ a(p, { "data-slot": "dialog-portal", ...t });
 }
-function C({
+function T({
   ...t
 }) {
-  return /* @__PURE__ */ a(s, { "data-slot": "dialog-close", ...t });
+  return /* @__PURE__ */ a(l, { "data-slot": "dialog-close", ...t });
 }
-function D({
+function b({
   className: t,
   ...e
 }) {
   return /* @__PURE__ */ a(
-    m,
+    x,
     {
       "data-slot": "dialog-overlay",
       className: o(
@@ -39,29 +39,37 @@ function D({
     }
   );
 }
-function k({
+function j({
   className: t,
   children: e,
+  size: r = "default",
   ...i
 }) {
-  return /* @__PURE__ */ n(x, { "data-slot": "dialog-portal", children: [
-    /* @__PURE__ */ a(D, {}),
+  const d = {
+    sm: "sm:max-w-sm",
+    default: "sm:max-w-lg",
+    lg: "sm:max-w-2xl",
+    xl: "sm:max-w-3xl"
+  }[r];
+  return /* @__PURE__ */ n(v, { "data-slot": "dialog-portal", children: [
+    /* @__PURE__ */ a(b, {}),
     /* @__PURE__ */ n(
-      c,
+      f,
       {
         "data-slot": "dialog-content",
-        onPointerDownOutside: (r) => {
-          r.target?.closest("[data-command-portal]") && r.preventDefault();
+        onPointerDownOutside: (s) => {
+          s.target?.closest("[data-command-portal]") && s.preventDefault();
         },
         className: o(
-          "bg-d-background data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 border-d-border fixed top-[50%] left-[50%] z-50 grid w-full max-w-[calc(100%-2rem)] translate-x-[-50%] translate-y-[-50%] gap-4 rounded-lg border p-6 shadow-lg duration-200 sm:max-w-lg",
+          "bg-d-background data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 border-d-border fixed top-[50%] left-[50%] z-50 grid w-full max-w-[calc(100%-2rem)] translate-x-[-50%] translate-y-[-50%] gap-4 rounded-lg border p-6 shadow-lg duration-200",
+          d,
           t
         ),
         ...i,
         children: [
           e,
-          /* @__PURE__ */ n(s, { className: "ring-offset-d-background focus:ring-d-ring data-[state=open]:bg-d-accent data-[state=open]:text-d-secondary-foreground absolute top-4 right-4 rounded-xs opacity-70 transition-opacity hover:opacity-100 focus:ring-2 focus:ring-offset-2 focus:outline-hidden disabled:pointer-events-none [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4", children: [
-            /* @__PURE__ */ a(p, {}),
+          /* @__PURE__ */ n(l, { className: "ring-offset-d-background focus:ring-d-ring data-[state=open]:bg-d-accent data-[state=open]:text-d-secondary-foreground absolute top-4 right-4 rounded-xs opacity-70 transition-opacity hover:opacity-100 focus:ring-2 focus:ring-offset-2 focus:outline-hidden disabled:pointer-events-none [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4", children: [
+            /* @__PURE__ */ a(D, {}),
             /* @__PURE__ */ a("span", { className: "sr-only", children: "Close" })
           ] })
         ]
@@ -69,7 +77,7 @@ function k({
     )
   ] });
 }
-function T({ className: t, ...e }) {
+function O({ className: t, ...e }) {
   return /* @__PURE__ */ a(
     "div",
     {
@@ -79,7 +87,7 @@ function T({ className: t, ...e }) {
     }
   );
 }
-function j({ className: t, ...e }) {
+function P({ className: t, ...e }) {
   return /* @__PURE__ */ a(
     "div",
     {
@@ -92,12 +100,12 @@ function j({ className: t, ...e }) {
     }
   );
 }
-function O({
+function _({
   className: t,
   ...e
 }) {
   return /* @__PURE__ */ a(
-    g,
+    u,
     {
       "data-slot": "dialog-title",
       className: o("text-lg leading-none font-semibold", t),
@@ -105,12 +113,12 @@ function O({
     }
   );
 }
-function P({
+function F({
   className: t,
   ...e
 }) {
   return /* @__PURE__ */ a(
-    f,
+    m,
     {
       "data-slot": "dialog-description",
       className: o("text-d-secondary-foreground text-sm", t),
@@ -119,14 +127,14 @@ function P({
   );
 }
 export {
-  w as Dialog,
-  C as DialogClose,
-  k as DialogContent,
-  P as DialogDescription,
-  j as DialogFooter,
-  T as DialogHeader,
-  D as DialogOverlay,
-  x as DialogPortal,
-  O as DialogTitle,
-  z as DialogTrigger
+  C as Dialog,
+  T as DialogClose,
+  j as DialogContent,
+  F as DialogDescription,
+  P as DialogFooter,
+  O as DialogHeader,
+  b as DialogOverlay,
+  v as DialogPortal,
+  _ as DialogTitle,
+  k as DialogTrigger
 };

package/dist/components/ui/message-input.js

@@ -127,7 +127,7 @@ function ie({
               onPaste: K,
               onKeyDown: U,
               className: C(
-                "border-d-input bg-d-background ring-offset-d-background placeholder:text-d-secondary-foreground focus-visible:border-d-primary z-10 w-full grow resize-none rounded-xl border p-3 pr-24 text-sm transition-[border] focus-visible:outline-none disabled:cursor-not-allowed disabled:opacity-50",
+                "border-d-input bg-d-input ring-offset-d-background placeholder:text-d-muted-foreground focus-visible:border-d-primary z-10 w-full grow resize-none rounded-xl border p-3 pr-24 text-sm transition-[border] focus-visible:outline-none disabled:cursor-not-allowed disabled:opacity-50",
                 z && "pb-16",
                 n
               ),

package/dist/components/ui/message-list.js

@@ -1,29 +1,33 @@
 "use client";
-import { jsxs as c, jsx as o } from "react/jsx-runtime";
-import { ChatMessage as f } from "./chat-message.js";
-import { TypingIndicator as p } from "./typing-indicator.js";
-function d({
-  messages: r,
+import { jsxs as l, jsx as r } from "react/jsx-runtime";
+import { ChatMessage as s } from "./chat-message.js";
+import { TypingIndicator as m } from "./typing-indicator.js";
+function x({
+  messages: n,
   showTimeStamps: e = !0,
-  isTyping: n = !1,
-  messageOptions: t
+  isTyping: a = !1,
+  messageOptions: i,
+  onEdit: t,
+  onMessageSave: c
 }) {
-  return /* @__PURE__ */ c("div", { className: "space-y-4 overflow-visible", children: [
-    r.map((i, a) => {
-      const s = typeof t == "function" ? t(i) : t;
-      return /* @__PURE__ */ o(
-        f,
+  return /* @__PURE__ */ l("div", { className: "space-y-4 overflow-visible", children: [
+    n.map((o, f) => {
+      const p = typeof i == "function" ? i(o) : i;
+      return /* @__PURE__ */ r(
+        s,
         {
           showTimeStamp: e,
-          ...i,
-          ...s
+          ...o,
+          ...p,
+          onEdit: t ? (d) => t(o.id, d) : void 0,
+          onMessageSave: c
         },
-        a
+        f
       );
     }),
-    n && /* @__PURE__ */ o(p, {})
+    a && /* @__PURE__ */ r(m, {})
   ] });
 }
 export {
-  d as MessageList
+  x as MessageList
 };

package/dist/index.d.ts

@@ -436,7 +436,7 @@ export declare function ChartTooltipContent({ active, payload, className, indica
     labelKey?: string;
 } & React_2.HTMLAttributes<HTMLDivElement>): JSX.Element | null;
 
-export declare function Chat({ messages, handleSubmit, input, handleInputChange, stop, isGenerating, append, suggestions, className, onRateResponse, setMessages, transcribeAudio, }: ChatProps): JSX.Element;
+export declare function Chat({ messages, handleSubmit, input, handleInputChange, stop, isGenerating, append, suggestions, className, onRateResponse, setMessages, transcribeAudio, allowAttachments, onEdit, onMessageSave, }: ChatProps): JSX.Element;
 
 export declare namespace Chat {
     var displayName: string;
@@ -471,6 +471,8 @@ export declare interface ChatMessageProps extends Message {
     showTimeStamp?: boolean;
     animation?: Animation_2;
     actions?: default_3.ReactNode;
+    onEdit?: (newContent: string) => void;
+    onMessageSave?: (messageId: string, content: string) => void;
 }
 
 export declare function ChatMessages({ messages, children, }: React.PropsWithChildren<{
@@ -494,6 +496,9 @@ declare interface ChatPropsBase {
     onRateResponse?: (messageId: string, rating: "thumbs-up" | "thumbs-down") => void;
     setMessages?: (messages: any[]) => void;
     transcribeAudio?: (blob: Blob) => Promise<string>;
+    allowAttachments?: boolean;
+    onEdit?: (messageId: string, newContent: string) => void;
+    onMessageSave?: (messageId: string, content: string) => void;
 }
 
 declare interface ChatPropsWithoutSuggestions extends ChatPropsBase {
@@ -791,7 +796,9 @@ declare interface DatePickerProps {
 
 export declare function Dialog({ ...props }: React_2.ComponentProps<typeof DialogPrimitive.Root>): JSX.Element;
 
-export declare function DialogContent({ className, children, ...props }: React_2.ComponentProps<typeof DialogPrimitive.Content>): JSX.Element;
+export declare function DialogContent({ className, children, size, ...props }: React_2.ComponentProps<typeof DialogPrimitive.Content> & {
+    size?: "sm" | "default" | "lg";
+}): JSX.Element;
 
 export declare function DialogDescription({ className, ...props }: React_2.ComponentProps<typeof DialogPrimitive.Description>): JSX.Element;
 
@@ -1171,13 +1178,15 @@ declare interface MessageInputWithoutAttachmentProps extends MessageInputBasePro
     allowAttachments?: false;
 }
 
-export declare function MessageList({ messages, showTimeStamps, isTyping, messageOptions, }: MessageListProps): JSX.Element;
+export declare function MessageList({ messages, showTimeStamps, isTyping, messageOptions, onEdit, onMessageSave, }: MessageListProps): JSX.Element;
 
 declare interface MessageListProps {
     messages: Message[];
     showTimeStamps?: boolean;
     isTyping?: boolean;
     messageOptions?: AdditionalMessageOptions | ((message: Message) => AdditionalMessageOptions);
+    onEdit?: (messageId: string, newContent: string) => void;
+    onMessageSave?: (messageId: string, content: string) => void;
 }
 
 declare type MessagePart = TextPart | ReasoningPart | ToolInvocationPart | SourcePart;