Skip to content

Commit 0c3977a

Browse files
authored
[ML] Add ElasticInferenceServiceCompletionServiceSettings (#123155) (#123174)
Adding the missing NamedWriteable to the registry.
1 parent 673c268 commit 0c3977a

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
@@ -59,6 +59,7 @@
5959
import org.elasticsearch.xpack.inference.services.cohere.rerank.CohereRerankServiceSettings;
6060
import org.elasticsearch.xpack.inference.services.cohere.rerank.CohereRerankTaskSettings;
6161
import org.elasticsearch.xpack.inference.services.elastic.ElasticInferenceServiceSparseEmbeddingsServiceSettings;
62+
import org.elasticsearch.xpack.inference.services.elastic.completion.ElasticInferenceServiceCompletionServiceSettings;
6263
import org.elasticsearch.xpack.inference.services.elasticsearch.CustomElandInternalServiceSettings;
6364
import org.elasticsearch.xpack.inference.services.elasticsearch.CustomElandInternalTextEmbeddingServiceSettings;
6465
import org.elasticsearch.xpack.inference.services.elasticsearch.ElasticRerankerServiceSettings;
@@ -634,5 +635,12 @@ private static void addEisNamedWriteables(List<NamedWriteableRegistry.Entry> nam
634635
ElasticInferenceServiceSparseEmbeddingsServiceSettings::new
635636
)
636637
);
638+
namedWriteables.add(
639+
new NamedWriteableRegistry.Entry(
640+
ServiceSettings.class,
641+
ElasticInferenceServiceCompletionServiceSettings.NAME,
642+
ElasticInferenceServiceCompletionServiceSettings::new
643+
)
644+
);
637645
}
638646
}

0 commit comments

Comments
 (0)