30
30
31
31
import org .springframework .ai .document .Document ;
32
32
import org .springframework .ai .document .DocumentMetadata ;
33
- import org .springframework .ai .embedding .BatchingStrategy ;
34
33
import org .springframework .ai .embedding .EmbeddingModel ;
35
34
import org .springframework .ai .embedding .EmbeddingOptionsBuilder ;
36
- import org .springframework .ai .embedding .TokenCountBatchingStrategy ;
37
35
import org .springframework .ai .observation .conventions .VectorStoreProvider ;
38
36
import org .springframework .ai .observation .conventions .VectorStoreSimilarityMetric ;
39
37
import org .springframework .ai .vectorstore .AbstractVectorStoreBuilder ;
@@ -180,8 +178,6 @@ public class Neo4jVectorStore extends AbstractObservationVectorStore implements
180
178
181
179
private final boolean initializeSchema ;
182
180
183
- private final BatchingStrategy batchingStrategy ;
184
-
185
181
protected Neo4jVectorStore (Builder builder ) {
186
182
super (builder );
187
183
@@ -198,7 +194,6 @@ protected Neo4jVectorStore(Builder builder) {
198
194
this .idProperty = SchemaNames .sanitize (builder .idProperty ).orElseThrow ();
199
195
this .constraintName = SchemaNames .sanitize (builder .constraintName ).orElseThrow ();
200
196
this .initializeSchema = builder .initializeSchema ;
201
- this .batchingStrategy = new TokenCountBatchingStrategy ();
202
197
}
203
198
204
199
@ Override
@@ -418,8 +413,6 @@ public static class Builder extends AbstractVectorStoreBuilder<Builder> {
418
413
419
414
private String constraintName = DEFAULT_CONSTRAINT_NAME ;
420
415
421
- private BatchingStrategy batchingStrategy = new TokenCountBatchingStrategy ();
422
-
423
416
private boolean initializeSchema = false ;
424
417
425
418
private Builder (Driver driver , EmbeddingModel embeddingModel ) {
@@ -545,11 +538,6 @@ public Builder initializeSchema(boolean initializeSchema) {
545
538
return this ;
546
539
}
547
540
548
- public Builder batchingStrategy (BatchingStrategy batchingStrategy ) {
549
- this .batchingStrategy = batchingStrategy ;
550
- return this ;
551
- }
552
-
553
541
@ Override
554
542
public Neo4jVectorStore build () {
555
543
return new Neo4jVectorStore (this );
0 commit comments