diff --git a/src/main/java/org/opensearch/neuralsearch/query/ModelInferenceQueryBuilder.java b/src/main/java/org/opensearch/neuralsearch/query/ModelInferenceQueryBuilder.java index a1001c455..23db567d4 100644 --- a/src/main/java/org/opensearch/neuralsearch/query/ModelInferenceQueryBuilder.java +++ b/src/main/java/org/opensearch/neuralsearch/query/ModelInferenceQueryBuilder.java @@ -8,7 +8,6 @@ * 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. @@ -16,12 +15,12 @@ public interface ModelInferenceQueryBuilder { 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(); } diff --git a/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryEnricherProcessorIT.java b/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryEnricherProcessorIT.java index 74d0912d6..0024eb8c5 100644 --- a/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryEnricherProcessorIT.java +++ b/src/test/java/org/opensearch/neuralsearch/processor/NeuralQueryEnricherProcessorIT.java @@ -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 {