⚡ (chat) Improve chat API compatibility with preview mode

apps/viewer/src/features/chat/api/utils/executeGroup.ts

@@ -23,7 +23,7 @@ export const executeGroup =
   (state: SessionState, currentReply?: ChatReply) =>
   async (
     group: Group
-  ): Promise<ChatReply & { newSessionState?: SessionState }> => {
+  ): Promise<ChatReply & { newSessionState: SessionState }> => {
     const messages: ChatReply['messages'] = currentReply?.messages ?? []
     let logic: ChatReply['logic'] = currentReply?.logic
     let integrations: ChatReply['integrations'] = currentReply?.integrations
@@ -72,8 +72,10 @@ export const executeGroup =
         integrations = executionResponse.integrations
       if (executionResponse.newSessionState)
         newSessionState = executionResponse.newSessionState
-      if (executionResponse.outgoingEdgeId)
+      if (executionResponse.outgoingEdgeId) {
         nextEdgeId = executionResponse.outgoingEdgeId
+        break
+      }
     }
 
     if (!nextEdgeId) return { messages, newSessionState, logic, integrations }

apps/viewer/src/features/chat/api/utils/startBotFlow.ts

@@ -1,13 +1,26 @@
+import { TRPCError } from '@trpc/server'
 import { ChatReply, SessionState } from 'models'
 import { executeGroup } from './executeGroup'
 import { getNextGroup } from './getNextGroup'
 
 export const startBotFlow = async (
-  state: SessionState
-): Promise<ChatReply & { newSessionState?: SessionState }> => {
+  state: SessionState,
+  startGroupId?: string
+): Promise<ChatReply & { newSessionState: SessionState }> => {
+  if (startGroupId) {
+    const group = state.typebot.groups.find(
+      (group) => group.id === startGroupId
+    )
+    if (!group)
+      throw new TRPCError({
+        code: 'BAD_REQUEST',
+        message: "startGroupId doesn't exist",
+      })
+    return executeGroup(state)(group)
+  }
   const firstEdgeId = state.typebot.groups[0].blocks[0].outgoingEdgeId
-  if (!firstEdgeId) return { messages: [] }
+  if (!firstEdgeId) return { messages: [], newSessionState: state }
   const nextGroup = getNextGroup(state)(firstEdgeId)
-  if (!nextGroup) return { messages: [] }
+  if (!nextGroup) return { messages: [], newSessionState: state }
   return executeGroup(state)(nextGroup.group)
 }