⚡ (chat) Improve chat API compatibility with preview mode

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)
 }