Skip to content

Commit 00b0aab

Browse files
committed
Refactor Cassandra ChatMemory Autoconfiguration
- Move Cassandra ChatMemory autoconfiguration into dedicated autoconfiguration Signed-off-by: Ilayaperumal Gopinathan <[email protected]>
1 parent 1d4bde8 commit 00b0aab

File tree

8 files changed

+17
-46
lines changed

8 files changed

+17
-46
lines changed

Diff for: auto-configurations/models/chat/memory/spring-ai-autoconfigure-model-chat-memory/pom.xml renamed to auto-configurations/models/chat/memory/spring-ai-autoconfigure-model-chat-memory-cassandra/pom.xml

+3-3
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,10 @@
99
<version>1.0.0-SNAPSHOT</version>
1010
<relativePath>../../../../../pom.xml</relativePath>
1111
</parent>
12-
<artifactId>spring-ai-autoconfigure-model-chat-memory</artifactId>
12+
<artifactId>spring-ai-autoconfigure-model-chat-memory-cassandra</artifactId>
1313
<packaging>jar</packaging>
14-
<name>Spring AI Chat Memory Auto Configuration</name>
15-
<description>Spring AI Chat Memory Auto Configuration</description>
14+
<name>Spring AI Cassandra Chat Memory Auto Configuration</name>
15+
<description>Spring AI Cassandra Chat Memory Auto Configuration</description>
1616
<url>https://github.com/spring-projects/spring-ai</url>
1717

1818
<scm>
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
import org.slf4j.Logger;
2222
import org.slf4j.LoggerFactory;
2323

24-
import org.springframework.ai.model.chat.memory.autoconfigure.CommonChatMemoryProperties;
2524
import org.springframework.ai.chat.memory.cassandra.CassandraChatMemoryConfig;
2625
import org.springframework.boot.context.properties.ConfigurationProperties;
2726
import org.springframework.lang.Nullable;
@@ -34,7 +33,7 @@
3433
* @since 1.0.0
3534
*/
3635
@ConfigurationProperties(CassandraChatMemoryProperties.CONFIG_PREFIX)
37-
public class CassandraChatMemoryProperties extends CommonChatMemoryProperties {
36+
public class CassandraChatMemoryProperties {
3837

3938
public static final String CONFIG_PREFIX = "spring.ai.chat.memory.cassandra";
4039

@@ -48,6 +47,16 @@ public class CassandraChatMemoryProperties extends CommonChatMemoryProperties {
4847

4948
private String userColumn = CassandraChatMemoryConfig.DEFAULT_USER_COLUMN_NAME;
5049

50+
private boolean initializeSchema = true;
51+
52+
public boolean isInitializeSchema() {
53+
return this.initializeSchema;
54+
}
55+
56+
public void setInitializeSchema(boolean initializeSchema) {
57+
this.initializeSchema = initializeSchema;
58+
}
59+
5160
private Duration timeToLive = null;
5261

5362
public String getKeyspace() {
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
1515
#
16-
cassandra.org.springframework.ai.model.chat.memory.autoconfigure.CassandraChatMemoryAutoConfiguration
16+
org.springframework.ai.model.chat.memory.cassandra.autoconfigure.CassandraChatMemoryAutoConfiguration
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,12 @@
1414
* limitations under the License.
1515
*/
1616

17-
package org.springframework.ai.model.chat.memory.autoconfigure;
17+
package org.springframework.ai.model.chat.memory.cassandra.autoconfigure;
1818

1919
import java.time.Duration;
2020

2121
import org.junit.jupiter.api.Test;
2222

23-
import org.springframework.ai.model.chat.memory.cassandra.autoconfigure.CassandraChatMemoryProperties;
2423
import org.springframework.ai.chat.memory.cassandra.CassandraChatMemoryConfig;
2524

2625
import static org.assertj.core.api.Assertions.assertThat;

Diff for: auto-configurations/models/chat/memory/spring-ai-autoconfigure-model-chat-memory/src/main/java/org/springframework/ai/model/chat/memory/autoconfigure/CommonChatMemoryProperties.java

-37
This file was deleted.

Diff for: pom.xml

+1-1
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141

4242
<module>auto-configurations/models/chat/client/spring-ai-autoconfigure-model-chat-client</module>
4343

44-
<module>auto-configurations/models/chat/memory/spring-ai-autoconfigure-model-chat-memory</module>
44+
<module>auto-configurations/models/chat/memory/spring-ai-autoconfigure-model-chat-memory-cassandra</module>
4545
<module>auto-configurations/models/chat/memory/spring-ai-autoconfigure-model-chat-memory-neo4j</module>
4646

4747
<module>auto-configurations/models/chat/observation/spring-ai-autoconfigure-model-chat-observation</module>

0 commit comments

Comments
 (0)