From c5b7dd9c46ada0ca4843e7449eee1854da9e5485 Mon Sep 17 00:00:00 2001
From: Josh Long <josh@joshlong.com>
Date: Mon, 6 May 2024 12:36:37 +0200
Subject: [PATCH] i ran the spring javaformat maven plugin to clean up any
 source code incompatabilities.

---
 .../vectorstore/azure/AzureVectorStore.java   | 42 +++++++++----------
 .../CassandraFilterExpressionConverter.java   |  3 +-
 .../ai/vectorstore/ChromaVectorStore.java     |  1 +
 .../vectorstore/ElasticsearchVectorStore.java |  1 +
 .../ai/vectorstore/MilvusVectorStore.java     |  2 +-
 .../ai/vectorstore/Neo4jVectorStore.java      | 10 ++---
 .../ai/vectorstore/PgVectorStore.java         | 16 +++----
 .../vectorstore/qdrant/QdrantVectorStore.java | 25 +++++------
 8 files changed, 53 insertions(+), 47 deletions(-)

diff --git a/vector-stores/spring-ai-azure-vector-store/src/main/java/org/springframework/ai/vectorstore/azure/AzureVectorStore.java b/vector-stores/spring-ai-azure-vector-store/src/main/java/org/springframework/ai/vectorstore/azure/AzureVectorStore.java
index 071a48fa5..9a5205cfb 100644
--- a/vector-stores/spring-ai-azure-vector-store/src/main/java/org/springframework/ai/vectorstore/azure/AzureVectorStore.java
+++ b/vector-stores/spring-ai-azure-vector-store/src/main/java/org/springframework/ai/vectorstore/azure/AzureVectorStore.java
@@ -109,37 +109,37 @@ public class AzureVectorStore implements VectorStore, ApplicationListener<Applic
 		List<SearchField> fields = new ArrayList<>();
 
 		fields.add(new SearchField(ID_FIELD_NAME, SearchFieldDataType.STRING).setKey(true)
-				.setFilterable(true)
-				.setSortable(true));
+			.setFilterable(true)
+			.setSortable(true));
 		fields.add(new SearchField(EMBEDDING_FIELD_NAME, SearchFieldDataType.collection(SearchFieldDataType.SINGLE))
-				.setSearchable(true)
-				.setVectorSearchDimensions(dimensions)
-				// This must match a vector search configuration name.
-				.setVectorSearchProfileName(SPRING_AI_VECTOR_PROFILE));
+			.setSearchable(true)
+			.setVectorSearchDimensions(dimensions)
+			// This must match a vector search configuration name.
+			.setVectorSearchProfileName(SPRING_AI_VECTOR_PROFILE));
 		fields.add(new SearchField(CONTENT_FIELD_NAME, SearchFieldDataType.STRING).setSearchable(true)
-				.setFilterable(true));
+			.setFilterable(true));
 		fields.add(new SearchField(METADATA_FIELD_NAME, SearchFieldDataType.STRING).setSearchable(true)
-				.setFilterable(true));
+			.setFilterable(true));
 
 		for (MetadataField filterableMetadataField : this.filterMetadataFields) {
 			fields.add(new SearchField(METADATA_FIELD_PREFIX + filterableMetadataField.name(),
 					filterableMetadataField.fieldType())
-					.setSearchable(false)
-					.setFacetable(true));
+				.setSearchable(false)
+				.setFacetable(true));
 		}
 
 		SearchIndex searchIndex = new SearchIndex(this.indexName).setFields(fields)
-				// VectorSearch configuration is required for a vector field. The name used
-				// for the vector search algorithm configuration must match the configuration
-				// used by the search field used for vector search.
-				.setVectorSearch(new VectorSearch()
-						.setProfiles(Collections
-								.singletonList(new VectorSearchProfile(SPRING_AI_VECTOR_PROFILE, SPRING_AI_VECTOR_CONFIG)))
-						.setAlgorithms(Collections.singletonList(new HnswAlgorithmConfiguration(SPRING_AI_VECTOR_CONFIG)
-								.setParameters(new HnswParameters().setM(4)
-										.setEfConstruction(400)
-										.setEfSearch(1000)
-										.setMetric(VectorSearchAlgorithmMetric.COSINE)))));
+			// VectorSearch configuration is required for a vector field. The name used
+			// for the vector search algorithm configuration must match the configuration
+			// used by the search field used for vector search.
+			.setVectorSearch(new VectorSearch()
+				.setProfiles(Collections
+					.singletonList(new VectorSearchProfile(SPRING_AI_VECTOR_PROFILE, SPRING_AI_VECTOR_CONFIG)))
+				.setAlgorithms(Collections.singletonList(new HnswAlgorithmConfiguration(SPRING_AI_VECTOR_CONFIG)
+					.setParameters(new HnswParameters().setM(4)
+						.setEfConstruction(400)
+						.setEfSearch(1000)
+						.setMetric(VectorSearchAlgorithmMetric.COSINE)))));
 
 		SearchIndex index = this.searchIndexClient.createOrUpdateIndex(searchIndex);
 
diff --git a/vector-stores/spring-ai-cassandra/src/main/java/org/springframework/ai/vectorstore/CassandraFilterExpressionConverter.java b/vector-stores/spring-ai-cassandra/src/main/java/org/springframework/ai/vectorstore/CassandraFilterExpressionConverter.java
index 15028f725..f3270ac81 100644
--- a/vector-stores/spring-ai-cassandra/src/main/java/org/springframework/ai/vectorstore/CassandraFilterExpressionConverter.java
+++ b/vector-stores/spring-ai-cassandra/src/main/java/org/springframework/ai/vectorstore/CassandraFilterExpressionConverter.java
@@ -32,7 +32,8 @@ import java.util.function.Function;
 import java.util.stream.Collectors;
 
 /**
- * Converts {@link org.springframework.ai.vectorstore.filter.Filter.Expression} into CQL where clauses.
+ * Converts {@link org.springframework.ai.vectorstore.filter.Filter.Expression} into CQL
+ * where clauses.
  *
  * @author Mick Semb Wever
  * @since 1.0.0
diff --git a/vector-stores/spring-ai-chroma-store/src/main/java/org/springframework/ai/vectorstore/ChromaVectorStore.java b/vector-stores/spring-ai-chroma-store/src/main/java/org/springframework/ai/vectorstore/ChromaVectorStore.java
index 411398a6d..3c9741cbe 100644
--- a/vector-stores/spring-ai-chroma-store/src/main/java/org/springframework/ai/vectorstore/ChromaVectorStore.java
+++ b/vector-stores/spring-ai-chroma-store/src/main/java/org/springframework/ai/vectorstore/ChromaVectorStore.java
@@ -153,4 +153,5 @@ public class ChromaVectorStore implements VectorStore, ApplicationListener<Appli
 		}
 		this.collectionId = collection.id();
 	}
+
 }
diff --git a/vector-stores/spring-ai-elasticsearch-store/src/main/java/org/springframework/ai/vectorstore/ElasticsearchVectorStore.java b/vector-stores/spring-ai-elasticsearch-store/src/main/java/org/springframework/ai/vectorstore/ElasticsearchVectorStore.java
index 8dcb1b5bc..cff3a8efd 100644
--- a/vector-stores/spring-ai-elasticsearch-store/src/main/java/org/springframework/ai/vectorstore/ElasticsearchVectorStore.java
+++ b/vector-stores/spring-ai-elasticsearch-store/src/main/java/org/springframework/ai/vectorstore/ElasticsearchVectorStore.java
@@ -226,4 +226,5 @@ public class ElasticsearchVectorStore implements VectorStore, ApplicationListene
 			createIndexMapping();
 		}
 	}
+
 }
diff --git a/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/MilvusVectorStore.java b/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/MilvusVectorStore.java
index 76a54fddb..b86438b6c 100644
--- a/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/MilvusVectorStore.java
+++ b/vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/MilvusVectorStore.java
@@ -64,7 +64,7 @@ import org.springframework.util.StringUtils;
  * @author Christian Tzolov
  * @author Josh Long
  */
-public class MilvusVectorStore implements VectorStore, ApplicationListener <ApplicationReadyEvent> {
+public class MilvusVectorStore implements VectorStore, ApplicationListener<ApplicationReadyEvent> {
 
 	private static final Logger logger = LoggerFactory.getLogger(MilvusVectorStore.class);
 
diff --git a/vector-stores/spring-ai-neo4j-store/src/main/java/org/springframework/ai/vectorstore/Neo4jVectorStore.java b/vector-stores/spring-ai-neo4j-store/src/main/java/org/springframework/ai/vectorstore/Neo4jVectorStore.java
index bb1799584..8b5b3024f 100644
--- a/vector-stores/spring-ai-neo4j-store/src/main/java/org/springframework/ai/vectorstore/Neo4jVectorStore.java
+++ b/vector-stores/spring-ai-neo4j-store/src/main/java/org/springframework/ai/vectorstore/Neo4jVectorStore.java
@@ -52,8 +52,8 @@ public class Neo4jVectorStore implements VectorStore, ApplicationListener<Applic
 			this.name = name;
 		}
 
-
 	}
+
 	/**
 	 * Configuration for the Neo4j vector store.
 	 */
@@ -74,6 +74,7 @@ public class Neo4jVectorStore implements VectorStore, ApplicationListener<Applic
 		// needed for similarity search call
 
 		private final String indexNameNotSanitized;
+
 		private final String idProperty;
 
 		private final String constraintName;
@@ -251,7 +252,6 @@ public class Neo4jVectorStore implements VectorStore, ApplicationListener<Applic
 				return new Neo4jVectorStoreConfig(this);
 			}
 
-
 		}
 
 	}
@@ -262,9 +262,9 @@ public class Neo4jVectorStore implements VectorStore, ApplicationListener<Applic
 		try (var session = this.driver.session(this.config.sessionConfig)) {
 
 			session
-					.run("CREATE CONSTRAINT %s IF NOT EXISTS FOR (n:%s) REQUIRE n.%s IS UNIQUE"
-							.formatted(this.config.constraintName, this.config.label, this.config.idProperty))
-					.consume();
+				.run("CREATE CONSTRAINT %s IF NOT EXISTS FOR (n:%s) REQUIRE n.%s IS UNIQUE"
+					.formatted(this.config.constraintName, this.config.label, this.config.idProperty))
+				.consume();
 
 			var statement = """
 					CREATE VECTOR INDEX %s IF NOT EXISTS FOR (n:%s) ON (n.%s)
diff --git a/vector-stores/spring-ai-pgvector-store/src/main/java/org/springframework/ai/vectorstore/PgVectorStore.java b/vector-stores/spring-ai-pgvector-store/src/main/java/org/springframework/ai/vectorstore/PgVectorStore.java
index eab9e1024..2eb9eefee 100644
--- a/vector-stores/spring-ai-pgvector-store/src/main/java/org/springframework/ai/vectorstore/PgVectorStore.java
+++ b/vector-stores/spring-ai-pgvector-store/src/main/java/org/springframework/ai/vectorstore/PgVectorStore.java
@@ -52,7 +52,7 @@ import org.springframework.util.StringUtils;
  * @author Christian Tzolov
  * @author Josh Long
  */
-public class PgVectorStore implements VectorStore, ApplicationListener <ApplicationReadyEvent> {
+public class PgVectorStore implements VectorStore, ApplicationListener<ApplicationReadyEvent> {
 
 	private static final Logger logger = LoggerFactory.getLogger(PgVectorStore.class);
 
@@ -82,7 +82,7 @@ public class PgVectorStore implements VectorStore, ApplicationListener <Applicat
 
 	@Override
 	public void onApplicationEvent(ApplicationReadyEvent event) {
-        try {
+		try {
 			// Enable the PGVector, JSONB and UUID support.
 			this.jdbcTemplate.execute("CREATE EXTENSION IF NOT EXISTS vector");
 			this.jdbcTemplate.execute("CREATE EXTENSION IF NOT EXISTS hstore");
@@ -105,12 +105,14 @@ public class PgVectorStore implements VectorStore, ApplicationListener <Applicat
 			if (this.createIndexMethod != PgIndexType.NONE) {
 				this.jdbcTemplate.execute(String.format("""
 						CREATE INDEX IF NOT EXISTS %s ON %s USING %s (embedding %s)
-						""", VECTOR_INDEX_NAME, VECTOR_TABLE_NAME, this.createIndexMethod, this.getDistanceType().index));
+						""", VECTOR_INDEX_NAME, VECTOR_TABLE_NAME, this.createIndexMethod,
+						this.getDistanceType().index));
 			}
-		} catch (Exception e) {
-            throw new RuntimeException(e);
-        }
-    }
+		}
+		catch (Exception e) {
+			throw new RuntimeException(e);
+		}
+	}
 
 	/**
 	 * By default, pgvector performs exact nearest neighbor search, which provides perfect
diff --git a/vector-stores/spring-ai-qdrant/src/main/java/org/springframework/ai/vectorstore/qdrant/QdrantVectorStore.java b/vector-stores/spring-ai-qdrant/src/main/java/org/springframework/ai/vectorstore/qdrant/QdrantVectorStore.java
index 72dcf09b5..1b54c2f00 100644
--- a/vector-stores/spring-ai-qdrant/src/main/java/org/springframework/ai/vectorstore/qdrant/QdrantVectorStore.java
+++ b/vector-stores/spring-ai-qdrant/src/main/java/org/springframework/ai/vectorstore/qdrant/QdrantVectorStore.java
@@ -58,6 +58,7 @@ import static io.qdrant.client.WithPayloadSelectorFactory.enable;
 public class QdrantVectorStore implements VectorStore, ApplicationListener<ApplicationReadyEvent> {
 
 	private static final String CONTENT_FIELD_NAME = "doc_content";
+
 	private static final String DISTANCE_FIELD_NAME = "distance";
 
 	public static final String DEFAULT_COLLECTION_NAME = "vector_store";
@@ -89,6 +90,7 @@ public class QdrantVectorStore implements VectorStore, ApplicationListener<Appli
 		private QdrantVectorStoreConfig(Builder builder) {
 			this.collectionName = builder.collectionName;
 		}
+
 		/**
 		 * Start building a new configuration.
 		 * @return The entry point for creating a new configuration.
@@ -127,10 +129,10 @@ public class QdrantVectorStore implements VectorStore, ApplicationListener<Appli
 				return new QdrantVectorStoreConfig(this);
 			}
 
-
 		}
 
 	}
+
 	/**
 	 * Constructs a new QdrantVectorStore.
 	 * @param config The configuration for the store.
@@ -165,16 +167,16 @@ public class QdrantVectorStore implements VectorStore, ApplicationListener<Appli
 		// Create the collection if it does not exist.
 		if (!isCollectionExists()) {
 			var vectorParams = VectorParams.newBuilder()
-					.setDistance(Distance.Cosine)
-					.setSize(this.embeddingClient.dimensions())
-					.build();
-            try {
-                this.qdrantClient.createCollectionAsync(this.collectionName, vectorParams).get();
-            } //
-			catch (Exception  e) {
-                throw new RuntimeException(e);
-            }
-        }
+				.setDistance(Distance.Cosine)
+				.setSize(this.embeddingClient.dimensions())
+				.build();
+			try {
+				this.qdrantClient.createCollectionAsync(this.collectionName, vectorParams).get();
+			} //
+			catch (Exception e) {
+				throw new RuntimeException(e);
+			}
+		}
 	}
 
 	/**
@@ -303,7 +305,6 @@ public class QdrantVectorStore implements VectorStore, ApplicationListener<Appli
 		return doubleList.stream().map(d -> d.floatValue()).toList();
 	}
 
-
 	private boolean isCollectionExists() {
 		try {
 			return this.qdrantClient.listCollectionsAsync().get().stream().anyMatch(c -> c.equals(this.collectionName));