Skip to content

Commit 2d6689d

Browse files
authored
[ML] Add ElasticInferenceServiceCompletionServiceSettings (#123155) (#123173)
Adding the missing NamedWriteable to the registry.
1 parent fcfde3b commit 2d6689d

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

docs/changelog/123155.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
pr: 123155
2+
summary: Add `ElasticInferenceServiceCompletionServiceSettings`
3+
area: Machine Learning
4+
type: bug
5+
issues: []

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/InferenceNamedWriteablesProvider.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@
5757
import org.elasticsearch.xpack.inference.services.cohere.rerank.CohereRerankServiceSettings;
5858
import org.elasticsearch.xpack.inference.services.cohere.rerank.CohereRerankTaskSettings;
5959
import org.elasticsearch.xpack.inference.services.elastic.ElasticInferenceServiceSparseEmbeddingsServiceSettings;
60+
import org.elasticsearch.xpack.inference.services.elastic.completion.ElasticInferenceServiceCompletionServiceSettings;
6061
import org.elasticsearch.xpack.inference.services.elasticsearch.CustomElandInternalServiceSettings;
6162
import org.elasticsearch.xpack.inference.services.elasticsearch.CustomElandInternalTextEmbeddingServiceSettings;
6263
import org.elasticsearch.xpack.inference.services.elasticsearch.ElasticRerankerServiceSettings;
@@ -618,5 +619,12 @@ private static void addEisNamedWriteables(List<NamedWriteableRegistry.Entry> nam
618619
ElasticInferenceServiceSparseEmbeddingsServiceSettings::new
619620
)
620621
);
622+
namedWriteables.add(
623+
new NamedWriteableRegistry.Entry(
624+
ServiceSettings.class,
625+
ElasticInferenceServiceCompletionServiceSettings.NAME,
626+
ElasticInferenceServiceCompletionServiceSettings::new
627+
)
628+
);
621629
}
622630
}

0 commit comments

Comments
 (0)