From cae045d52bd9d0005273e297763f25c65077ea38 Mon Sep 17 00:00:00 2001 From: mck Date: Sun, 5 May 2024 19:53:22 +0200 Subject: [PATCH] Add the autoconfigure property for the CassandraVectorStore option to return embeddings in documents from similarity searches ref: https://github.com/spring-projects/spring-ai/pull/673 --- .../CassandraVectorStoreAutoConfiguration.java | 3 +++ .../cassandra/CassandraVectorStoreProperties.java | 12 +++++++++++- 2 files changed, 14 insertions(+), 1 deletion(-) 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; }