diff --git a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreAutoConfiguration.java b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreAutoConfiguration.java index 23a077ee3..f9e760169 100644 --- a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreAutoConfiguration.java +++ b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreAutoConfiguration.java @@ -57,6 +57,9 @@ public class CassandraVectorStoreAutoConfiguration { if (properties.getDisallowSchemaCreation()) { builder = builder.disallowSchemaChanges(); } + if (properties.getReturnEmbeddings()) { + builder = builder.returnEmbeddings(); + } return new CassandraVectorStore(builder.build(), embeddingClient); } diff --git a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreProperties.java b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreProperties.java index 27af7605e..9725e7559 100644 --- a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreProperties.java +++ b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vectorstore/cassandra/CassandraVectorStoreProperties.java @@ -41,6 +41,8 @@ public class CassandraVectorStoreProperties { private boolean disallowSchemaChanges = false; + private boolean returnEmbeddings = false; + private int fixedThreadPoolExecutorSize = CassandraVectorStoreConfig.DEFAULT_ADD_CONCURRENCY; public String getKeyspace() { @@ -83,7 +85,7 @@ public class CassandraVectorStoreProperties { this.embeddingColumnName = embeddingColumnName; } - public Boolean getDisallowSchemaCreation() { + public boolean getDisallowSchemaCreation() { return this.disallowSchemaChanges; } @@ -91,6 +93,14 @@ public class CassandraVectorStoreProperties { this.disallowSchemaChanges = disallowSchemaCreation; } + public boolean getReturnEmbeddings() { + return this.returnEmbeddings; + } + + public void setReturnEmbeddings(boolean returnEmbeddings) { + this.returnEmbeddings = returnEmbeddings; + } + public int getFixedThreadPoolExecutorSize() { return this.fixedThreadPoolExecutorSize; }