Refactor chat memory repository artifacts for clarity · spring-projects/spring-ai@2d517ee (original) (raw)
`@@ -22,6 +22,7 @@
`
22
22
`import org.junit.Test;
`
23
23
`import org.junit.runner.RunWith;
`
24
24
``
``
25
`+
import org.springframework.ai.chat.memory.repository.jdbc.JdbcChatMemoryRepository;
`
25
26
`import org.springframework.ai.chat.messages.AssistantMessage;
`
26
27
`import org.springframework.ai.chat.messages.Message;
`
27
28
`import org.springframework.ai.chat.messages.UserMessage;
`
`@@ -39,7 +40,7 @@
`
39
40
`import static org.assertj.core.api.Assertions.fail;
`
40
41
``
41
42
`@RunWith(SpringRunner.class)
`
42
``
`-
@SpringBootTest(classes = JdbcChatMemoryHsqldbAutoConfigurationIT.TestConfig.class,
`
``
43
`+
@SpringBootTest(classes = JdbcChatMemoryRepositoryHsqldbAutoConfigurationIT.TestConfig.class,
`
43
44
`properties = { "spring.datasource.url=jdbc:hsqldb:mem:chat_memory_auto_configuration_test;DB_CLOSE_DELAY=-1",
`
44
45
`"spring.datasource.username=sa", "spring.datasource.password=",
`
45
46
`"spring.datasource.driver-class-name=org.hsqldb.jdbcDriver",
`
`@@ -54,7 +55,7 @@
`
54
55
`org.springframework.boot.autoconfigure.jdbc.JdbcTemplateAutoConfiguration.class,
`
55
56
`org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration.class,
`
56
57
`SqlInitializationAutoConfiguration.class })
`
57
``
`-
public class JdbcChatMemoryHsqldbAutoConfigurationIT {
`
``
58
`+
public class JdbcChatMemoryRepositoryHsqldbAutoConfigurationIT {
`
58
59
``
59
60
`@Autowired
`
60
61
`private ApplicationContext context;
`
`@@ -130,7 +131,7 @@ public void useAutoConfiguredChatMemoryWithJdbc() {
`
130
131
`try {
`
131
132
`java.util.Enumeration<java.net.URL> resources = Thread.currentThread()
`
132
133
` .getContextClassLoader()
`
133
``
`-
.getResources("org/springframework/ai/chat/memory/jdbc/schema-hsqldb.sql");
`
``
134
`+
.getResources("org/springframework/ai/chat/memory/repository/jdbc/schema-hsqldb.sql");
`
134
135
`System.out.println("--- schema-hsqldb.sql resources found on classpath ---");
`
135
136
`while (resources.hasMoreElements()) {
`
136
137
`System.out.println(resources.nextElement());
`
`@@ -157,7 +158,7 @@ public void useAutoConfiguredChatMemoryWithJdbc() {
`
157
158
``
158
159
`// Now test the ChatMemory functionality
`
159
160
`assertThat(context.getBean(org.springframework.ai.chat.memory.ChatMemory.class)).isNotNull();
`
160
``
`-
assertThat(context.getBean(org.springframework.ai.chat.memory.jdbc.JdbcChatMemoryRepository.class)).isNotNull();
`
``
161
`+
assertThat(context.getBean(JdbcChatMemoryRepository.class)).isNotNull();
`
161
162
``
162
163
`var chatMemory = context.getBean(org.springframework.ai.chat.memory.ChatMemory.class);
`
163
164
`var conversationId = java.util.UUID.randomUUID().toString();
`