139 lines
3.6 KiB
TypeScript
139 lines
3.6 KiB
TypeScript
import { z } from 'zod'
|
|
import {
|
|
chatCompletionMessageCustomRoles,
|
|
chatCompletionMessageRoles,
|
|
chatCompletionResponseValues,
|
|
openAITasks,
|
|
openAIVoices,
|
|
} 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)
|
|
export type ChatCompletionOpenAIOptions = z.infer<
|
|
typeof chatCompletionOptionsSchema
|
|
>
|
|
|
|
const createImageOptionsSchema = z
|
|
.object({
|
|
task: z.literal(openAITasks[2]),
|
|
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 type CreateImageOpenAIOptions = z.infer<typeof createImageOptionsSchema>
|
|
|
|
const createSpeechOptionsSchema = openAIBaseOptionsSchema.extend({
|
|
task: z.literal(openAITasks[1]),
|
|
model: z.string().optional(),
|
|
input: z.string().optional(),
|
|
voice: z.enum(openAIVoices).optional(),
|
|
saveUrlInVariableId: z.string().optional(),
|
|
})
|
|
export type CreateSpeechOpenAIOptions = z.infer<
|
|
typeof createSpeechOptionsSchema
|
|
>
|
|
|
|
export const openAIBlockSchema = blockBaseSchema.merge(
|
|
z.object({
|
|
type: z.enum([IntegrationBlockType.OPEN_AI]),
|
|
options: z
|
|
.discriminatedUnion('task', [
|
|
initialOptionsSchema,
|
|
chatCompletionOptionsSchema,
|
|
createImageOptionsSchema,
|
|
createSpeechOptionsSchema,
|
|
])
|
|
.optional(),
|
|
})
|
|
)
|
|
export type OpenAIBlock = z.infer<typeof openAIBlockSchema>
|
|
|
|
export const openAICredentialsSchema = z
|
|
.object({
|
|
type: z.literal('openai'),
|
|
data: z.object({
|
|
apiKey: z.string(),
|
|
}),
|
|
})
|
|
.merge(credentialsBaseSchema)
|
|
export type OpenAICredentials = z.infer<typeof openAICredentialsSchema>
|