Skip to content

Commit

Permalink
nit
Browse files Browse the repository at this point in the history
Signed-off-by: zhichao-aws <[email protected]>
  • Loading branch information
zhichao-aws committed Feb 29, 2024
1 parent 1c341a7 commit 1189b4e
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -8,20 +8,19 @@
* Query builders which calls ml-commons API to do model inference.
* The model inference result is used for search on target field.
*/

public interface ModelInferenceQueryBuilder {
/**
* Get the model id used by ml-commons model inference. Return null if the model id is absent.
*/
public String modelId();

/**
* Set a new model id for the query builder.
*/
* Set a new model id for the query builder.
*/
public ModelInferenceQueryBuilder modelId(String modelId);

/**
* Get the field name for search.
*/
* Get the field name for search.
*/
public String fieldName();
}
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,11 @@
import org.opensearch.neuralsearch.BaseNeuralSearchIT;
import org.opensearch.neuralsearch.query.HybridQueryBuilder;
import org.opensearch.neuralsearch.query.NeuralQueryBuilder;
import org.opensearch.neuralsearch.query.NeuralSparseQueryBuilder;

import com.google.common.primitives.Floats;

import lombok.SneakyThrows;
import org.opensearch.neuralsearch.query.NeuralSparseQueryBuilder;

public class NeuralQueryEnricherProcessorIT extends BaseNeuralSearchIT {

Expand Down

0 comments on commit 1189b4e

Please sign in to comment.