feat: GH-4433 Add a method to ChatMemory for retrieving all conversation IDs, and add corresponding unit tests.

Signed-off-by: Sun Yuhan <[email protected]>

Author: sunyuhan1998

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

@@ -26,6 +26,7 @@
  *
  * @author Christian Tzolov
  * @author Thomas Vitale
+ * @author Sun Yuhan
  * @since 1.0.0
  */
 public interface ChatMemory {
@@ -61,4 +62,10 @@ default void add(String conversationId, Message message) {
 	 */
 	void clear(String conversationId);
 
+	/**
+	 * Get all conversation IDs
+	 * @return All conversation IDs
+	 */
+	List<String> getConversations();
+
 }

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

@@ -37,6 +37,7 @@
  *
  * @author Thomas Vitale
  * @author Ilayaperumal Gopinathan
+ * @author Sun Yuhan
  * @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 List<String> getConversations() {
+		return this.chatMemoryRepository.findConversationIds();
+	}
+
 	private List<Message> process(List<Message> memoryMessages, List<Message> newMessages) {
 		List<Message> processedMessages = new ArrayList<>();
 

spring-ai-model/src/test/java/org/springframework/ai/chat/memory/MessageWindowChatMemoryTests.java

@@ -292,4 +292,50 @@ void mixedMessagesWithLimitEqualToSystemMessageCount() {
 				new SystemMessage("System instruction 2"));
 	}
 
+	@Test
+	void getConversationsReturnsAllConversationIds() {
+		MessageWindowChatMemory chatMemory = MessageWindowChatMemory.builder().build();
+
+		String conversationId1 = UUID.randomUUID().toString();
+		String conversationId2 = UUID.randomUUID().toString();
+		String conversationId3 = UUID.randomUUID().toString();
+
+		chatMemory.add(conversationId1, new UserMessage("Hello from conversation 1"));
+		chatMemory.add(conversationId2, new UserMessage("Hello from conversation 2"));
+		chatMemory.add(conversationId3, new UserMessage("Hello from conversation 3"));
+
+		List<String> conversations = chatMemory.getConversations();
+
+		assertThat(conversations).contains(conversationId1, conversationId2, conversationId3);
+		assertThat(conversations).hasSize(3);
+	}
+
+	@Test
+	void getConversationsWithCustomRepository() {
+		InMemoryChatMemoryRepository customRepository = new InMemoryChatMemoryRepository();
+		MessageWindowChatMemory chatMemory = MessageWindowChatMemory.builder()
+			.chatMemoryRepository(customRepository)
+			.build();
+
+		String conversationId1 = UUID.randomUUID().toString();
+		String conversationId2 = UUID.randomUUID().toString();
+
+		chatMemory.add(conversationId1, new UserMessage("Message in conversation 1"));
+		chatMemory.add(conversationId2, new UserMessage("Message in conversation 2"));
+
+		List<String> conversations = chatMemory.getConversations();
+
+		assertThat(conversations).contains(conversationId1, conversationId2);
+		assertThat(conversations).hasSize(2);
+	}
+
+	@Test
+	void getConversationsReturnsEmptyListWhenNoConversations() {
+		MessageWindowChatMemory chatMemory = MessageWindowChatMemory.builder().build();
+
+		List<String> conversations = chatMemory.getConversations();
+
+		assertThat(conversations).isEmpty();
+	}
+
 }