feat: Allow fieldname as parameter (#215)

Co-authored-by: Alexander Reelsen <[email protected]>

Author: spinscale

src/queries/vector-queries/sparse-vector-query.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const { Query } = require('../../core');
+const { isNil } = require('lodash');
 
 /**
  * The sparse vector query executes a query consisting of sparse vectors, such as built by a learned sparse retrieval model,
@@ -15,8 +16,10 @@ const { Query } = require('../../core');
  */
 class SparseVectorQuery extends Query {
     // eslint-disable-next-line require-jsdoc
-    constructor() {
+
+    constructor(field) {
         super('sparse_vector');
+        if (!isNil(field)) this._queryOpts.field = field;
     }
 
     /**

test/queries-test/sparse-vector-query.test.js

@@ -1,5 +1,5 @@
 import test from 'ava';
-import { SparseVectorQuery } from '../../src';
+import esb, { SparseVectorQuery } from '../../src';
 
 test('with inference id and query', t => {
     const q = new SparseVectorQuery();
@@ -136,3 +136,15 @@ test('with pruning config for multiple elements', t => {
     };
     t.deepEqual(q.toJSON(), expected);
 });
+
+test('call sparse vector query via esb', t => {
+    const q = esb.sparseVectorQuery('my_field').query('my query');
+
+    const expected = {
+        sparse_vector: {
+            field: 'my_field',
+            query: 'my query'
+        }
+    };
+    t.deepEqual(q.toJSON(), expected);
+});