Add vmsfields config for nearest neighbor search on supported tensor fields.

2 files changed, 55 insertions, 0 deletions

diff --git a/config-model/src/test/derived/nearestneighbor_streaming/test.sd b/config-model/src/test/derived/nearestneighbor_streaming/test.sd
new file mode 100644
index 00000000000..4427fa08ab6
--- /dev/null
+++ b/config-model/src/test/derived/nearestneighbor_streaming/test.sd
@@ -0,0 +1,24 @@
+# Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+schema test {
+  document test {
+    field vec_a type tensor<float>(x[16]) {
+      indexing: attribute
+    }
+    field vec_b type tensor<float>(x[16]) {
+      indexing: attribute
+      attribute {
+        distance-metric: angular
+      }
+    }
+    field vec_c type tensor<float>(m{},x[16]) {
+      indexing: attribute
+      attribute {
+        distance-metric: innerproduct
+      }
+    }
+    # This tensor field can not be used with nearest neighbor search.
+    field vec_d type tensor<float>(x{}) {
+      indexing: attribute
+    }
+  }
+}
diff --git a/config-model/src/test/derived/nearestneighbor_streaming/vsmfields.cfg b/config-model/src/test/derived/nearestneighbor_streaming/vsmfields.cfg
new file mode 100644
index 00000000000..f8b1cf62048
--- /dev/null
+++ b/config-model/src/test/derived/nearestneighbor_streaming/vsmfields.cfg
@@ -0,0 +1,31 @@
+documentverificationlevel 0
+searchall 1
+fieldspec[].name "vec_a"
+fieldspec[].searchmethod NEAREST_NEIGHBOR
+fieldspec[].arg1 "EUCLIDEAN"
+fieldspec[].maxlength 1048576
+fieldspec[].fieldtype ATTRIBUTE
+fieldspec[].name "vec_b"
+fieldspec[].searchmethod NEAREST_NEIGHBOR
+fieldspec[].arg1 "ANGULAR"
+fieldspec[].maxlength 1048576
+fieldspec[].fieldtype ATTRIBUTE
+fieldspec[].name "vec_c"
+fieldspec[].searchmethod NEAREST_NEIGHBOR
+fieldspec[].arg1 "INNERPRODUCT"
+fieldspec[].maxlength 1048576
+fieldspec[].fieldtype ATTRIBUTE
+fieldspec[].name "vec_d"
+fieldspec[].searchmethod NONE
+fieldspec[].arg1 ""
+fieldspec[].maxlength 1048576
+fieldspec[].fieldtype ATTRIBUTE
+documenttype[].name "test"
+documenttype[].index[].name "vec_a"
+documenttype[].index[].field[].name "vec_a"
+documenttype[].index[].name "vec_b"
+documenttype[].index[].field[].name "vec_b"
+documenttype[].index[].name "vec_c"
+documenttype[].index[].field[].name "vec_c"
+documenttype[].index[].name "vec_d"
+documenttype[].index[].field[].name "vec_d"