Skip to content

Commit b5eae66

Browse files
committed
[ML] Add ElasticInferenceServiceCompletionServiceSettings
Adding the missing NamedWriteable to the registry.
1 parent a3313a3 commit b5eae66

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

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)