test fixes

mongodb · Sep 25, 2024 · 6065df2 · 6065df2
1 parent 6da1001
commit 6065df2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/examples/src/test/kotlin/AggregatesBuilderTest.kt b/examples/src/test/kotlin/AggregatesBuilderTest.kt
@@ -28,7 +28,8 @@ import com.mongodb.client.model.geojson.Position
 import com.mongodb.client.model.search.SearchOperator
 import com.mongodb.client.model.search.SearchOptions
 import com.mongodb.client.model.search.SearchPath
-import com.mongodb.client.model.search.VectorSearchOptions.vectorSearchOptions
+import com.mongodb.client.model.search.VectorSearchOptions
+import com.mongodb.client.model.search.VectorSearchOptions.exactVectorSearchOptions
 import com.mongodb.kotlin.client.coroutine.MongoClient
 import config.getConfig
 import kotlinx.coroutines.flow.firstOrNull
@@ -981,13 +982,13 @@ class AggregatesBuilderTest {
                     SearchPath.fieldPath(MovieAlt::plotEmbedding.name),
                     listOf(-0.0072121937, -0.030757688, -0.012945653),
                     "mflix_movies_embedding_index",
-                    2.toLong(),
                     1.toLong(),
-                    vectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016))
+                    exactVectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016))
                 )
                 // :snippet-end:
             )
         )
+
         val results = resultsFlow.toList()
         assertEquals(1, resultsFlow.toList().size)
         assertEquals(1, results.first().get("count", Document::class.java).get("lowerBound", java.lang.Long::class.java)?.toInt())

diff --git a/examples/src/test/kotlin/SearchIndexesTest.kt b/examples/src/test/kotlin/SearchIndexesTest.kt
@@ -1,5 +1,6 @@
 
 import com.mongodb.client.model.SearchIndexModel
+import com.mongodb.client.model.SearchIndexType
 import com.mongodb.kotlin.client.coroutine.MongoClient
 import config.getConfig
 import kotlinx.coroutines.flow.toList

diff --git a/source/examples/generated/AggregatesBuilderTest.snippet.vector-search.kt b/source/examples/generated/AggregatesBuilderTest.snippet.vector-search.kt
@@ -2,7 +2,6 @@ Aggregates.vectorSearch(
     SearchPath.fieldPath(MovieAlt::plotEmbedding.name),
     listOf(-0.0072121937, -0.030757688, -0.012945653),
     "mflix_movies_embedding_index",
-    2.toLong(),
     1.toLong(),
-    vectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016))
+    exactVectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016))
 )