Optimise CassandraChatMemoryRepository for MessageWindowChatMemory usage pattern #3097
Conversation
michaelsembwever
force-pushed
the
mck/CassandraChatMemoryRepository_0
branch
2 times, most recently
from
af0c80d to
a5649da
Compare
…age pattern Time-series each chat window in Cassandra, keeping past (and deleted) windows still in the db. Add ability to store different MessageTypes. Signed-off-by: mck <[email protected]>
michaelsembwever
force-pushed
the
mck/CassandraChatMemoryRepository_0
branch
from
a5649da to
73e31d1
Compare
| stmt = stmt.whereColumn(columnName).isEqualTo(QueryBuilder.bindMarker(columnName)); | ||
| private Message getMessage(UdtValue udt) { | ||
| String content = udt.getString(this.conf.messageUdtContentColumn); | ||
| Map<String, Object> props = Map.of(CONVERSATION_TS, udt.getInstant(this.conf.messageUdtTimestampColumn)); |
There was a problem hiding this comment.
@ZYMCao messages have the timestamp value in their metadata, as an Instant instance, under the CONVERSATION_TS key.
The literal for that key is found here: https://github.com/spring-projects/spring-ai/blob/main/memory/spring-ai-model-chat-memory-cassandra/src/main/java/org/springframework/ai/chat/memory/cassandra/CassandraChatMemoryRepository.java#L50-L51
There was a problem hiding this comment.
@michaelsembwever
unix timestamp is great, but I am not sure about fraction part:
[
{
"messageType": "USER",
"metadata": {
"CassandraChatMemoryRepository_message_timestamp": 1747564565.737000000,
"messageType": "USER"
},
"media": [],
"text": "SSE development using springboot?"
},
{
"messageType": "ASSISTANT",
"metadata": {
"CassandraChatMemoryRepository_message_timestamp": 1747564594.962000000,
"messageType": "ASSISTANT"
},
"toolCalls": [],
"media": [],
"text": "..."
}
]
|
@michaelsembwever Thanks for the PR! Rebased and merged as 87b680a |
4 participants
Time-series each chat window in Cassandra, keeping past (and deleted) windows still in the db. Add ability to store different MessageTypes.
ref: #2998