Add a method to delete all chat memories in ChatMemory

fanxt0218 · fanxt0218 · commit b15b7fab3d7f · 2025-06-18T16:58:17.000+08:00
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-cassandra/src/main/java/org/springframework/ai/chat/memory/repository/cassandra/CassandraChatMemoryRepository.java b/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());
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/HsqldbChatMemoryRepositoryDialect.java b/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";
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/JdbcChatMemoryRepository.java b/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();
 	}
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/JdbcChatMemoryRepositoryDialect.java b/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.
 	 */
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/MysqlChatMemoryRepositoryDialect.java b/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";
+	}
 
 }
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/PostgresChatMemoryRepositoryDialect.java b/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";
+	}
+
 }
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/SqlServerChatMemoryRepositoryDialect.java b/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";
+	}
+
 }
diff --git a/memory/repository/spring-ai-model-chat-memory-repository-neo4j/src/main/java/org/springframework/ai/chat/memory/repository/neo4j/Neo4jChatMemoryRepository.java b/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;
 	}
diff --git a/spring-ai-model/src/main/java/org/springframework/ai/chat/memory/ChatMemory.java b/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();
+
 }
diff --git a/spring-ai-model/src/main/java/org/springframework/ai/chat/memory/ChatMemoryRepository.java b/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();
 }
diff --git a/spring-ai-model/src/main/java/org/springframework/ai/chat/memory/InMemoryChatMemoryRepository.java b/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();
+	}
+
 }
diff --git a/spring-ai-model/src/main/java/org/springframework/ai/chat/memory/MessageWindowChatMemory.java b/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<>();
 

Original file line number	Diff line number	Diff line change
`@@ -43,5 +43,9 @@ public String getSelectConversationIdsSql() {`
`43`	`43`	`public String getDeleteMessagesSql() {`
`44`	`44`	`return "DELETE FROM SPRING_AI_CHAT_MEMORY WHERE conversation_id = ?";`
`45`	`45`	`}`
	`46`	`+ @Override`
	`47`	`+ public String getDeleteAllMessageSql() {`
	`48`	`+ return "DELETE FROM SPRING_AI_CHAT_MEMORY";`
	`49`	`+ }`
`46`	`50`
`47`	`51`	`}`
Original file line number	Diff line number	Diff line change
`@@ -44,4 +44,9 @@ public String getDeleteMessagesSql() {`
`44`	`44`	`return "DELETE FROM SPRING_AI_CHAT_MEMORY WHERE conversation_id = ?";`
`45`	`45`	`}`
`46`	`46`
	`47`	`+ @Override`
	`48`	`+ public String getDeleteAllMessageSql() {`
	`49`	`+ return "DELETE FROM SPRING_AI_CHAT_MEMORY";`
	`50`	`+ }`
	`51`	`+`
`47`	`52`	`}`
Original file line number	Diff line number	Diff line change
`@@ -45,6 +45,7 @@`
`45`	`45`	`*`
`46`	`46`	`* @author Enrico Rampazzo`
`47`	`47`	`* @author Michael J. Simons`
	`48`	`+ * @author Xiaotong Fan`
`48`	`49`	`* @since 1.0.0`
`49`	`50`	`*/`
`50`	`51`
`@@ -166,6 +167,14 @@ OPTIONAL MATCH (m)-[:HAS_TOOL_CALL]->(tc:%s)`
`166`	`167`	`}`
`167`	`168`	`}`
`168`	`169`
	`170`	`+ @Override`
	`171`	`+ public void deleteConversationIds() {`
	`172`	`+ List<String> conversationIds = findConversationIds();`
	`173`	`+ for (String conversationId : conversationIds) {`
	`174`	`+ deleteByConversationId(conversationId);`
	`175`	`+ }`
	`176`	`+ }`
	`177`	`+`
`169`	`178`	`public Neo4jChatMemoryRepositoryConfig getConfig() {`
`170`	`179`	`return this.config;`
`171`	`180`	`}`
Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,7 @@`
`24`	`24`	`* A repository for storing and retrieving chat messages.`
`25`	`25`	`*`
`26`	`26`	`* @author Thomas Vitale`
	`27`	`+ * @author Xiaotong Fan`
`27`	`28`	`* @since 1.0.0`
`28`	`29`	`*/`
`29`	`30`	`public interface ChatMemoryRepository {`
`@@ -40,4 +41,8 @@ public interface ChatMemoryRepository {`
`40`	`41`
`41`	`42`	`void deleteByConversationId(String conversationId);`
`42`	`43`
	`44`	`+ /**`
	`45`	`+ * Deletes all the conversation IDs.`
	`46`	`+ */`
	`47`	`+ void deleteConversationIds();`
`43`	`48`	`}`