Skip to content

Commit 5c483b0

Browse files
markpollackleijendary
authored andcommitted
Update class level javadoc in some vector stores to match builder signatures
Signed-off-by: leijendary <[email protected]>
1 parent 11801d7 commit 5c483b0

File tree

11 files changed

+24
-55
lines changed

11 files changed

+24
-55
lines changed

vector-stores/spring-ai-cassandra-store/src/main/java/org/springframework/ai/vectorstore/cassandra/CassandraVectorStore.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,8 @@
104104
* Basic usage example:
105105
* </p>
106106
* <pre>{@code
107-
* CassandraVectorStore vectorStore = CassandraVectorStore.builder()
107+
* CassandraVectorStore vectorStore = CassandraVectorStore.builder(embeddingModel)
108108
* .session(cqlSession)
109-
* .embeddingModel(embeddingModel)
110109
* .keyspace("my_keyspace")
111110
* .table("my_vectors")
112111
* .build();
@@ -130,9 +129,8 @@
130129
* Advanced configuration example:
131130
* </p>
132131
* <pre>{@code
133-
* CassandraVectorStore vectorStore = CassandraVectorStore.builder()
132+
* CassandraVectorStore vectorStore = CassandraVectorStore.builder(embeddingModel)
134133
* .session(cqlSession)
135-
* .embeddingModel(embeddingModel)
136134
* .keyspace("my_keyspace")
137135
* .table("my_vectors")
138136
* .partitionKeys(List.of(new SchemaColumn("id", DataTypes.TEXT)))

vector-stores/spring-ai-elasticsearch-store/src/main/java/org/springframework/ai/vectorstore/elasticsearch/ElasticsearchVectorStore.java

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,7 @@
8383
* Basic usage example:
8484
* </p>
8585
* <pre>{@code
86-
* ElasticsearchVectorStore vectorStore = ElasticsearchVectorStore.builder()
87-
* .restClient(restClient)
88-
* .embeddingModel(embeddingModel)
86+
* ElasticsearchVectorStore vectorStore = ElasticsearchVectorStore.builder(restClient, embeddingModel)
8987
* .initializeSchema(true)
9088
* .build();
9189
*
@@ -113,9 +111,7 @@
113111
* options.setSimilarity(SimilarityFunction.dot_product);
114112
* options.setDimensions(1536);
115113
*
116-
* ElasticsearchVectorStore vectorStore = ElasticsearchVectorStore.builder()
117-
* .restClient(restClient)
118-
* .embeddingModel(embeddingModel)
114+
* ElasticsearchVectorStore vectorStore = ElasticsearchVectorStore.builder(restClient, embeddingModel)
119115
* .options(options)
120116
* .initializeSchema(true)
121117
* .batchingStrategy(new TokenCountBatchingStrategy())

vector-stores/spring-ai-mariadb-store/src/main/java/org/springframework/ai/vectorstore/mariadb/MariaDBVectorStore.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
import org.springframework.ai.embedding.TokenCountBatchingStrategy;
3434
import org.springframework.ai.observation.conventions.VectorStoreProvider;
3535
import org.springframework.ai.observation.conventions.VectorStoreSimilarityMetric;
36+
import org.springframework.ai.observation.conventions.VectorStoreSimilarityMetric;
3637
import org.springframework.ai.util.JacksonUtils;
3738
import org.springframework.ai.vectorstore.AbstractVectorStoreBuilder;
3839
import org.springframework.ai.vectorstore.SearchRequest;
@@ -73,8 +74,7 @@
7374
* Basic usage example:
7475
* </p>
7576
* <pre>{@code
76-
* MariaDBVectorStore vectorStore = MariaDBVectorStore.builder(jdbcTemplate)
77-
* .embeddingModel(embeddingModel)
77+
* MariaDBVectorStore vectorStore = MariaDBVectorStore.builder(jdbcTemplate, embeddingModel)
7878
* .initializeSchema(true)
7979
* .build();
8080
*
@@ -97,8 +97,7 @@
9797
* Advanced configuration example:
9898
* </p>
9999
* <pre>{@code
100-
* MariaDBVectorStore vectorStore = MariaDBVectorStore.builder(jdbcTemplate)
101-
* .embeddingModel(embeddingModel)
100+
* MariaDBVectorStore vectorStore = MariaDBVectorStore.builder(jdbcTemplate, embeddingModel)
102101
* .schemaName("mydb")
103102
* .distanceType(MariaDBDistanceType.COSINE)
104103
* .dimensions(1536)
@@ -295,7 +294,8 @@ protected MariaDBVectorStore(MariaDBBuilder builder) {
295294
this.vectorTableName = builder.vectorTableName.isEmpty() ? DEFAULT_TABLE_NAME
296295
: MariaDBSchemaValidator.validateAndEnquoteIdentifier(builder.vectorTableName.trim(), false);
297296

298-
logger.info("Using the vector table name: {}. Is empty: {}", this.vectorTableName, vectorTableName.isEmpty());
297+
logger.info("Using the vector table name: {}. Is empty: {}", this.vectorTableName,
298+
builder.vectorTableName.isEmpty());
299299

300300
this.schemaName = builder.schemaName == null ? null
301301
: MariaDBSchemaValidator.validateAndEnquoteIdentifier(builder.schemaName, false);

vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/milvus/vectorstore/MilvusVectorStore.java

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -92,16 +92,12 @@
9292
* <p>
9393
* Example usage: <pre>{@code
9494
* // Create a basic Milvus vector store
95-
* MilvusVectorStore vectorStore = MilvusVectorStore.builder()
96-
* .milvusClient(milvusClient)
97-
* .embeddingModel(embeddingModel)
95+
* MilvusVectorStore vectorStore = MilvusVectorStore.builder(milvusServiceClient, embeddingModel)
9896
* .initializeSchema(true)
9997
* .build();
10098
*
10199
* // Create a customized Milvus vector store
102-
* MilvusVectorStore customVectorStore = MilvusVectorStore.builder()
103-
* .milvusClient(milvusClient)
104-
* .embeddingModel(embeddingModel)
100+
* MilvusVectorStore customVectorStore = MilvusVectorStore.builder(milvusServiceClient, embeddingModel)
105101
* .databaseName("my_database")
106102
* .collectionName("my_collection")
107103
* .metricType(MetricType.COSINE)
@@ -267,8 +263,8 @@ protected MilvusVectorStore(Builder builder) {
267263
* recommended way to instantiate a MilvusBuilder.
268264
* @return a new MilvusBuilder instance
269265
*/
270-
public static Builder builder(MilvusServiceClient milvusClient, EmbeddingModel embeddingModel) {
271-
return new Builder(milvusClient, embeddingModel);
266+
public static Builder builder(MilvusServiceClient milvusServiceClient, EmbeddingModel embeddingModel) {
267+
return new Builder(milvusServiceClient, embeddingModel);
272268
}
273269

274270
@Override

vector-stores/spring-ai-mongodb-atlas-store/src/main/java/org/springframework/ai/vectorstore/mongodb/atlas/MongoDBAtlasVectorStore.java

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,7 @@
7272
* Basic usage example:
7373
* </p>
7474
* <pre>{@code
75-
* MongoDBAtlasVectorStore vectorStore = MongoDBAtlasVectorStore.builder()
76-
* .mongoTemplate(mongoTemplate)
77-
* .embeddingModel(embeddingModel)
75+
* MongoDBAtlasVectorStore vectorStore = MongoDBAtlasVectorStore.builder(mongoTemplate, embeddingModel)
7876
* .collectionName("vector_store")
7977
* .initializeSchema(true)
8078
* .build();
@@ -98,9 +96,7 @@
9896
* Advanced configuration example:
9997
* </p>
10098
* <pre>{@code
101-
* MongoDBAtlasVectorStore vectorStore = MongoDBAtlasVectorStore.builder()
102-
* .mongoTemplate(mongoTemplate)
103-
* .embeddingModel(embeddingModel)
99+
* MongoDBAtlasVectorStore vectorStore = MongoDBAtlasVectorStore.builder(mongoTemplate, embeddingModel)
104100
* .collectionName("custom_vectors")
105101
* .vectorIndexName("custom_vector_index")
106102
* .pathName("custom_embedding")

vector-stores/spring-ai-neo4j-store/src/main/java/org/springframework/ai/vectorstore/neo4j/Neo4jVectorStore.java

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,7 @@
7272
* Basic usage example:
7373
* </p>
7474
* <pre>{@code
75-
* Neo4jVectorStore vectorStore = Neo4jVectorStore.builder()
76-
* .driver(driver)
77-
* .embeddingModel(embeddingModel)
75+
* Neo4jVectorStore vectorStore = Neo4jVectorStore.builder(driver, embeddingModel)
7876
* .initializeSchema(true)
7977
* .build();
8078
*
@@ -97,9 +95,7 @@
9795
* Advanced configuration example:
9896
* </p>
9997
* <pre>{@code
100-
* Neo4jVectorStore vectorStore = Neo4jVectorStore.builder()
101-
* .driver(driver)
102-
* .embeddingModel(embeddingModel)
98+
* Neo4jVectorStore vectorStore = Neo4jVectorStore.builder(driver, embeddingModel)
10399
* .databaseName("neo4j")
104100
* .distanceType(Neo4jDistanceType.COSINE)
105101
* .dimensions(1536)

vector-stores/spring-ai-opensearch-store/src/main/java/org/springframework/ai/vectorstore/opensearch/OpenSearchVectorStore.java

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -82,9 +82,7 @@
8282
* Basic usage example:
8383
* </p>
8484
* <pre>{@code
85-
* OpenSearchVectorStore vectorStore = OpenSearchVectorStore.builder()
86-
* .openSearchClient(openSearchClient)
87-
* .embeddingModel(embeddingModel)
85+
* OpenSearchVectorStore vectorStore = OpenSearchVectorStore.builder(openSearchClient, embeddingModel)
8886
* .initializeSchema(true)
8987
* .build();
9088
*
@@ -107,9 +105,7 @@
107105
* Advanced configuration example:
108106
* </p>
109107
* <pre>{@code
110-
* OpenSearchVectorStore vectorStore = OpenSearchVectorStore.builder()
111-
* .openSearchClient(openSearchClient)
112-
* .embeddingModel(embeddingModel)
108+
* OpenSearchVectorStore vectorStore = OpenSearchVectorStore.builder(openSearchClient, embeddingModel)
113109
* .index("custom-index")
114110
* .mappingJson(customMapping)
115111
* .similarityFunction("l2")

vector-stores/spring-ai-pgvector-store/src/main/java/org/springframework/ai/vectorstore/pgvector/PgVectorStore.java

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,7 @@
8585
* Basic usage example:
8686
* </p>
8787
* <pre>{@code
88-
* PgVectorStore vectorStore = PgVectorStore.builder()
89-
* .jdbcTemplate(jdbcTemplate)
90-
* .embeddingModel(embeddingModel)
88+
* PgVectorStore vectorStore = PgVectorStore.builder(jdbcTemplate, embeddingModel)
9189
* .dimensions(1536) // Optional: defaults to model dimensions or 1536
9290
* .distanceType(PgDistanceType.COSINE_DISTANCE)
9391
* .indexType(PgIndexType.HNSW)
@@ -112,9 +110,7 @@
112110
* Advanced configuration example:
113111
* </p>
114112
* <pre>{@code
115-
* PgVectorStore vectorStore = PgVectorStore.builder()
116-
* .jdbcTemplate(jdbcTemplate)
117-
* .embeddingModel(embeddingModel)
113+
* PgVectorStore vectorStore = PgVectorStore.builder(jdbcTemplate, embeddingModel)
118114
* .schemaName("custom_schema")
119115
* .vectorTableName("custom_vectors")
120116
* .distanceType(PgDistanceType.NEGATIVE_INNER_PRODUCT)

vector-stores/spring-ai-qdrant-store/src/main/java/org/springframework/ai/vectorstore/qdrant/QdrantVectorStore.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,7 @@
101101
* Advanced configuration example:
102102
* </p>
103103
* <pre>{@code
104-
* QdrantVectorStore vectorStore = QdrantVectorStore.builder(qdrantClient)
105-
* .embeddingModel(embeddingModel)
104+
* QdrantVectorStore vectorStore = QdrantVectorStore.builder(qdrantClient, embeddingModel)
106105
* .collectionName("custom-collection")
107106
* .initializeSchema(true)
108107
* .batchingStrategy(new TokenCountBatchingStrategy())

vector-stores/spring-ai-redis-store/src/main/java/org/springframework/ai/vectorstore/redis/RedisVectorStore.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -102,9 +102,7 @@
102102
* Basic usage example:
103103
* </p>
104104
* <pre>{@code
105-
* RedisVectorStore vectorStore = RedisVectorStore.builder()
106-
* .jedis(jedisPooled)
107-
* .embeddingModel(embeddingModel)
105+
* RedisVectorStore vectorStore = RedisVectorStore.builder(jedisPooled, embeddingModel)
108106
* .indexName("custom-index") // Optional: defaults to "spring-ai-index"
109107
* .prefix("custom-prefix") // Optional: defaults to "embedding:"
110108
* .vectorAlgorithm(Algorithm.HNSW)

0 commit comments

Comments
 (0)