Update knn search and query template autocomplete (elastic#207187)

(cherry picked from commit 1dc2aca)
kibanamachine · Jan 20, 2025 · c9eda11 · c9eda11
1 parent 2d9d9c6
commit c9eda11
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 3 deletions.
diff --git a/src/platform/plugins/shared/console/server/lib/spec_definitions/js/query/dsl.ts b/src/platform/plugins/shared/console/server/lib/spec_definitions/js/query/dsl.ts
@@ -382,6 +382,35 @@ export const query = (specService: SpecDefinitionsService) => {
       _scope: '',
       query: {},
     },
+    knn: {
+      __template: {
+        field: '',
+        k: 10,
+        num_candidates: 100,
+        query_vector_builder: {
+          text_embedding: {
+            model_id: '',
+            model_text: '',
+          },
+        },
+      },
+      field: '{field}',
+      filter: { __scope_link: 'GLOBAL.filter' },
+      k: 10,
+      num_candidates: 100,
+      query_vector: [],
+      query_vector_builder: {
+        text_embedding: {
+          model_id: '',
+          model_text: '',
+        },
+      },
+      rescore_vector: {
+        oversample: 1.5,
+      },
+      similarity: { __one_of: ['l2_norm', 'cosine', 'dot_product', 'max_inner_product'] },
+      boost: 1.0,
+    },
     match_all: {
       boost: 1,
     },

diff --git a/src/platform/plugins/shared/console/server/lib/spec_definitions/js/search.ts b/src/platform/plugins/shared/console/server/lib/spec_definitions/js/search.ts
@@ -226,6 +226,12 @@ export const search = (specService: SpecDefinitionsService) => {
       knn: {
         __template: {
           field: '',
+          query_vector_builder: {
+            text_embedding: {
+              model_id: '',
+              model_text: '',
+            },
+          },
           k: 10,
           num_candidates: 100,
         },
@@ -236,9 +242,17 @@ export const search = (specService: SpecDefinitionsService) => {
             k: 10,
             num_candidates: 100,
             query_vector: [],
-            query_vector_builder: {},
-            similarity: { __one_of: ['l2_norm', 'cosine', 'dot_product'] },
+            query_vector_builder: {
+              text_embedding: {
+                model_id: '',
+                model_text: '',
+              },
+            },
+            similarity: { __one_of: ['l2_norm', 'cosine', 'dot_product', 'max_inner_product'] },
             boost: 1.0,
+            rescore_vector: {
+              oversample: 1.5,
+            },
           },
           [
             {
@@ -248,8 +262,11 @@ export const search = (specService: SpecDefinitionsService) => {
               num_candidates: 100,
               query_vector: [],
               query_vector_builder: {},
-              similarity: { __one_of: ['l2_norm', 'cosine', 'dot_product'] },
+              similarity: { __one_of: ['l2_norm', 'cosine', 'dot_product', 'max_inner_product'] },
               boost: 1.0,
+              rescore_vector: {
+                oversample: 1.5,
+              },
             },
           ],
         ],