3030
3131import org .springframework .ai .document .Document ;
3232import org .springframework .ai .document .DocumentMetadata ;
33- import org .springframework .ai .embedding .BatchingStrategy ;
3433import org .springframework .ai .embedding .EmbeddingModel ;
3534import org .springframework .ai .embedding .EmbeddingOptionsBuilder ;
36- import org .springframework .ai .embedding .TokenCountBatchingStrategy ;
3735import org .springframework .ai .observation .conventions .VectorStoreProvider ;
3836import org .springframework .ai .observation .conventions .VectorStoreSimilarityMetric ;
3937import org .springframework .ai .vectorstore .AbstractVectorStoreBuilder ;
@@ -180,8 +178,6 @@ public class Neo4jVectorStore extends AbstractObservationVectorStore implements
180178
181179 private final boolean initializeSchema ;
182180
183- private final BatchingStrategy batchingStrategy ;
184-
185181 protected Neo4jVectorStore (Builder builder ) {
186182 super (builder );
187183
@@ -198,7 +194,6 @@ protected Neo4jVectorStore(Builder builder) {
198194 this .idProperty = SchemaNames .sanitize (builder .idProperty ).orElseThrow ();
199195 this .constraintName = SchemaNames .sanitize (builder .constraintName ).orElseThrow ();
200196 this .initializeSchema = builder .initializeSchema ;
201- this .batchingStrategy = new TokenCountBatchingStrategy ();
202197 }
203198
204199 @ Override
@@ -418,8 +413,6 @@ public static class Builder extends AbstractVectorStoreBuilder<Builder> {
418413
419414 private String constraintName = DEFAULT_CONSTRAINT_NAME ;
420415
421- private BatchingStrategy batchingStrategy = new TokenCountBatchingStrategy ();
422-
423416 private boolean initializeSchema = false ;
424417
425418 private Builder (Driver driver , EmbeddingModel embeddingModel ) {
@@ -545,11 +538,6 @@ public Builder initializeSchema(boolean initializeSchema) {
545538 return this ;
546539 }
547540
548- public Builder batchingStrategy (BatchingStrategy batchingStrategy ) {
549- this .batchingStrategy = batchingStrategy ;
550- return this ;
551- }
552-
553541 @ Override
554542 public Neo4jVectorStore build () {
555543 return new Neo4jVectorStore (this );
0 commit comments