wehub-resource-sync
45 lines
1.6 KiB
TypeScript
45 lines
1.6 KiB
TypeScript
import { UserResolverWithCaching } from '../user-resolver'
|
|
import { MessageFormatter, MessageTextExtractor, MessageTextOutput } from './message-formatter'
|
|
import { Message } from './message-types'
|
|
|
|
export type TranscriptFormatter = (extractedMessages: MessageTextOutput[]) => Promise<string> | string
|
|
|
|
export const buildConversationTranscript = async ({
|
|
messages,
|
|
ctx,
|
|
client,
|
|
customMessageFormatters,
|
|
customTranscriptFormatter,
|
|
}: {
|
|
ctx: { botUserId: string }
|
|
client: ConstructorParameters<typeof UserResolverWithCaching>[0]
|
|
messages: Message[]
|
|
customMessageFormatters?: { [K in Message['type']]?: MessageFormatter }
|
|
customTranscriptFormatter?: TranscriptFormatter
|
|
}): Promise<string> => {
|
|
const messageExtractor = new MessageTextExtractor({
|
|
botUserId: ctx.botUserId,
|
|
customFormatters: customMessageFormatters,
|
|
userResolver: new UserResolverWithCaching(client),
|
|
})
|
|
|
|
const extractedMessages = await messageExtractor.extractTextFromMessages(messages)
|
|
const transcriptFormatter = customTranscriptFormatter ?? DEFAULT_TRANSCRIPT_FORMATTER
|
|
const transcript = await transcriptFormatter(extractedMessages)
|
|
|
|
return transcript
|
|
}
|
|
|
|
const DEFAULT_TRANSCRIPT_FORMATTER: TranscriptFormatter = (extractedMessages) => {
|
|
const formattedMessages = extractedMessages.map((message) => {
|
|
const emoji = message.isBot ? '🤖' : '👤'
|
|
const name = message.user.name ?? (message.isBot ? 'Botpress' : 'User')
|
|
const header = `${emoji} ${name}:`
|
|
const body = message.text.join('\n')
|
|
|
|
return `${header}\n${body}`
|
|
})
|
|
|
|
return formattedMessages.join('\n\n---\n\n')
|
|
}
|