Add a method to delete all chat memories in ChatMemory

Signed-off-by: fanxt0218 <[email protected]>

Author: fanxt0218

memory/repository/spring-ai-model-chat-memory-repository-cassandra/src/main/java/org/springframework/ai/chat/memory/repository/cassandra/CassandraChatMemoryRepository.java

@@ -45,6 +45,7 @@
  * An implementation of {@link ChatMemoryRepository} for Apache Cassandra.
  *
  * @author Mick Semb Wever
+ * @author Xiaotong Fan
  * @since 1.0.0
  */
 public final class CassandraChatMemoryRepository implements ChatMemoryRepository {
@@ -165,6 +166,14 @@ public void deleteByConversationId(String conversationId) {
 		saveAll(conversationId, List.of());
 	}
 
+	@Override
+	public void deleteConversationIds() {
+		List<String> conversationIds = findConversationIds();
+		for (String conversationId : conversationIds) {
+			deleteByConversationId(conversationId);
+		}
+	}
+
 	private PreparedStatement prepareAddStmt() {
 		RegularInsert stmt = null;
 		InsertInto stmtStart = QueryBuilder.insertInto(this.conf.schema.keyspace(), this.conf.schema.table());

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/HsqldbChatMemoryRepositoryDialect.java

@@ -36,6 +36,11 @@ public String getDeleteMessagesSql() {
 		return "DELETE FROM SPRING_AI_CHAT_MEMORY WHERE conversation_id = ?";
 	}
 
+	@Override
+	public String getDeleteAllMessageSql() {
+		return "DELETE FROM SPRING_AI_CHAT_MEMORY";
+	}
+
 	@Override
 	public String getSelectConversationIdsSql() {
 		return "SELECT DISTINCT conversation_id FROM SPRING_AI_CHAT_MEMORY";

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/JdbcChatMemoryRepository.java

@@ -53,6 +53,7 @@
  * @author Linar Abzaltdinov
  * @author Mark Pollack
  * @author Yanming Zhou
+ * @author Xiaotong Fan
  * @since 1.0.0
  */
 public final class JdbcChatMemoryRepository implements ChatMemoryRepository {
@@ -106,6 +107,12 @@ public void deleteByConversationId(String conversationId) {
 		this.jdbcTemplate.update(this.dialect.getDeleteMessagesSql(), conversationId);
 	}
 
+	@Override
+	public void deleteConversationIds() {
+		this.jdbcTemplate.update(this.dialect.getDeleteAllMessageSql());
+	}
+
+
 	public static Builder builder() {
 		return new Builder();
 	}

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/JdbcChatMemoryRepositoryDialect.java

@@ -45,6 +45,11 @@ public interface JdbcChatMemoryRepositoryDialect {
 	 */
 	String getDeleteMessagesSql();
 
+	/**
+	 * Returns the SQL to delete all conversation IDs.
+	 */
+	String getDeleteAllMessageSql();
+
 	/**
 	 * Optionally, dialect can provide more advanced SQL as needed.
 	 */

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/MysqlChatMemoryRepositoryDialect.java

@@ -43,5 +43,9 @@ public String getSelectConversationIdsSql() {
 	public String getDeleteMessagesSql() {
 		return "DELETE FROM SPRING_AI_CHAT_MEMORY WHERE conversation_id = ?";
 	}
+	@Override
+	public String getDeleteAllMessageSql() {
+		return "DELETE FROM SPRING_AI_CHAT_MEMORY";
+	}
 
 }

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/PostgresChatMemoryRepositoryDialect.java

@@ -44,4 +44,9 @@ public String getDeleteMessagesSql() {
 		return "DELETE FROM SPRING_AI_CHAT_MEMORY WHERE conversation_id = ?";
 	}
 
+	@Override
+	public String getDeleteAllMessageSql() {
+		return "DELETE FROM SPRING_AI_CHAT_MEMORY";
+	}
+
 }

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/SqlServerChatMemoryRepositoryDialect.java

@@ -44,4 +44,9 @@ public String getDeleteMessagesSql() {
 		return "DELETE FROM SPRING_AI_CHAT_MEMORY WHERE conversation_id = ?";
 	}
 
+	@Override
+	public String getDeleteAllMessageSql() {
+		return "DELETE FROM SPRING_AI_CHAT_MEMORY";
+	}
+
 }

memory/repository/spring-ai-model-chat-memory-repository-neo4j/src/main/java/org/springframework/ai/chat/memory/repository/neo4j/Neo4jChatMemoryRepository.java

@@ -45,6 +45,7 @@
  *
  * @author Enrico Rampazzo
  * @author Michael J. Simons
+ * @author Xiaotong Fan
  * @since 1.0.0
  */
 
@@ -166,6 +167,14 @@ OPTIONAL MATCH (m)-[:HAS_TOOL_CALL]->(tc:%s)
 		}
 	}
 
+	@Override
+	public void deleteConversationIds() {
+		List<String> conversationIds = findConversationIds();
+		for (String conversationId : conversationIds) {
+			deleteByConversationId(conversationId);
+		}
+	}
+
 	public Neo4jChatMemoryRepositoryConfig getConfig() {
 		return this.config;
 	}

spring-ai-model/src/main/java/org/springframework/ai/chat/memory/ChatMemory.java

@@ -26,6 +26,7 @@
  *
  * @author Christian Tzolov
  * @author Thomas Vitale
+ * @author Xiaotong Fan
  * @since 1.0.0
  */
 public interface ChatMemory {
@@ -61,4 +62,9 @@ default void add(String conversationId, Message message) {
 	 */
 	void clear(String conversationId);
 
+	/**
+	  Clear all the chat memory.
+	 */
+	void clear();
+
 }

spring-ai-model/src/main/java/org/springframework/ai/chat/memory/ChatMemoryRepository.java

@@ -24,6 +24,7 @@
  * A repository for storing and retrieving chat messages.
  *
  * @author Thomas Vitale
+ * @author Xiaotong Fan
  * @since 1.0.0
  */
 public interface ChatMemoryRepository {
@@ -40,4 +41,8 @@ public interface ChatMemoryRepository {
 
 	void deleteByConversationId(String conversationId);
 
+	/**
+	* Deletes all the conversation IDs.
+	*/
+	void deleteConversationIds();
 }

spring-ai-model/src/main/java/org/springframework/ai/chat/memory/InMemoryChatMemoryRepository.java

@@ -28,6 +28,7 @@
  * An in-memory implementation of {@link ChatMemoryRepository}.
  *
  * @author Thomas Vitale
+ * @author Xiaotong Fan
  * @since 1.0.0
  */
 public final class InMemoryChatMemoryRepository implements ChatMemoryRepository {
@@ -60,4 +61,9 @@ public void deleteByConversationId(String conversationId) {
 		this.chatMemoryStore.remove(conversationId);
 	}
 
+	@Override
+	public void deleteConversationIds() {
+		this.chatMemoryStore.clear();
+	}
+
 }

spring-ai-model/src/main/java/org/springframework/ai/chat/memory/MessageWindowChatMemory.java

@@ -37,6 +37,7 @@
  *
  * @author Thomas Vitale
  * @author Ilayaperumal Gopinathan
+ * @author Xiaotong Fan
  * @since 1.0.0
  */
 public final class MessageWindowChatMemory implements ChatMemory {
@@ -77,6 +78,11 @@ public void clear(String conversationId) {
 		this.chatMemoryRepository.deleteByConversationId(conversationId);
 	}
 
+	@Override
+	public void clear() {
+		this.chatMemoryRepository.deleteConversationIds();
+	}
+
 	private List<Message> process(List<Message> memoryMessages, List<Message> newMessages) {
 		List<Message> processedMessages = new ArrayList<>();