♻️ Introduce typebot v6 with events (#1013)

Closes #885

packages/schemas/features/blocks/integrations/openai/schema.ts

@@ -0,0 +1,126 @@
+import { z } from 'zod'
+import {
+  chatCompletionMessageCustomRoles,
+  chatCompletionMessageRoles,
+  chatCompletionResponseValues,
+  openAITasks,
+} from './constants'
+import { variableStringSchema } from '../../../utils'
+import { blockBaseSchema, credentialsBaseSchema } from '../../shared'
+import { IntegrationBlockType } from '../constants'
+
+const openAIBaseOptionsSchema = z.object({
+  credentialsId: z.string().optional(),
+  baseUrl: z.string().optional(),
+  apiVersion: z.string().optional(),
+})
+
+const initialOptionsSchema = z
+  .object({
+    task: z.undefined(),
+  })
+  .merge(openAIBaseOptionsSchema)
+
+export const nativeMessageSchema = z.object({
+  id: z.string(),
+  role: z.enum(chatCompletionMessageRoles).optional(),
+  content: z.string().optional(),
+  name: z.string().optional(),
+})
+
+const messageSequenceItemSchema = z.object({
+  id: z.string(),
+  role: z.literal('Messages sequence ✨'),
+  content: z
+    .object({
+      assistantMessagesVariableId: z.string().optional(),
+      userMessagesVariableId: z.string().optional(),
+    })
+    .optional(),
+})
+
+const dialogueItemSchema = z.object({
+  id: z.string(),
+  role: z.literal('Dialogue'),
+  dialogueVariableId: z.string().optional(),
+  startsBy: z.enum(['user', 'assistant']).optional(),
+})
+
+const chatCompletionOptionsSchema = z
+  .object({
+    task: z.literal(openAITasks[0]),
+    model: z.string().optional(),
+    messages: z
+      .array(
+        z.union([
+          nativeMessageSchema,
+          messageSequenceItemSchema,
+          dialogueItemSchema,
+        ])
+      )
+      .optional(),
+    advancedSettings: z
+      .object({
+        temperature: z.number().or(variableStringSchema).optional(),
+      })
+      .optional(),
+    responseMapping: z
+      .array(
+        z.object({
+          id: z.string(),
+          valueToExtract: z.preprocess(
+            (val) => (!val ? 'Message content' : val),
+            z.enum(chatCompletionResponseValues)
+          ),
+          variableId: z.string().optional(),
+        })
+      )
+      .optional(),
+  })
+  .merge(openAIBaseOptionsSchema)
+
+const createImageOptionsSchema = z
+  .object({
+    task: z.literal(openAITasks[1]),
+    prompt: z.string().optional(),
+    advancedOptions: z.object({
+      size: z.enum(['256x256', '512x512', '1024x1024']).optional(),
+    }),
+    responseMapping: z.array(
+      z.object({
+        id: z.string(),
+        valueToExtract: z.enum(['Image URL']),
+        variableId: z.string().optional(),
+      })
+    ),
+  })
+  .merge(openAIBaseOptionsSchema)
+
+export const openAIBlockSchema = blockBaseSchema.merge(
+  z.object({
+    type: z.enum([IntegrationBlockType.OPEN_AI]),
+    options: z
+      .discriminatedUnion('task', [
+        initialOptionsSchema,
+        chatCompletionOptionsSchema,
+        createImageOptionsSchema,
+      ])
+      .optional(),
+  })
+)
+
+export const openAICredentialsSchema = z
+  .object({
+    type: z.literal('openai'),
+    data: z.object({
+      apiKey: z.string(),
+    }),
+  })
+  .merge(credentialsBaseSchema)
+
+export type OpenAICredentials = z.infer<typeof openAICredentialsSchema>
+export type OpenAIBlock = z.infer<typeof openAIBlockSchema>
+export type ChatCompletionOpenAIOptions = z.infer<
+  typeof chatCompletionOptionsSchema
+>
+export type CreateImageOpenAIOptions = z.infer<typeof createImageOptionsSchema>