Skip to content

Commit 1cd337d

Browse files
committed
fix JdbcChatMemoryRepository to support developer message
Signed-off-by: Andres da Silva Santos <40636137+andresssantos@users.noreply.github.com>
1 parent 38086bc commit 1cd337d

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

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

+2
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
import org.springframework.ai.chat.messages.Message;
3131
import org.springframework.ai.chat.messages.MessageType;
3232
import org.springframework.ai.chat.messages.SystemMessage;
33+
import org.springframework.ai.chat.messages.DeveloperMessage;
3334
import org.springframework.ai.chat.messages.ToolResponseMessage;
3435
import org.springframework.ai.chat.messages.UserMessage;
3536
import org.springframework.jdbc.core.BatchPreparedStatementSetter;
@@ -129,6 +130,7 @@ public Message mapRow(ResultSet rs, int i) throws SQLException {
129130
case USER -> new UserMessage(content);
130131
case ASSISTANT -> new AssistantMessage(content);
131132
case SYSTEM -> new SystemMessage(content);
133+
case DEVELOPER -> new DeveloperMessage(content);
132134
// The content is always stored empty for ToolResponseMessages.
133135
// If we want to capture the actual content, we need to extend
134136
// AddBatchPreparedStatement to support it.

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

+2
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
import org.springframework.ai.chat.messages.MessageType;
2525
import org.springframework.ai.chat.messages.AssistantMessage;
2626
import org.springframework.ai.chat.messages.SystemMessage;
27+
import org.springframework.ai.chat.messages.DeveloperMessage;
2728
import org.springframework.ai.chat.messages.UserMessage;
2829
import org.springframework.beans.factory.annotation.Autowired;
2930
import org.springframework.boot.SpringBootConfiguration;
@@ -72,6 +73,7 @@ void saveMessagesSingleMessage(String content, MessageType messageType) {
7273
case ASSISTANT -> new AssistantMessage(content + " - " + conversationId);
7374
case USER -> new UserMessage(content + " - " + conversationId);
7475
case SYSTEM -> new SystemMessage(content + " - " + conversationId);
76+
case DEVELOPER -> new DeveloperMessage(content + " - " + conversationId);
7577
default -> throw new IllegalArgumentException("Type not supported: " + messageType);
7678
};
7779

0 commit comments

Comments
 (0)