Skip to content

Commit 4f7b964

Browse files
Switch sparse embedding name from gateway
1 parent 58ec62d commit 4f7b964

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

x-pack/plugin/inference/qa/inference-service-tests/src/javaRestTest/java/org/elasticsearch/xpack/inference/MockElasticInferenceServiceAuthorizationServer.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ public static MockElasticInferenceServiceAuthorizationServer enabledWithSparseAn
3131
"models": [
3232
{
3333
"model_name": "model-a",
34-
"task_types": ["embedding/text/sparse", "chat/completion"]
34+
"task_types": ["embed/text/sparse", "chat"]
3535
}
3636
]
3737
}

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/external/response/elastic/ElasticInferenceServiceAuthorizationResponseEntity.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,9 @@ public class ElasticInferenceServiceAuthorizationResponseEntity implements Infer
4040

4141
public static final String NAME = "elastic_inference_service_auth_results";
4242
private static final Map<String, TaskType> ELASTIC_INFERENCE_SERVICE_TASK_TYPE_MAPPING = Map.of(
43-
"embedding/text/sparse",
43+
"embed/text/sparse",
4444
TaskType.SPARSE_EMBEDDING,
45-
"chat/completion",
45+
"chat",
4646
TaskType.CHAT_COMPLETION
4747
);
4848

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/external/http/sender/HttpRequestSenderTests.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ public void testSendWithoutQueuing_SendsRequestAndReceivesResponse() throws Exce
150150
"models": [
151151
{
152152
"model_name": "model-a",
153-
"task_types": ["embedding/text/sparse", "chat/completion"]
153+
"task_types": ["embed/text/sparse", "chat"]
154154
}
155155
]
156156
}

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ public void testGetAuthorization_FailsWhenAnInvalidFieldIsFound() throws IOExcep
110110
"models": [
111111
{
112112
"invalid-field": "model-a",
113-
"task-types": ["embedding/text/sparse", "chat/completion"]
113+
"task-types": ["embed/text/sparse", "chat"]
114114
}
115115
]
116116
}
@@ -160,7 +160,7 @@ public void testGetAuthorization_ReturnsAValidResponse() throws IOException {
160160
"models": [
161161
{
162162
"model_name": "model-a",
163-
"task_types": ["embedding/text/sparse", "chat/completion"]
163+
"task_types": ["embed/text/sparse", "chat"]
164164
}
165165
]
166166
}

0 commit comments

Comments
 (0)