Class CassandraChatMemoryStore

java.lang.Object

dev.langchain4j.store.memory.chat.cassandra.CassandraChatMemoryStore

All Implemented Interfaces:

ChatMemoryStore

public class CassandraChatMemoryStore
extends Object
implements ChatMemoryStore

Implementation of ChatMemoryStore using Astra DB Vector Search. Table contains all chats. (default name is message_store). Each chat with multiple messages is a partition.Message id is a time uuid.

See Also:

• Astra Vector Store Documentation

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	CassandraChatMemoryStore.Builder
static class	CassandraChatMemoryStore.BuilderAstra

Field Summary

Fields

Modifier and Type	Field	Description
static final String	DEFAULT_TABLE_NAME	Default message store.

Constructor Summary

Constructors

Constructor	Description
CassandraChatMemoryStore(com.datastax.oss.driver.api.core.CqlSession session)	Constructor for message store
CassandraChatMemoryStore(com.datastax.oss.driver.api.core.CqlSession session, String tableName)	Constructor for message store

Method Summary

Modifier and Type	Method	Description
static CassandraChatMemoryStore.Builder	builder()
static CassandraChatMemoryStore.BuilderAstra	builderAstra()
void	clear()	Delete all rows.
void	create()	Create the table if not exist.
void	delete()	Delete the table.
void	deleteMessages(@NonNull Object memoryId)
com.datastax.oss.driver.api.core.CqlSession	getCassandraSession()	Access the cassandra session for fined grained operation.
List<ChatMessage>	getMessages(@NonNull Object memoryId)
void	updateMessages(@NonNull Object memoryId, @NonNull List<ChatMessage> messages)

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

DEFAULT_TABLE_NAME

public static final String DEFAULT_TABLE_NAME

Default message store.

See Also:

• Constant Field Values

Constructor Details

CassandraChatMemoryStore

public CassandraChatMemoryStore(com.datastax.oss.driver.api.core.CqlSession session)

Constructor for message store

Parameters:

session - cassandra session

CassandraChatMemoryStore

public CassandraChatMemoryStore(com.datastax.oss.driver.api.core.CqlSession session,
 String tableName)

Constructor for message store

Parameters:

session - cassandra session

tableName - table name

Method Details

create

public void create()

Create the table if not exist.

delete

public void delete()

Delete the table.

clear

public void clear()

Delete all rows.

getCassandraSession

public com.datastax.oss.driver.api.core.CqlSession getCassandraSession()

Access the cassandra session for fined grained operation.

Returns:

current cassandra session

getMessages

public List<ChatMessage> getMessages(@NonNull Object memoryId)

Specified by:

getMessages in interface ChatMemoryStore

updateMessages

public void updateMessages(@NonNull Object memoryId,
 @NonNull List<ChatMessage> messages)

Specified by:

updateMessages in interface ChatMemoryStore

deleteMessages

public void deleteMessages(@NonNull Object memoryId)

Specified by:

deleteMessages in interface ChatMemoryStore

builder

public static CassandraChatMemoryStore.Builder builder()

builderAstra

public static CassandraChatMemoryStore.BuilderAstra builderAstra()