Skip to content

Commit 541c917

Browse files
committed
fix more tests
1 parent ca1c709 commit 541c917

File tree

4 files changed

+10
-4
lines changed

4 files changed

+10
-4
lines changed

server/src/main/java/org/elasticsearch/inference/InferenceServiceRegistry.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@ public Map<String, InferenceService> getServices() {
4646
}
4747

4848
public Optional<InferenceService> getService(String serviceName) {
49-
5049
if ("elser".equals(serviceName)) { // ElserService.NAME before removal
5150
// here we are aliasing the elser service to use the elasticsearch service instead
5251
return Optional.ofNullable(services.get("elasticsearch")); // ElasticsearchInternalService.NAME

x-pack/plugin/esql/qa/server/src/main/java/org/elasticsearch/xpack/esql/qa/rest/SemanticMatchTestCase.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import java.io.IOException;
1919
import java.util.Map;
2020

21+
import static org.hamcrest.Matchers.equalTo;
2122
import static org.hamcrest.core.StringContains.containsString;
2223

2324
public abstract class SemanticMatchTestCase extends ESRestTestCase {
@@ -98,7 +99,12 @@ public void setUpTextEmbeddingInferenceEndpoint() throws IOException {
9899
}
99100
}
100101
""");
101-
adminClient().performRequest(request);
102+
try {
103+
adminClient().performRequest(request);
104+
} catch (ResponseException exc) {
105+
// in case the removal failed
106+
assertThat(exc.getResponse().getStatusLine().getStatusCode(), equalTo(400));
107+
}
102108
}
103109

104110
@After

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,9 @@ public void testSkipValidationAndStart() throws IOException {
147147
{
148148
"service": "openai",
149149
"service_settings": {
150-
"api_key": "XXXX"
150+
"api_key": "XXXX",
151+
"dimensions": 128,
152+
"similarity": "cosine"
151153
},
152154
"task_settings": {
153155
"model": "text-embedding-ada-002"

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/openai/embeddings/OpenAiEmbeddingsServiceSettings.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,6 @@ private static CommonFields fromMap(
9696
ValidationException validationException,
9797
ConfigurationParseContext context
9898
) {
99-
10099
String url = extractOptionalString(map, URL, ModelConfigurations.SERVICE_SETTINGS, validationException);
101100
String organizationId = extractOptionalString(map, ORGANIZATION, ModelConfigurations.SERVICE_SETTINGS, validationException);
102101
SimilarityMeasure similarity = extractSimilarity(map, ModelConfigurations.SERVICE_SETTINGS, validationException);

0 commit comments

Comments
 (0)