diff --git a/web/src/components/playground/MessageActions.js b/web/src/components/playground/MessageActions.js index 7156a8ce..0ed40950 100644 --- a/web/src/components/playground/MessageActions.js +++ b/web/src/components/playground/MessageActions.js @@ -8,6 +8,7 @@ import { Copy, Trash2, UserCheck, + Edit, } from 'lucide-react'; import { useTranslation } from 'react-i18next'; @@ -18,13 +19,16 @@ const MessageActions = ({ onMessageCopy, onMessageDelete, onRoleToggle, - isAnyMessageGenerating = false + onMessageEdit, + isAnyMessageGenerating = false, + isEditing = false }) => { const { t } = useTranslation(); const isLoading = message.status === 'loading' || message.status === 'incomplete'; - const shouldDisableActions = isAnyMessageGenerating; + const shouldDisableActions = isAnyMessageGenerating || isEditing; const canToggleRole = message.role === 'assistant' || message.role === 'system'; + const canEdit = !isLoading && message.content && typeof onMessageEdit === 'function' && !isEditing; return (
@@ -57,6 +61,21 @@ const MessageActions = ({ )} + {canEdit && ( + +