Skip to content

Commit 1d9a898

Browse files
committed
Fixing formatting violations
1 parent 0a1158d commit 1d9a898

File tree

1 file changed

+15
-26
lines changed

1 file changed

+15
-26
lines changed

vector-stores/spring-ai-mongodb-atlas-store/src/test/java/org/springframework/ai/vectorstore/mongodb/atlas/MongoDBAtlasVectorStoreIT.java

Lines changed: 15 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -272,10 +272,10 @@ void deleteByFilter() {
272272
Thread.sleep(5000); // Wait for indexing
273273

274274
SearchRequest searchRequest = SearchRequest.builder()
275-
.query("The World")
276-
.topK(5)
277-
.similarityThresholdAll()
278-
.build();
275+
.query("The World")
276+
.topK(5)
277+
.similarityThresholdAll()
278+
.build();
279279

280280
List<Document> results = vectorStore.similaritySearch(searchRequest);
281281
assertThat(results).hasSize(3);
@@ -307,11 +307,7 @@ void deleteWithStringFilterExpression() {
307307
vectorStore.add(List.of(bgDocument, nlDocument, bgDocument2));
308308
Thread.sleep(5000); // Wait for indexing
309309

310-
var searchRequest = SearchRequest.builder()
311-
.query("The World")
312-
.topK(5)
313-
.similarityThresholdAll()
314-
.build();
310+
var searchRequest = SearchRequest.builder().query("The World").topK(5).similarityThresholdAll().build();
315311

316312
List<Document> results = vectorStore.similaritySearch(searchRequest);
317313
assertThat(results).hasSize(3);
@@ -340,29 +336,22 @@ void deleteWithComplexFilterExpression() {
340336
// Complex filter expression: (type == 'A' AND priority > 1)
341337
Filter.Expression priorityFilter = new Filter.Expression(Filter.ExpressionType.GT,
342338
new Filter.Key("priority"), new Filter.Value(1));
343-
Filter.Expression typeFilter = new Filter.Expression(Filter.ExpressionType.EQ,
344-
new Filter.Key("type"), new Filter.Value("A"));
345-
Filter.Expression complexFilter = new Filter.Expression(Filter.ExpressionType.AND,
346-
typeFilter, priorityFilter);
339+
Filter.Expression typeFilter = new Filter.Expression(Filter.ExpressionType.EQ, new Filter.Key("type"),
340+
new Filter.Value("A"));
341+
Filter.Expression complexFilter = new Filter.Expression(Filter.ExpressionType.AND, typeFilter,
342+
priorityFilter);
347343

348344
vectorStore.delete(complexFilter);
349345
Thread.sleep(1000); // Wait for deletion to be processed
350346

351-
var results = vectorStore.similaritySearch(SearchRequest.builder()
352-
.query("Content")
353-
.topK(5)
354-
.similarityThresholdAll()
355-
.build());
347+
var results = vectorStore
348+
.similaritySearch(SearchRequest.builder().query("Content").topK(5).similarityThresholdAll().build());
356349

357350
assertThat(results).hasSize(2);
358-
assertThat(results.stream()
359-
.map(doc -> doc.getMetadata().get("type"))
360-
.collect(Collectors.toList()))
361-
.containsExactlyInAnyOrder("A", "B");
362-
assertThat(results.stream()
363-
.map(doc -> doc.getMetadata().get("priority"))
364-
.collect(Collectors.toList()))
365-
.containsExactlyInAnyOrder(1, 1);
351+
assertThat(results.stream().map(doc -> doc.getMetadata().get("type")).collect(Collectors.toList()))
352+
.containsExactlyInAnyOrder("A", "B");
353+
assertThat(results.stream().map(doc -> doc.getMetadata().get("priority")).collect(Collectors.toList()))
354+
.containsExactlyInAnyOrder(1, 1);
366355
});
367356
}
368357

0 commit comments

Comments
 (0)