diff --git a/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx b/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx
index 0f57c5c0551..e2aa9d94339 100644
--- a/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx
+++ b/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.test.tsx
@@ -1,7 +1,7 @@
/**
* @vitest-environment jsdom
*/
-import { act, type ReactNode } from 'react'
+import { act, type ComponentProps, type ReactNode } from 'react'
import { createRoot, type Root } from 'react-dom/client'
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
@@ -19,6 +19,11 @@ vi.mock('next/navigation', () => ({
}))
vi.mock('@sim/emcn', () => ({
+ Button: ({
+ variant,
+ iconSize,
+ ...props
+ }: ComponentProps<'button'> & { variant?: string; iconSize?: string }) => ,
Check: () => null,
Duplicate: () => null,
Split: () => null,
diff --git a/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.tsx b/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.tsx
index 6207a1e247a..28780a945dd 100644
--- a/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.tsx
+++ b/apps/sim/app/workspace/[workspaceId]/components/message-actions/message-actions.tsx
@@ -2,6 +2,7 @@
import { memo, useEffect, useRef, useState } from 'react'
import {
+ Button,
Check,
ChipModal,
ChipModalBody,
@@ -27,8 +28,6 @@ import { useForkMothershipChat } from '@/hooks/queries/mothership-chats'
import { useFolderStore } from '@/stores/folders/store'
const ICON_CLASS = 'size-[14px]'
-const BUTTON_CLASS =
- 'flex size-[26px] items-center justify-center rounded-[6px] text-[var(--text-icon)] transition-colors hover-hover:bg-[var(--surface-hover)] focus-visible:outline-hidden'
interface MessageActionsProps {
content: string
@@ -162,14 +161,15 @@ export const MessageActions = memo(function MessageActions({
{canCopyContent && (
-
+
{copied ? 'Copied message' : 'Copy message'}
@@ -180,27 +180,29 @@ export const MessageActions = memo(function MessageActions({
<>
-
+
Good response
-
+
Bad response
@@ -209,15 +211,16 @@ export const MessageActions = memo(function MessageActions({
{canFork && (
-
+
Fork in new chat