|
@@ -504,7 +504,7 @@ public class ElasticInferenceServiceTests extends ESTestCase {
|
|
|
assertThat(request.getHeader(HttpHeaders.CONTENT_TYPE), Matchers.equalTo(XContentType.JSON.mediaType()));
|
|
|
|
|
|
var requestMap = entityAsMap(request.getBody());
|
|
|
- assertThat(requestMap, is(Map.of("input", List.of("input text"), "model_id", "my-model-id", "usage_context", "search")));
|
|
|
+ assertThat(requestMap, is(Map.of("input", List.of("input text"), "model", "my-model-id", "usage_context", "search")));
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -562,7 +562,7 @@ public class ElasticInferenceServiceTests extends ESTestCase {
|
|
|
);
|
|
|
|
|
|
var requestMap = entityAsMap(webServer.requests().get(0).getBody());
|
|
|
- assertThat(requestMap, is(Map.of("input", List.of("input text"), "model_id", "my-model-id", "usage_context", "ingest")));
|
|
|
+ assertThat(requestMap, is(Map.of("input", List.of("input text"), "model", "my-model-id", "usage_context", "ingest")));
|
|
|
}
|
|
|
}
|
|
|
|