feat: assistantActionBarCopy - extend thread config properties for copy action

Author: bsebben

packages/react-ui/src/ui/assistant-action-bar.tsx

@@ -99,23 +99,29 @@ namespace AssistantActionBarCopy {
 const AssistantActionBarCopy = forwardRef<
   AssistantActionBarCopy.Element,
   AssistantActionBarCopy.Props
->(({ copiedDuration, ...props }, ref) => {
+>(({ copiedDuration, onClick: copyAction, ...props }, ref) => {
   const {
+    assistantMessage: { copy: { icon: { Check = CheckIcon, Copy = CheckIcon } = {}, onCopy } = {} } = {},
     strings: {
       assistantMessage: { copy: { tooltip = "Copy" } = {} } = {},
     } = {},
   } = useThreadConfig();
 
+  const handleClick = e => {
+    copyAction();
+    onCopy?.();
+  }
+
   return (
     <ActionBarPrimitive.Copy copiedDuration={copiedDuration} asChild>
-      <TooltipIconButton tooltip={tooltip} {...props} ref={ref}>
+      <TooltipIconButton tooltip={tooltip} {...props} onClick={handleClick} ref={ref}>
         {props.children ?? (
           <>
             <MessagePrimitive.If copied>
-              <CheckIcon />
+              <Check />
             </MessagePrimitive.If>
             <MessagePrimitive.If copied={false}>
-              <CopyIcon />
+              <Copy />
             </MessagePrimitive.If>
           </>
         )}

packages/react-ui/src/ui/thread-config.tsx

@@ -34,6 +34,14 @@ export type UserMessageConfig = {
   allowEdit?: boolean | undefined;
 };
 
+export type AssistantMessageCopyConfig = {
+  icon: {
+    copy: ComponentType,
+    copied: ComponentType
+  },
+  onCopy?: () => void;
+}
+
 export type AssistantMessageConfig = {
   allowReload?: boolean | undefined;
   allowCopy?: boolean | undefined;
@@ -48,6 +56,7 @@ export type AssistantMessageConfig = {
         Footer?: ComponentType | undefined;
       }
     | undefined;
+  copy?: AssistantMessageCopyConfig | undefined;
 };
 
 export type BranchPickerConfig = {