52 lines
1.7 KiB
TypeScript
52 lines
1.7 KiB
TypeScript
import { byId, isDefined } from '@typebot.io/lib'
|
|
import { ContinueChatResponse, SessionState } from '@typebot.io/schemas'
|
|
import { ChatCompletionOpenAIOptions } from '@typebot.io/schemas/features/blocks/integrations/openai'
|
|
import { VariableWithUnknowValue } from '@typebot.io/schemas/features/typebot/variable'
|
|
import { updateVariablesInSession } from '@typebot.io/variables/updateVariablesInSession'
|
|
|
|
export const resumeChatCompletion =
|
|
(
|
|
state: SessionState,
|
|
{
|
|
outgoingEdgeId,
|
|
options,
|
|
logs = [],
|
|
}: {
|
|
outgoingEdgeId?: string
|
|
options: ChatCompletionOpenAIOptions
|
|
logs?: ContinueChatResponse['logs']
|
|
}
|
|
) =>
|
|
async (message: string, totalTokens?: number) => {
|
|
let newSessionState = state
|
|
const newVariables = options.responseMapping?.reduce<
|
|
VariableWithUnknowValue[]
|
|
>((newVariables, mapping) => {
|
|
const { typebot } = newSessionState.typebotsQueue[0]
|
|
const existingVariable = typebot.variables.find(byId(mapping.variableId))
|
|
if (!existingVariable) return newVariables
|
|
if (mapping.valueToExtract === 'Message content') {
|
|
newVariables.push({
|
|
...existingVariable,
|
|
value: Array.isArray(existingVariable.value)
|
|
? existingVariable.value.concat(message)
|
|
: message,
|
|
})
|
|
}
|
|
if (mapping.valueToExtract === 'Total tokens' && isDefined(totalTokens)) {
|
|
newVariables.push({
|
|
...existingVariable,
|
|
value: totalTokens,
|
|
})
|
|
}
|
|
return newVariables
|
|
}, [])
|
|
if (newVariables && newVariables.length > 0)
|
|
newSessionState = updateVariablesInSession(newSessionState)(newVariables)
|
|
return {
|
|
outgoingEdgeId,
|
|
newSessionState,
|
|
logs,
|
|
}
|
|
}
|