another example

firebase · Aug 8, 2024 · d9eea7a · d9eea7a
1 parent 07a6409
commit d9eea7a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/Firestore/Swift/Tests/Integration/VectorIntegrationTests.swift b/Firestore/Swift/Tests/Integration/VectorIntegrationTests.swift
@@ -34,6 +34,21 @@ class VectorIntegrationTests: FSTIntegrationTestCase {
     try await vectorQuery.getDocuments(source: VectorSource.server)
   }
 
+  func exampleFindNearestWithOptions() async throws {
+    let collection = collectionRef()
+
+    let vectorQuery = collection.findNearest(
+      fieldPath: "embedding",
+      queryVector: [1.0, 2.0, 3.0],
+      limit: 10,
+      distanceMeasure: FirestoreDistanceMeasure.cosine,
+      options: FindNearestOptions().withDistanceResultFieldPath("distance")
+        .withDistanceThreshold(0.5)
+    )
+
+    try await vectorQuery.getDocuments(source: VectorSource.server)
+  }
+
   func testWriteAndReadVectorEmbeddings() async throws {
     let collection = collectionRef()