Skip to content

Commit 3d5d827

Browse files
authored
[8.18][Inference API] Rename model_id prop to model in EIS sparse inference request body (#122398)
* Rename model_id prop in EIS sparse inference request body * Update docs/changelog/122398.yaml
1 parent d907cbc commit 3d5d827

File tree

6 files changed

+18
-16
lines changed

6 files changed

+18
-16
lines changed

docs/changelog/122398.yaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
pr: 122398
2+
summary: "[8.18][Inference API] Rename `model_id` prop to model in EIS sparse inference\
3+
\ request body"
4+
area: Inference
5+
type: enhancement
6+
issues: []

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/external/request/elastic/ElasticInferenceServiceSparseEmbeddingsRequestEntity.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,7 @@ public record ElasticInferenceServiceSparseEmbeddingsRequestEntity(
2323
) implements ToXContentObject {
2424

2525
private static final String INPUT_FIELD = "input";
26-
27-
private static final String MODEL_ID_FIELD = "model_id";
28-
26+
private static final String MODEL_FIELD = "model";
2927
private static final String USAGE_CONTEXT = "usage_context";
3028

3129
public ElasticInferenceServiceSparseEmbeddingsRequestEntity {
@@ -44,7 +42,7 @@ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws
4442

4543
builder.endArray();
4644

47-
builder.field(MODEL_ID_FIELD, modelId);
45+
builder.field(MODEL_FIELD, modelId);
4846

4947
// optional field
5048
if ((usageContext == ElasticInferenceServiceUsageContext.UNSPECIFIED) == false) {

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/external/action/elastic/ElasticInferenceServiceActionCreatorTests.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ public void testExecute_ReturnsSuccessfulResponse_ForElserAction() throws IOExce
124124
assertThat(requestMap.get("input"), instanceOf(List.class));
125125
var inputList = (List<String>) requestMap.get("input");
126126
assertThat(inputList, contains("hello world"));
127-
assertThat(requestMap.get("model_id"), is("my-model-id"));
127+
assertThat(requestMap.get("model"), is("my-model-id"));
128128
}
129129
}
130130

@@ -179,7 +179,7 @@ public void testSend_FailsFromInvalidResponseFormat_ForElserAction() throws IOEx
179179
assertThat(requestMap.get("input"), instanceOf(List.class));
180180
var inputList = (List<String>) requestMap.get("input");
181181
assertThat(inputList, contains("hello world"));
182-
assertThat(requestMap.get("model_id"), is("my-model-id"));
182+
assertThat(requestMap.get("model"), is("my-model-id"));
183183
}
184184
}
185185

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/external/request/elastic/ElasticInferenceServiceSparseEmbeddingsRequestEntityTests.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ public void testToXContent_SingleInput_UnspecifiedUsageContext() throws IOExcept
3131
assertThat(xContentString, equalToIgnoringWhitespaceInJsonString("""
3232
{
3333
"input": ["abc"],
34-
"model_id": "my-model-id"
34+
"model": "my-model-id"
3535
}"""));
3636
}
3737

@@ -48,7 +48,7 @@ public void testToXContent_MultipleInputs_UnspecifiedUsageContext() throws IOExc
4848
"abc",
4949
"def"
5050
],
51-
"model_id": "my-model-id"
51+
"model": "my-model-id"
5252
}
5353
"""));
5454
}
@@ -63,7 +63,7 @@ public void testToXContent_MultipleInputs_SearchUsageContext() throws IOExceptio
6363
assertThat(xContentString, equalToIgnoringWhitespaceInJsonString("""
6464
{
6565
"input": ["abc"],
66-
"model_id": "my-model-id",
66+
"model": "my-model-id",
6767
"usage_context": "search"
6868
}
6969
"""));
@@ -79,7 +79,7 @@ public void testToXContent_MultipleInputs_IngestUsageContext() throws IOExceptio
7979
assertThat(xContentString, equalToIgnoringWhitespaceInJsonString("""
8080
{
8181
"input": ["abc"],
82-
"model_id": "my-model-id",
82+
"model": "my-model-id",
8383
"usage_context": "ingest"
8484
}
8585
"""));

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/external/request/elastic/ElasticInferenceServiceSparseEmbeddingsRequestTests.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ public void testCreateHttpRequest_UsageContextSearch() throws IOException {
4747

4848
assertThat(requestMap.size(), equalTo(3));
4949
assertThat(requestMap.get("input"), is(List.of(input)));
50-
assertThat(requestMap.get("model_id"), is(modelId));
50+
assertThat(requestMap.get("model"), is(modelId));
5151
assertThat(requestMap.get("usage_context"), equalTo("search"));
5252
}
5353

@@ -84,7 +84,7 @@ public void testTruncate_ReducesInputTextSizeByHalf() throws IOException {
8484
var requestMap = entityAsMap(httpPost.getEntity().getContent());
8585
assertThat(requestMap, aMapWithSize(2));
8686
assertThat(requestMap.get("input"), is(List.of("ab")));
87-
assertThat(requestMap.get("model_id"), is(modelId));
87+
assertThat(requestMap.get("model"), is(modelId));
8888
}
8989

9090
public void testIsTruncated_ReturnsTrue() {

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/elastic/ElasticInferenceServiceTests.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -504,8 +504,7 @@ public void testInfer_SendsEmbeddingsRequest() throws IOException {
504504
assertThat(request.getHeader(HttpHeaders.CONTENT_TYPE), Matchers.equalTo(XContentType.JSON.mediaType()));
505505

506506
var requestMap = entityAsMap(request.getBody());
507-
508-
assertThat(requestMap, is(Map.of("input", List.of("input text"), "model_id", "my-model-id", "usage_context", "search")));
507+
assertThat(requestMap, is(Map.of("input", List.of("input text"), "model", "my-model-id", "usage_context", "search")));
509508
}
510509
}
511510

@@ -562,8 +561,7 @@ public void testChunkedInfer_PassesThrough() throws IOException {
562561
);
563562

564563
var requestMap = entityAsMap(webServer.requests().get(0).getBody());
565-
566-
assertThat(requestMap, is(Map.of("input", List.of("input text"), "model_id", "my-model-id", "usage_context", "ingest")));
564+
assertThat(requestMap, is(Map.of("input", List.of("input text"), "model", "my-model-id", "usage_context", "ingest")));
567565
}
568566
}
569567

0 commit comments

Comments
 (0)