Skip to content

Commit faca63f

Browse files
committed
Rename model_id prop to model in EIS sparse inference request body
1 parent 3f021a1 commit faca63f

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

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

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

2525
private static final String INPUT_FIELD = "input";
26-
private static final String MODEL_ID_FIELD = "model_id";
26+
private static final String MODEL_FIELD = "model";
2727
private static final String USAGE_CONTEXT = "usage_context";
2828

2929
public ElasticInferenceServiceSparseEmbeddingsRequestEntity {
@@ -42,7 +42,7 @@ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws
4242

4343
builder.endArray();
4444

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

4747
// optional field
4848
if ((usageContext == ElasticInferenceServiceUsageContext.UNSPECIFIED) == false) {

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
@@ -46,7 +46,7 @@ public void testCreateHttpRequest_UsageContextSearch() throws IOException {
4646
var requestMap = entityAsMap(httpPost.getEntity().getContent());
4747
assertThat(requestMap.size(), equalTo(3));
4848
assertThat(requestMap.get("input"), is(List.of(input)));
49-
assertThat(requestMap.get("model_id"), is(modelId));
49+
assertThat(requestMap.get("model"), is(modelId));
5050
assertThat(requestMap.get("usage_context"), equalTo("search"));
5151
}
5252

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

8989
public void testIsTruncated_ReturnsTrue() {

0 commit comments

Comments
 (0)