Writers for Haystack.
Module haystack_experimental.components.writers.chat_message_writer
ChatMessageWriter
Writes chat messages to an underlying ChatMessageStore.
Usage example:
from haystack.dataclasses import ChatMessage
from haystack_experimental.components.writers import ChatMessageWriter
from haystack_experimental.chat_message_stores.in_memory import InMemoryChatMessageStore
messages = [
ChatMessage.from_assistant("Hello, how can I help you?"),
ChatMessage.from_user("I have a question about Python."),
]
message_store = InMemoryChatMessageStore()
writer = ChatMessageWriter(message_store)
writer.run(messages)
ChatMessageWriter.__init__
def __init__(message_store: ChatMessageStore)
Create a ChatMessageWriter component.
Arguments:
message_store
: The ChatMessageStore where the chat messages are to be written.
ChatMessageWriter.to_dict
def to_dict() -> Dict[str, Any]
Serializes the component to a dictionary.
Returns:
Dictionary with serialized data.
ChatMessageWriter.from_dict
@classmethod
def from_dict(cls, data: Dict[str, Any]) -> "ChatMessageWriter"
Deserializes the component from a dictionary.
Arguments:
data
: The dictionary to deserialize from.
Raises:
DeserializationError
: If the message store is not properly specified in the serialization data or its type cannot be imported.
Returns:
The deserialized component.
ChatMessageWriter.run
@component.output_types(messages_written=int)
def run(messages: List[ChatMessage])
Run the ChatMessageWriter on the given input data.
Arguments:
messages
: A list of chat messages to write to the store.
Raises:
ValueError
: If the specified message store is not found.
Returns:
messages_written
: Number of messages written to the ChatMessageStore.