Feature/Hide or Show agent messages for share chatbot (#3140)

ability to hide/show agent messages for share chatbot
FlowiseAI · Sep 3, 2024 · f464704 · f464704
1 parent e691838
commit f464704
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/ui/src/views/chatflows/ShareChatbot.jsx b/packages/ui/src/views/chatflows/ShareChatbot.jsx
@@ -65,6 +65,7 @@ const ShareChatbot = ({ isSessionMemory, isAgentCanvas }) => {
     const [backgroundColor, setBackgroundColor] = useState(chatbotConfig?.backgroundColor ?? defaultConfig.backgroundColor)
     const [fontSize, setFontSize] = useState(chatbotConfig?.fontSize ?? defaultConfig.fontSize)
     const [poweredByTextColor, setPoweredByTextColor] = useState(chatbotConfig?.poweredByTextColor ?? defaultConfig.poweredByTextColor)
+    const [showAgentMessages, setShowAgentMessages] = useState(chatbotConfig?.showAgentMessages || (isAgentCanvas ? true : undefined))
 
     const [botMessageBackgroundColor, setBotMessageBackgroundColor] = useState(
         chatbotConfig?.botMessage?.backgroundColor ?? defaultConfig.botMessage.backgroundColor
@@ -139,7 +140,14 @@ const ShareChatbot = ({ isSessionMemory, isAgentCanvas }) => {
 
         if (chatbotConfig?.starterPrompts) obj.starterPrompts = chatbotConfig.starterPrompts
 
-        if (isAgentCanvas) obj.showAgentMessages = true
+        if (isAgentCanvas) {
+            // if showAgentMessages is undefined, default to true
+            if (showAgentMessages === undefined || showAgentMessages === null) {
+                obj.showAgentMessages = true
+            } else {
+                obj.showAgentMessages = showAgentMessages
+            }
+        }
 
         return obj
     }
@@ -301,6 +309,9 @@ const ShareChatbot = ({ isSessionMemory, isAgentCanvas }) => {
             case 'generateNewSession':
                 setGenerateNewSession(value)
                 break
+            case 'showAgentMessages':
+                setShowAgentMessages(value)
+                break
         }
     }
 
@@ -428,6 +439,7 @@ const ShareChatbot = ({ isSessionMemory, isAgentCanvas }) => {
             {colorField(backgroundColor, 'backgroundColor', 'Background Color')}
             {textField(fontSize, 'fontSize', 'Font Size', 'number')}
             {colorField(poweredByTextColor, 'poweredByTextColor', 'PoweredBy TextColor')}
+            {booleanField(showAgentMessages, 'showAgentMessages', 'Show Agent Reasoning')}
 
             {/*BOT Message*/}
             <Typography variant='h4' sx={{ mb: 1, mt: 2 }}>