53 lines
1.7 KiB
TypeScript
53 lines
1.7 KiB
TypeScript
import type { OpenAI } from 'openai'
|
|
import { options as createChatCompletionOption } from '../actions/createChatCompletion'
|
|
import { ReadOnlyVariableStore } from '@typebot.io/forge'
|
|
import { isNotEmpty } from '@typebot.io/lib'
|
|
import { z } from '@typebot.io/forge/zod'
|
|
|
|
export const parseChatCompletionMessages = ({
|
|
options: { messages },
|
|
variables,
|
|
}: {
|
|
options: Pick<z.infer<typeof createChatCompletionOption>, 'messages'>
|
|
variables: ReadOnlyVariableStore
|
|
}): OpenAI.Chat.ChatCompletionMessageParam[] => {
|
|
const parsedMessages = messages
|
|
?.flatMap((message) => {
|
|
if (!message.role) return
|
|
|
|
if (message.role === 'Dialogue') {
|
|
if (!message.dialogueVariableId) return
|
|
const dialogue = variables.get(message.dialogueVariableId) ?? []
|
|
const dialogueArr = Array.isArray(dialogue) ? dialogue : [dialogue]
|
|
|
|
return dialogueArr.map((dialogueItem, index) => {
|
|
if (index === 0 && message.startsBy === 'assistant')
|
|
return {
|
|
role: 'assistant',
|
|
content: dialogueItem,
|
|
}
|
|
return {
|
|
role:
|
|
index % (message.startsBy === 'assistant' ? 1 : 2) === 0
|
|
? 'user'
|
|
: 'assistant',
|
|
content: dialogueItem,
|
|
}
|
|
})
|
|
}
|
|
|
|
if (!message.content) return
|
|
|
|
return {
|
|
role: message.role,
|
|
content: variables.parse(message.content),
|
|
} satisfies OpenAI.Chat.ChatCompletionMessageParam
|
|
})
|
|
.filter(
|
|
(message) =>
|
|
isNotEmpty(message?.role) && isNotEmpty(message?.content?.toString())
|
|
) as OpenAI.Chat.ChatCompletionMessageParam[]
|
|
|
|
return parsedMessages
|
|
}
|