🐛 (sendMessage) Correctly preprocess and parse fetched bot

apps/viewer/src/features/blocks/logic/typebotLink/executeTypebotLink.ts

@@ -8,9 +8,10 @@ import {
   SessionState,
   Variable,
   ReplyLog,
-  Typebot,
   VariableWithValue,
   Edge,
+  typebotInSessionStateSchema,
+  TypebotInSession,
 } from '@typebot.io/schemas'
 import { ExecuteLogicResponse } from '@/features/chat/types'
 import { createId } from '@paralleldrive/cuid2'
@@ -73,7 +74,7 @@ export const executeTypebotLink = async (
 const addLinkedTypebotToState = async (
   state: SessionState,
   block: TypebotLinkBlock,
-  linkedTypebot: Pick<Typebot, 'id' | 'edges' | 'groups' | 'variables'>
+  linkedTypebot: TypebotInSession
 ): Promise<SessionState> => {
   const currentTypebotInQueue = state.typebotsQueue[0]
   const isPreview = isNotDefined(currentTypebotInQueue.resultId)
@@ -191,17 +192,19 @@ const fetchTypebot = async (state: SessionState, typebotId: string) => {
     const typebot = await prisma.typebot.findUnique({
       where: { id: typebotId },
       select: {
+        version: true,
         id: true,
         edges: true,
         groups: true,
         variables: true,
       },
     })
-    return typebot as Pick<Typebot, 'id' | 'edges' | 'groups' | 'variables'>
+    return typebotInSessionStateSchema.parse(typebot)
   }
   const typebot = await prisma.publicTypebot.findUnique({
     where: { typebotId },
     select: {
+      version: true,
       id: true,
       edges: true,
       groups: true,
@@ -209,8 +212,8 @@ const fetchTypebot = async (state: SessionState, typebotId: string) => {
     },
   })
   if (!typebot) return null
-  return {
+  return typebotInSessionStateSchema.parse({
     ...typebot,
     id: typebotId,
-  } as Pick<Typebot, 'id' | 'edges' | 'groups' | 'variables'>
+  })
 }

apps/viewer/src/features/chat/api/sendMessage.ts

@@ -11,8 +11,8 @@ import {
   SessionState,
   StartParams,
   StartTypebot,
+  startTypebotSchema,
   Theme,
-  Typebot,
   Variable,
   VariableWithValue,
 } from '@typebot.io/schemas'
@@ -152,6 +152,7 @@ const startSession = async (
       {
         resultId: result?.id,
         typebot: {
+          version: typebot.version,
           id: typebot.id,
           groups: typebot.groups,
           edges: typebot.edges,
@@ -268,17 +269,14 @@ const getTypebot = async (
 
   const parsedTypebot =
     typebotQuery && 'typebot' in typebotQuery
-      ? ({
+      ? {
           id: typebotQuery.typebotId,
           ...omit(typebotQuery.typebot, 'workspace'),
           ...omit(typebotQuery, 'typebot', 'typebotId'),
-        } as StartTypebot & Pick<Typebot, 'isArchived' | 'isClosed'>)
-      : (typebotQuery as StartTypebot & Pick<Typebot, 'isArchived'>)
+        }
+      : typebotQuery
 
-  if (
-    !parsedTypebot ||
-    ('isArchived' in parsedTypebot && parsedTypebot.isArchived)
-  )
+  if (!parsedTypebot || parsedTypebot.isArchived)
     throw new TRPCError({
       code: 'NOT_FOUND',
       message: 'Typebot not found',
@@ -299,7 +297,7 @@ const getTypebot = async (
       message: 'Typebot is closed',
     })
 
-  return parsedTypebot
+  return startTypebotSchema.parse(parsedTypebot)
 }
 
 const getResult = async ({

apps/viewer/src/features/chat/chat.spec.ts

@@ -142,6 +142,7 @@ test('API chat execution should work on published bot', async ({ request }) => {
         data: { message: '8', sessionId: chatSessionId },
       })
     ).json()
+    console.log(messages, input)
     expect(messages[0].content.richText).toStrictEqual([
       {
         children: [{ text: "I'm gonna shoot multiple inputs now..." }],

apps/viewer/src/features/chat/helpers/continueBotFlow.ts

@@ -137,7 +137,7 @@ export const continueBotFlow =
           formattedReply !== reply ? formattedReply : undefined,
       }
 
-    const nextGroup = getNextGroup(newSessionState)(nextEdgeId)
+    const nextGroup = await getNextGroup(newSessionState)(nextEdgeId)
 
     newSessionState = nextGroup.newSessionState
 

apps/viewer/src/features/chat/helpers/executeGroup.ts

@@ -124,7 +124,9 @@ export const executeGroup =
     if (!nextEdgeId && state.typebotsQueue.length === 1)
       return { messages, newSessionState, clientSideActions, logs }
 
-    const nextGroup = getNextGroup(newSessionState)(nextEdgeId ?? undefined)
+    const nextGroup = await getNextGroup(newSessionState)(
+      nextEdgeId ?? undefined
+    )
 
     newSessionState = nextGroup.newSessionState
 

apps/viewer/src/features/chat/helpers/getNextGroup.ts

@@ -1,5 +1,6 @@
 import { byId } from '@typebot.io/lib'
 import { Group, SessionState } from '@typebot.io/schemas'
+import { upsertResult } from '../queries/upsertResult'
 
 export type NextGroup = {
   group?: Group
@@ -8,12 +9,20 @@ export type NextGroup = {
 
 export const getNextGroup =
   (state: SessionState) =>
-  (edgeId?: string): NextGroup => {
+  async (edgeId?: string): Promise<NextGroup> => {
     const nextEdge = state.typebotsQueue[0].typebot.edges.find(byId(edgeId))
     if (!nextEdge) {
       if (state.typebotsQueue.length > 1) {
         const nextEdgeId = state.typebotsQueue[0].edgeIdToTriggerWhenDone
         const isMergingWithParent = state.typebotsQueue[0].isMergingWithParent
+        const currentResultId = state.typebotsQueue[0].resultId
+        if (!isMergingWithParent && currentResultId)
+          await upsertResult({
+            resultId: currentResultId,
+            typebot: state.typebotsQueue[0].typebot,
+            isCompleted: true,
+            hasStarted: state.typebotsQueue[0].answers.length > 0,
+          })
         const newSessionState = {
           ...state,
           typebotsQueue: [
@@ -48,7 +57,7 @@ export const getNextGroup =
             ...state.typebotsQueue.slice(2),
           ],
         } satisfies SessionState
-        const nextGroup = getNextGroup(newSessionState)(nextEdgeId)
+        const nextGroup = await getNextGroup(newSessionState)(nextEdgeId)
         if (!nextGroup)
           return {
             newSessionState,

apps/viewer/src/features/chat/helpers/startBotFlow.ts

@@ -21,7 +21,7 @@ export const startBotFlow = async (
   const firstEdgeId =
     state.typebotsQueue[0].typebot.groups[0].blocks[0].outgoingEdgeId
   if (!firstEdgeId) return { messages: [], newSessionState: state }
-  const nextGroup = getNextGroup(state)(firstEdgeId)
+  const nextGroup = await getNextGroup(state)(firstEdgeId)
   if (!nextGroup.group) return { messages: [], newSessionState: state }
   return executeGroup(state)(nextGroup.group)
 }

apps/viewer/src/features/chat/queries/findPublicTypebot.ts

@@ -8,6 +8,7 @@ export const findPublicTypebot = ({ publicId }: Props) =>
   prisma.publicTypebot.findFirst({
     where: { typebot: { publicId } },
     select: {
+      version: true,
       groups: true,
       edges: true,
       settings: true,

apps/viewer/src/features/chat/queries/findTypebot.ts

@@ -9,6 +9,7 @@ export const findTypebot = ({ id, userId }: Props) =>
   prisma.typebot.findFirst({
     where: { id, workspace: { members: { some: { userId } } } },
     select: {
+      version: true,
       id: true,
       groups: true,
       edges: true,