MessagePassingDocumentExtractor

class MessagePassingDocumentExtractor(val messageRolesToExtract: Set<Message.Role> = setOf(Message.Role.User, Message.Role.Assistant)) : DocumentExtractor(source)

Default extractor that filters messages by role.

This extractor filters messages to only include those with roles in messageRolesToExtract, then converts each message's content into TextDocuments.

Constructors

constructor(messageRolesToExtract: Set<Message.Role> = setOf(Message.Role.User, Message.Role.Assistant))

The set of message roles to extract and persist. Defaults to setOf(Message.Role.User, Message.Role.Assistant).

open suspend override fun extract(messages: List<Message>): List<TextDocument>

Transforms a list of messages into a list of TextDocuments for storage.