From 6a0f8564f356f0286262ff5aaa53dd758b14d4b6 Mon Sep 17 00:00:00 2001
From: Enrico Ros <enrico.ros@gmail.com>
Date: Mon, 26 Feb 2024 13:12:28 -0800
Subject: [PATCH] Beam: as command

---
 src/apps/chat/AppChat.tsx | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/apps/chat/AppChat.tsx b/src/apps/chat/AppChat.tsx
index 40ff60e08..aae5461bb 100644
--- a/src/apps/chat/AppChat.tsx
+++ b/src/apps/chat/AppChat.tsx
@@ -166,6 +166,9 @@ export function AppChat() {
       const chatCommand = extractChatCommand(lastMessage.text)[0];
       if (chatCommand && chatCommand.type === 'cmd') {
         switch (chatCommand.providerId) {
+          case 'ass-beam':
+            return ConversationManager.getHandler(conversationId).beamStore.create(history);
+
           case 'ass-browse':
             setMessages(conversationId, history);
             return await runBrowseGetPageUpdatingState(conversationId, chatCommand.params!);
@@ -200,6 +203,9 @@ export function AppChat() {
             const helpMessage = createDMessage('assistant', 'Available Chat Commands:\n' + chatCommandsText);
             helpMessage.originLLM = Brand.Title.Base;
             return setMessages(conversationId, [...history, helpMessage]);
+
+          default:
+            return setMessages(conversationId, [...history, createDMessage('assistant', 'This command is not supported.')]);
         }
       }
     }
@@ -211,10 +217,7 @@ export function AppChat() {
           return await runAssistantUpdatingState(conversationId, history, chatLLMId, focusedSystemPurposeId, getUXLabsHighPerformance() ? 0 : getInstantAppChatPanesCount());
 
         case 'generate-text-beam':
-          if (!lastMessage?.text)
-            break;
-          ConversationManager.getHandler(conversationId).beamStore.create(history);
-          return;
+          return ConversationManager.getHandler(conversationId).beamStore.create(history);
 
         case 'append-user':
           return setMessages(conversationId, history);