diff --git a/web/src/components/playground/MessageActions.js b/web/src/components/playground/MessageActions.js index 7ad2b529..7156a8ce 100644 --- a/web/src/components/playground/MessageActions.js +++ b/web/src/components/playground/MessageActions.js @@ -7,15 +7,24 @@ import { RefreshCw, Copy, Trash2, + UserCheck, } from 'lucide-react'; import { useTranslation } from 'react-i18next'; -const MessageActions = ({ message, styleState, onMessageReset, onMessageCopy, onMessageDelete, isAnyMessageGenerating = false }) => { +const MessageActions = ({ + message, + styleState, + onMessageReset, + onMessageCopy, + onMessageDelete, + onRoleToggle, + isAnyMessageGenerating = false +}) => { const { t } = useTranslation(); const isLoading = message.status === 'loading' || message.status === 'incomplete'; - const shouldDisableActions = isAnyMessageGenerating; + const canToggleRole = message.role === 'assistant' || message.role === 'system'; return (
@@ -48,6 +57,30 @@ const MessageActions = ({ message, styleState, onMessageReset, onMessageCopy, on )} + {canToggleRole && !isLoading && ( + +