Skip to content

Commit de23341

Browse files
committed
apply spotless
Signed-off-by: Brian Flores <iflorbri@amazon.com>
1 parent f7d0f24 commit de23341

File tree

1 file changed

+19
-28
lines changed

1 file changed

+19
-28
lines changed

plugin/src/test/java/org/opensearch/ml/processor/MLInferenceSearchRequestProcessorTests.java

Lines changed: 19 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -1629,14 +1629,9 @@ public void testExecute_rewriteTermQueryWithNewQuerySuccess() throws Exception {
16291629

16301630
String queryTemplate = "${llm_query}";
16311631

1632-
String llmQuery = "{\n" +
1633-
" \"query\": {\n" +
1634-
" \"match_all\": {}\n" +
1635-
" }\n" +
1636-
"}";
1632+
String llmQuery = "{\n" + " \"query\": {\n" + " \"match_all\": {}\n" + " }\n" + "}";
16371633
Map content = Map.of("content", List.of(Map.of("text", llmQuery)));
16381634

1639-
16401635
List<Map<String, String>> optionalInputMap = new ArrayList<>();
16411636
Map<String, String> input = new HashMap<>();
16421637
input.put(modelInputField, originalQueryField);
@@ -1648,23 +1643,23 @@ public void testExecute_rewriteTermQueryWithNewQuerySuccess() throws Exception {
16481643
optionalOutputMap.add(output);
16491644

16501645
MLInferenceSearchRequestProcessor requestProcessor = new MLInferenceSearchRequestProcessor(
1651-
"model1",
1652-
queryTemplate,
1653-
null,
1654-
null,
1655-
optionalInputMap,
1656-
optionalOutputMap,
1657-
null,
1658-
DEFAULT_MAX_PREDICTION_TASKS,
1659-
PROCESSOR_TAG,
1660-
DESCRIPTION,
1661-
false,
1662-
"remote",
1663-
true,
1664-
false,
1665-
"{ \"parameters\": ${ml_inference.parameters} }",
1666-
client,
1667-
TEST_XCONTENT_REGISTRY_FOR_QUERY
1646+
"model1",
1647+
queryTemplate,
1648+
null,
1649+
null,
1650+
optionalInputMap,
1651+
optionalOutputMap,
1652+
null,
1653+
DEFAULT_MAX_PREDICTION_TASKS,
1654+
PROCESSOR_TAG,
1655+
DESCRIPTION,
1656+
false,
1657+
"remote",
1658+
true,
1659+
false,
1660+
"{ \"parameters\": ${ml_inference.parameters} }",
1661+
client,
1662+
TEST_XCONTENT_REGISTRY_FOR_QUERY
16681663
);
16691664

16701665
/**
@@ -1680,11 +1675,7 @@ public void testExecute_rewriteTermQueryWithNewQuerySuccess() throws Exception {
16801675
* } ]
16811676
* }
16821677
*/
1683-
ModelTensor modelTensor = ModelTensor
1684-
.builder()
1685-
.name("response")
1686-
.dataAsMap(content)
1687-
.build();
1678+
ModelTensor modelTensor = ModelTensor.builder().name("response").dataAsMap(content).build();
16881679
ModelTensors modelTensors = ModelTensors.builder().mlModelTensors(Arrays.asList(modelTensor)).build();
16891680
ModelTensorOutput mlModelTensorOutput = ModelTensorOutput.builder().mlModelOutputs(Arrays.asList(modelTensors)).build();
16901681

0 commit comments

Comments
 (0)