@@ -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