@uipath/apollo-vertex 0.0.0-stage → 0.3.0

package/dist/registry/ai-chat/components/ai-chat-input.d.ts

@@ -0,0 +1,22 @@
+import type { ContentPart } from "@tanstack/ai";
+import { type Ref } from "react";
+interface AiChatInputProps {
+    value: string;
+    onChange: (value: string) => void;
+    onSubmit: (parts?: ContentPart[]) => void;
+    onStop: () => void;
+    isLoading: boolean;
+    disabled?: boolean;
+    placeholder?: string;
+    hasMessages?: boolean;
+    acceptedFileTypes?: string;
+    quotedText?: string | null;
+    onClearQuote?: () => void;
+    ref?: Ref<AiChatInputHandle>;
+}
+export interface AiChatInputHandle {
+    focus: () => void;
+    addFiles: (files: FileList | File[]) => void;
+}
+export declare function AiChatInput({ value, onChange, onSubmit, onStop, isLoading, disabled, placeholder, hasMessages, acceptedFileTypes, quotedText, onClearQuote, ref, }: AiChatInputProps): import("react/jsx-runtime").JSX.Element;
+export {};

package/dist/registry/ai-chat/components/ai-chat-input.js

@@ -0,0 +1,271 @@
+"use client";
+import { Fragment, jsx, jsxs } from "react/jsx-runtime";
+import { ArrowUp, CircleStop, Paperclip, X } from "lucide-react";
+import { useImperativeHandle, useRef, useState } from "react";
+import { useVertexTranslation } from "../../../lib/vertex-translation.js";
+import { Button } from "../../button/button.js";
+import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuTrigger } from "../../dropdown-menu/dropdown-menu.js";
+import { Tooltip, TooltipContent, TooltipTrigger } from "../../tooltip/tooltip.js";
+import { cn } from "../../../lib/utils.js";
+import { pendingFilesToContentParts, usePendingFiles } from "../hooks/use-pending-files.js";
+import { AiChatImagePreview } from "./ai-chat-image-preview.js";
+import { AiChatInputGlow } from "./ai-chat-input-glow.js";
+import { AiChatPendingFiles } from "./ai-chat-pending-files.js";
+function AiChatInput({ value, onChange, onSubmit, onStop, isLoading, disabled = false, placeholder, hasMessages = false, acceptedFileTypes, quotedText, onClearQuote, ref }) {
+    const { t } = useVertexTranslation();
+    const textareaRef = useRef(null);
+    const fileInputRef = useRef(null);
+    const [focused, setFocused] = useState(false);
+    const displayPlaceholder = placeholder ?? t("shell_input_placeholder");
+    const attachmentsEnabled = !!acceptedFileTypes;
+    const { files: pendingFiles, add: addFiles, remove: removeFile, clear: clearFiles } = usePendingFiles();
+    const [previewUrl, setPreviewUrl] = useState(null);
+    useImperativeHandle(ref, ()=>({
+            focus: ()=>textareaRef.current?.focus(),
+            addFiles
+        }));
+    const adjustHeight = ()=>{
+        if (!hasMessages) return;
+        const el = textareaRef.current;
+        if (!el) return;
+        el.style.height = "auto";
+        el.style.height = `${Math.min(el.scrollHeight, 200)}px`;
+    };
+    const handleChange = (val)=>{
+        onChange(val);
+        requestAnimationFrame(adjustHeight);
+    };
+    const handleFileSelect = (e)=>{
+        if (e.target.files?.length) addFiles(e.target.files);
+        e.target.value = "";
+    };
+    const handlePaste = (e)=>{
+        if (!attachmentsEnabled) return;
+        const files = Array.from(e.clipboardData.items).filter((item)=>"file" === item.kind).map((item)=>item.getAsFile()).filter((f)=>null !== f);
+        if (0 === files.length) return;
+        e.preventDefault();
+        addFiles(files);
+    };
+    const submitMessage = async ()=>{
+        if (!value.trim() && 0 === pendingFiles.length) return;
+        const filesSnapshot = pendingFiles;
+        setPreviewUrl(null);
+        clearFiles();
+        requestAnimationFrame(()=>{
+            const el = textareaRef.current;
+            if (el) el.style.height = "auto";
+        });
+        const parts = await pendingFilesToContentParts(filesSnapshot);
+        onSubmit(parts);
+    };
+    const handleSubmit = (e)=>{
+        e.preventDefault();
+        submitMessage();
+    };
+    const handleKeyDown = (e)=>{
+        if ("Enter" === e.key && !e.shiftKey) {
+            e.preventDefault();
+            submitMessage();
+        }
+    };
+    const focusedStyle = {
+        borderColor: "transparent",
+        backgroundImage: "linear-gradient(var(--background), var(--background)), var(--ai-gradient-strong)",
+        backgroundOrigin: "border-box",
+        backgroundClip: "padding-box, border-box",
+        boxShadow: "0 0 0 3px color-mix(in oklch, var(--muted-foreground) 10%, transparent)"
+    };
+    const formProps = {
+        onSubmit: handleSubmit,
+        onFocus: ()=>setFocused(true),
+        onBlur: (e)=>{
+            if (!e.currentTarget.contains(e.relatedTarget)) setFocused(false);
+        },
+        className: "relative flex flex-col rounded-lg border-2 border-input transition-colors bg-background",
+        ...focused && {
+            style: focusedStyle
+        }
+    };
+    const pendingFilesChips = attachmentsEnabled ? /*#__PURE__*/ jsx(AiChatPendingFiles, {
+        files: pendingFiles,
+        onRemove: removeFile,
+        onPreview: setPreviewUrl
+    }) : null;
+    const quoteChip = quotedText ? /*#__PURE__*/ jsx("div", {
+        className: "flex items-center px-3 pt-2",
+        children: /*#__PURE__*/ jsxs("div", {
+            className: "inline-flex items-center gap-1.5 rounded-lg bg-ai-chat-muted px-2 py-1.5 text-xs text-ai-chat-muted-foreground max-w-full min-w-0",
+            children: [
+                /*#__PURE__*/ jsx("span", {
+                    className: "truncate",
+                    children: quotedText
+                }),
+                /*#__PURE__*/ jsx(Button, {
+                    type: "button",
+                    variant: "ghost",
+                    size: "icon-xs",
+                    onClick: onClearQuote,
+                    className: "shrink-0 hover:bg-ai-chat-border",
+                    "aria-label": t("remove_quoted_text"),
+                    children: /*#__PURE__*/ jsx(X, {
+                        "aria-hidden": "true"
+                    })
+                })
+            ]
+        })
+    }) : null;
+    const plusMenu = attachmentsEnabled ? /*#__PURE__*/ jsxs(DropdownMenu, {
+        children: [
+            /*#__PURE__*/ jsx(DropdownMenuTrigger, {
+                asChild: true,
+                children: /*#__PURE__*/ jsx(Button, {
+                    type: "button",
+                    variant: "ghost",
+                    size: "icon",
+                    className: "flex-shrink-0 hover:bg-ai-chat-muted text-ai-chat-muted-foreground",
+                    "aria-label": t("add_attachment"),
+                    children: /*#__PURE__*/ jsx(Paperclip, {
+                        "aria-hidden": "true"
+                    })
+                })
+            }),
+            /*#__PURE__*/ jsx(DropdownMenuContent, {
+                align: "start",
+                side: "top",
+                children: /*#__PURE__*/ jsx(DropdownMenuItem, {
+                    onClick: ()=>fileInputRef.current?.click(),
+                    children: t("upload_files")
+                })
+            })
+        ]
+    }) : null;
+    const sendStopButton = isLoading ? /*#__PURE__*/ jsxs(Tooltip, {
+        children: [
+            /*#__PURE__*/ jsx(TooltipTrigger, {
+                asChild: true,
+                children: /*#__PURE__*/ jsx(Button, {
+                    type: "button",
+                    variant: "secondary",
+                    size: "icon",
+                    className: "flex-shrink-0",
+                    onClick: onStop,
+                    "aria-label": t("stop"),
+                    children: /*#__PURE__*/ jsx(CircleStop, {
+                        className: "size-5 text-foreground",
+                        "aria-hidden": "true"
+                    })
+                })
+            }),
+            /*#__PURE__*/ jsx(TooltipContent, {
+                children: t("stop")
+            })
+        ]
+    }) : /*#__PURE__*/ jsxs(Tooltip, {
+        children: [
+            /*#__PURE__*/ jsx(TooltipTrigger, {
+                asChild: true,
+                children: /*#__PURE__*/ jsx(Button, {
+                    type: "submit",
+                    variant: "ghost",
+                    size: "icon",
+                    disabled: !value.trim() && 0 === pendingFiles.length || disabled,
+                    className: "flex-shrink-0 text-white hover:text-white hover:opacity-90 disabled:opacity-40 disabled:cursor-not-allowed",
+                    style: {
+                        background: "var(--ai-gradient-strong)"
+                    },
+                    "aria-label": t("send"),
+                    children: /*#__PURE__*/ jsx(ArrowUp, {
+                        className: "size-5",
+                        "aria-hidden": "true"
+                    })
+                })
+            }),
+            /*#__PURE__*/ jsx(TooltipContent, {
+                children: t("send")
+            })
+        ]
+    });
+    return /*#__PURE__*/ jsxs("div", {
+        className: "relative z-10 mt-auto pt-3 px-4",
+        children: [
+            attachmentsEnabled && /*#__PURE__*/ jsxs(Fragment, {
+                children: [
+                    /*#__PURE__*/ jsx("input", {
+                        ref: fileInputRef,
+                        type: "file",
+                        multiple: true,
+                        accept: acceptedFileTypes,
+                        hidden: true,
+                        onChange: handleFileSelect
+                    }),
+                    /*#__PURE__*/ jsx(AiChatImagePreview, {
+                        url: previewUrl,
+                        onClose: ()=>setPreviewUrl(null)
+                    })
+                ]
+            }),
+            /*#__PURE__*/ jsxs("div", {
+                className: "relative",
+                children: [
+                    /*#__PURE__*/ jsx("div", {
+                        className: `absolute pointer-events-none transition-opacity duration-500 -top-14 -left-12 -right-12 -bottom-16 ${hasMessages || focused ? "opacity-0" : "opacity-100"}`,
+                        "aria-hidden": "true",
+                        children: /*#__PURE__*/ jsx(AiChatInputGlow, {})
+                    }),
+                    hasMessages ? /*#__PURE__*/ jsxs("form", {
+                        ...formProps,
+                        children: [
+                            quoteChip,
+                            pendingFilesChips,
+                            /*#__PURE__*/ jsxs("div", {
+                                className: "flex items-end gap-2 pl-[8px] pr-[8px] pt-[4px] pb-[8px]",
+                                children: [
+                                    plusMenu,
+                                    /*#__PURE__*/ jsx("textarea", {
+                                        ref: textareaRef,
+                                        value: value,
+                                        onChange: (e)=>handleChange(e.target.value),
+                                        onKeyDown: handleKeyDown,
+                                        onPaste: handlePaste,
+                                        placeholder: displayPlaceholder,
+                                        "aria-label": displayPlaceholder,
+                                        className: "flex-1 resize-none bg-transparent text-sm text-foreground placeholder:text-muted-foreground focus-visible:outline-none max-h-[200px] overflow-y-auto h-[40px] pt-[12px] pb-[8px] leading-[20px]",
+                                        rows: 1,
+                                        disabled: disabled
+                                    }),
+                                    sendStopButton
+                                ]
+                            })
+                        ]
+                    }) : /*#__PURE__*/ jsxs("form", {
+                        ...formProps,
+                        children: [
+                            quoteChip,
+                            pendingFilesChips,
+                            /*#__PURE__*/ jsx("textarea", {
+                                ref: textareaRef,
+                                value: value,
+                                onChange: (e)=>handleChange(e.target.value),
+                                onKeyDown: handleKeyDown,
+                                onPaste: handlePaste,
+                                placeholder: displayPlaceholder,
+                                "aria-label": displayPlaceholder,
+                                className: "w-full resize-none bg-transparent text-sm text-foreground placeholder:text-muted-foreground focus-visible:outline-none h-[80px] px-[22px] pt-5 pb-2 leading-relaxed",
+                                rows: 1,
+                                disabled: disabled
+                            }),
+                            /*#__PURE__*/ jsxs("div", {
+                                className: cn("flex items-center px-[8px] pb-[8px]", plusMenu ? "justify-between" : "justify-end"),
+                                children: [
+                                    plusMenu,
+                                    sendStopButton
+                                ]
+                            })
+                        ]
+                    })
+                ]
+            })
+        ]
+    });
+}
+export { AiChatInput };

package/dist/registry/ai-chat/components/ai-chat-loading.d.ts

@@ -0,0 +1 @@
+export declare function AiChatLoading(): import("react/jsx-runtime").JSX.Element;

package/dist/registry/ai-chat/components/ai-chat-loading.js

@@ -0,0 +1,79 @@
+"use client";
+import { jsx, jsxs } from "react/jsx-runtime";
+import { motion, useReducedMotion } from "framer-motion";
+import { useVertexTranslation } from "../../../lib/vertex-translation.js";
+import { ENTRANCE_ANIMATE, ENTRANCE_EASE, ENTRANCE_INITIAL } from "../animations.js";
+import { AiChatThinking } from "./ai-chat-thinking.js";
+const ENTRANCE_DURATION = 0.5;
+const TEXT_DELAY = 0.9;
+const TEXT_DURATION = 0.3;
+const TEXT_OVERLAP_PX = -7;
+function AiChatLoading() {
+    const { t } = useVertexTranslation();
+    const prefersReducedMotion = useReducedMotion();
+    const thinkingLabel = t("thinking");
+    const shimmerStyle = prefersReducedMotion ? {} : {
+        display: "inline-block",
+        backgroundImage: "linear-gradient(90deg, var(--muted-foreground) 0%, var(--muted-foreground) 30%, var(--ai-gradient-start, #6C5AEF) 42%, var(--foreground) 50%, var(--muted-foreground) 58%, var(--muted-foreground) 70%, var(--muted-foreground) 100%)",
+        backgroundSize: "200% 100%",
+        backgroundClip: "text",
+        WebkitBackgroundClip: "text",
+        color: "transparent",
+        animation: "ap-chat-loading-shimmer 2.4s linear infinite"
+    };
+    return /*#__PURE__*/ jsxs(motion.div, {
+        role: "status",
+        "aria-label": thinkingLabel,
+        className: "flex justify-start py-2",
+        initial: ENTRANCE_INITIAL,
+        animate: ENTRANCE_ANIMATE,
+        transition: {
+            duration: ENTRANCE_DURATION,
+            ease: ENTRANCE_EASE
+        },
+        children: [
+            !prefersReducedMotion && /*#__PURE__*/ jsx("style", {
+                children: `
+          @keyframes ap-chat-loading-shimmer {
+            0% { background-position: 200% 0; }
+            100% { background-position: -200% 0; }
+          }
+        `
+            }),
+            /*#__PURE__*/ jsxs("div", {
+                className: "flex items-center gap-0",
+                children: [
+                    /*#__PURE__*/ jsx(AiChatThinking, {
+                        size: 40,
+                        isThinking: true
+                    }),
+                    /*#__PURE__*/ jsx(motion.div, {
+                        initial: {
+                            opacity: 0,
+                            x: -8
+                        },
+                        animate: {
+                            opacity: 1,
+                            x: 0
+                        },
+                        transition: {
+                            duration: TEXT_DURATION,
+                            delay: TEXT_DELAY,
+                            ease: ENTRANCE_EASE
+                        },
+                        className: "flex items-center",
+                        style: {
+                            marginLeft: TEXT_OVERLAP_PX
+                        },
+                        children: /*#__PURE__*/ jsx("span", {
+                            className: "text-sm font-medium leading-snug whitespace-nowrap",
+                            style: shimmerStyle,
+                            children: `${thinkingLabel}\u2026`
+                        })
+                    })
+                ]
+            })
+        ]
+    });
+}
+export { AiChatLoading };

package/dist/registry/ai-chat/components/ai-chat-markdown.d.ts

@@ -0,0 +1,7 @@
+interface AiChatMarkdownProps {
+    children: string;
+    className?: string;
+    preserveLineBreaks?: boolean;
+}
+export declare function AiChatMarkdown({ children, className, preserveLineBreaks, }: AiChatMarkdownProps): import("react/jsx-runtime").JSX.Element;
+export {};

package/dist/registry/ai-chat/components/ai-chat-markdown.js

@@ -0,0 +1,139 @@
+"use client";
+import { jsx } from "react/jsx-runtime";
+import react_markdown from "react-markdown";
+import remark_breaks from "remark-breaks";
+import remark_gfm from "remark-gfm";
+import { AiChatCodeBlock } from "./ai-chat-code-block.js";
+function extractCodeProps(props) {
+    const { className, children } = props;
+    const match = /language-(\w+)/.exec(className ?? "");
+    const language = match ? match[1] : "";
+    const code = ("string" == typeof children ? children : "").replace(/\n$/, "");
+    return {
+        language,
+        code
+    };
+}
+const components = {
+    p: ({ children })=>/*#__PURE__*/ jsx("p", {
+            className: "mt-5 mb-5 first:mt-0 last:mb-0",
+            children: children
+        }),
+    ul: ({ children })=>/*#__PURE__*/ jsx("ul", {
+            className: "mt-5 mb-5 first:mt-0 last:mb-0 ml-4 list-disc",
+            children: children
+        }),
+    ol: ({ children })=>/*#__PURE__*/ jsx("ol", {
+            className: "mt-5 mb-5 first:mt-0 last:mb-0 ml-4 list-decimal",
+            children: children
+        }),
+    li: ({ children })=>/*#__PURE__*/ jsx("li", {
+            className: "mb-3",
+            children: children
+        }),
+    pre: ({ children })=>/*#__PURE__*/ jsx("div", {
+            children: children
+        }),
+    code: ({ children, className, ...props })=>{
+        const isBlock = (className?.startsWith("language-") ?? false) || "string" == typeof children && children.includes("\n");
+        if (isBlock) {
+            const { language, code } = extractCodeProps({
+                className,
+                children,
+                ...props
+            });
+            return /*#__PURE__*/ jsx(AiChatCodeBlock, {
+                language: language,
+                children: code
+            });
+        }
+        return /*#__PURE__*/ jsx("code", {
+            className: "px-1.5 py-0.5 rounded bg-ai-chat-muted text-ai-chat-foreground font-mono text-xs",
+            children: children
+        });
+    },
+    a: ({ children, ...props })=>/*#__PURE__*/ jsx("a", {
+            className: "text-primary-700 dark:text-primary-400 hover:underline",
+            ...props,
+            children: children
+        }),
+    img: ({ src, alt, title })=>/*#__PURE__*/ jsx("img", {
+            src: "string" == typeof src ? src : "",
+            alt: alt ?? "",
+            title: title,
+            loading: "lazy",
+            decoding: "async",
+            className: "block max-w-full max-h-[400px] object-contain rounded-lg border border-ai-chat-border"
+        }),
+    strong: ({ children })=>/*#__PURE__*/ jsx("strong", {
+            className: "font-semibold",
+            children: children
+        }),
+    em: ({ children })=>/*#__PURE__*/ jsx("em", {
+            className: "italic",
+            children: children
+        }),
+    blockquote: ({ children })=>/*#__PURE__*/ jsx("blockquote", {
+            className: "border-l-2 border-ai-chat-accent pl-3 bg-ai-chat-accent/5 rounded-r italic my-2 py-1",
+            children: children
+        }),
+    h1: ({ children })=>/*#__PURE__*/ jsx("h1", {
+            className: "text-2xl font-bold mb-3 mt-6 first:mt-0",
+            children: children
+        }),
+    h2: ({ children })=>/*#__PURE__*/ jsx("h2", {
+            className: "text-xl font-bold mb-2 mt-5 first:mt-0",
+            children: children
+        }),
+    h3: ({ children })=>/*#__PURE__*/ jsx("h3", {
+            className: "text-lg font-bold mb-2 mt-4 first:mt-0",
+            children: children
+        }),
+    hr: ()=>/*#__PURE__*/ jsx("hr", {
+            className: "my-3 border-ai-chat-border"
+        }),
+    table: ({ children })=>/*#__PURE__*/ jsx("div", {
+            className: "overflow-x-auto my-2",
+            children: /*#__PURE__*/ jsx("table", {
+                className: "min-w-full divide-y divide-ai-chat-border",
+                children: children
+            })
+        }),
+    thead: ({ children })=>/*#__PURE__*/ jsx("thead", {
+            className: "bg-ai-chat-muted",
+            children: children
+        }),
+    tbody: ({ children })=>/*#__PURE__*/ jsx("tbody", {
+            className: "divide-y divide-ai-chat-border",
+            children: children
+        }),
+    tr: ({ children })=>/*#__PURE__*/ jsx("tr", {
+            children: children
+        }),
+    th: ({ children })=>/*#__PURE__*/ jsx("th", {
+            className: "px-3 py-2 text-left text-xs font-semibold",
+            children: children
+        }),
+    td: ({ children })=>/*#__PURE__*/ jsx("td", {
+            className: "px-3 py-2 text-xs",
+            children: children
+        })
+};
+const DEFAULT_WRAPPER = "py-1 text-base leading-relaxed bg-transparent text-foreground prose dark:prose-invert max-w-none";
+function AiChatMarkdown({ children, className, preserveLineBreaks = false }) {
+    const plugins = preserveLineBreaks ? [
+        remark_gfm,
+        remark_breaks
+    ] : [
+        remark_gfm
+    ];
+    return /*#__PURE__*/ jsx("div", {
+        className: className ?? DEFAULT_WRAPPER,
+        children: /*#__PURE__*/ jsx(react_markdown, {
+            remarkPlugins: plugins,
+            components: components,
+            children: children
+        })
+    });
+}
+export { AiChatMarkdown };

package/dist/registry/ai-chat/components/ai-chat-message-actions.d.ts

@@ -0,0 +1,14 @@
+import type { MessageFeedbackType } from "../types";
+interface AiChatMessageActionsProps {
+    content: string;
+    messageRole: "user" | "assistant";
+    /** When true, actions are always visible (used for the latest assistant message). */
+    isLatest?: boolean;
+    /** Current feedback for this message — drives the thumbs-up/down pressed state. */
+    feedback?: MessageFeedbackType | null;
+    onFeedback?: (type: MessageFeedbackType) => void;
+    onRegenerate?: () => void;
+    onEdit?: () => void;
+}
+export declare function AiChatMessageActions({ content, messageRole, isLatest, feedback, onFeedback, onRegenerate, onEdit, }: AiChatMessageActionsProps): import("react/jsx-runtime").JSX.Element;
+export {};

package/dist/registry/ai-chat/components/ai-chat-message-actions.js

@@ -0,0 +1,106 @@
+"use client";
+import { Fragment, jsx, jsxs } from "react/jsx-runtime";
+import { useClipboard } from "@mantine/hooks";
+import { Root } from "@radix-ui/react-tooltip";
+import { Check, Copy, Pencil, RefreshCw, ThumbsDown, ThumbsUp } from "lucide-react";
+import { useVertexTranslation } from "../../../lib/vertex-translation.js";
+import { Button } from "../../button/button.js";
+import { TooltipContent, TooltipProvider, TooltipTrigger } from "../../tooltip/tooltip.js";
+import { cn } from "../../../lib/utils.js";
+function IconActionButton({ icon, label, onClick, pressed }) {
+    return /*#__PURE__*/ jsxs(Root, {
+        children: [
+            /*#__PURE__*/ jsx(TooltipTrigger, {
+                asChild: true,
+                children: /*#__PURE__*/ jsx(Button, {
+                    type: "button",
+                    variant: "ghost",
+                    size: "icon-xs",
+                    "aria-label": label,
+                    ...null == pressed ? {} : {
+                        "aria-pressed": pressed
+                    },
+                    onClick: onClick,
+                    className: cn("hover:bg-ai-chat-muted", pressed && "bg-ai-chat-muted text-foreground"),
+                    children: icon
+                })
+            }),
+            /*#__PURE__*/ jsx(TooltipContent, {
+                children: label
+            })
+        ]
+    });
+}
+function CopyToClipboardButton({ value }) {
+    const { t } = useVertexTranslation();
+    const { copied, error, copy } = useClipboard({
+        timeout: 2000
+    });
+    const label = copied ? t("copied") : error ? t("copy_failed") : t("copy");
+    return /*#__PURE__*/ jsx(IconActionButton, {
+        icon: copied ? /*#__PURE__*/ jsx(Check, {
+            className: "size-3.5 text-success",
+            "aria-hidden": "true"
+        }) : /*#__PURE__*/ jsx(Copy, {
+            className: "size-3.5 text-ai-chat-muted-foreground",
+            "aria-hidden": "true"
+        }),
+        label: label,
+        onClick: ()=>copy(value)
+    });
+}
+const ICON_CLASSES = "size-3.5 text-ai-chat-muted-foreground";
+function AiChatMessageActions({ content, messageRole, isLatest = false, feedback, onFeedback, onRegenerate, onEdit }) {
+    const { t } = useVertexTranslation();
+    return /*#__PURE__*/ jsx(TooltipProvider, {
+        delayDuration: 0,
+        skipDelayDuration: 150,
+        children: /*#__PURE__*/ jsxs("div", {
+            className: cn("flex items-center gap-0.5 -ml-[7px] transition-opacity", isLatest ? "opacity-100" : "opacity-0 group-hover/message:opacity-100 group-focus-within/message:opacity-100 has-[[data-state=delayed-open]]:opacity-100 has-[[data-state=instant-open]]:opacity-100"),
+            children: [
+                content.trim() && /*#__PURE__*/ jsx(CopyToClipboardButton, {
+                    value: content
+                }),
+                "assistant" === messageRole && onFeedback && /*#__PURE__*/ jsxs(Fragment, {
+                    children: [
+                        /*#__PURE__*/ jsx(IconActionButton, {
+                            icon: /*#__PURE__*/ jsx(ThumbsUp, {
+                                className: ICON_CLASSES,
+                                "aria-hidden": "true"
+                            }),
+                            label: t("good_response"),
+                            onClick: ()=>onFeedback("positive"),
+                            pressed: "positive" === feedback
+                        }),
+                        /*#__PURE__*/ jsx(IconActionButton, {
+                            icon: /*#__PURE__*/ jsx(ThumbsDown, {
+                                className: ICON_CLASSES,
+                                "aria-hidden": "true"
+                            }),
+                            label: t("bad_response"),
+                            onClick: ()=>onFeedback("negative"),
+                            pressed: "negative" === feedback
+                        })
+                    ]
+                }),
+                "assistant" === messageRole && isLatest && onRegenerate && /*#__PURE__*/ jsx(IconActionButton, {
+                    icon: /*#__PURE__*/ jsx(RefreshCw, {
+                        className: ICON_CLASSES,
+                        "aria-hidden": "true"
+                    }),
+                    label: t("try_again"),
+                    onClick: onRegenerate
+                }),
+                "user" === messageRole && onEdit && /*#__PURE__*/ jsx(IconActionButton, {
+                    icon: /*#__PURE__*/ jsx(Pencil, {
+                        className: ICON_CLASSES,
+                        "aria-hidden": "true"
+                    }),
+                    label: t("edit"),
+                    onClick: onEdit
+                })
+            ]
+        })
+    });
+}
+export { AiChatMessageActions };

package/dist/registry/ai-chat/components/ai-chat-message.d.ts

@@ -0,0 +1,15 @@
+import type { UIMessage } from "@tanstack/ai-client";
+import { type ReactNode } from "react";
+import type { MessageFeedbackType } from "../types";
+interface AiChatMessageProps {
+    message: UIMessage;
+    children?: ReactNode;
+    hideActions?: boolean;
+    showActionsAlwaysVisible?: boolean;
+    feedback?: MessageFeedbackType | null;
+    onFeedback?: (type: MessageFeedbackType) => void;
+    onRegenerate?: () => void;
+    onEditMessage?: (content: string) => void;
+}
+export declare function AiChatMessage({ message, children, hideActions, showActionsAlwaysVisible, feedback, onFeedback, onRegenerate, onEditMessage, }: AiChatMessageProps): import("react/jsx-runtime").JSX.Element | null;
+export {};