Skip to content

Commit 6462cc9

Browse files
authored
[Inference API] Add ElasticInferenceServiceDenseTextEmbeddingsServiceSettings to InferenceNamedWriteablesProvider (#138484)
Needed for correct type identification during deserialization
1 parent f4d969d commit 6462cc9

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

docs/changelog/138484.yaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
pr: 138484
2+
summary: "[Inference API] Add `ElasticInferenceServiceDenseTextEmbeddingsServiceSettings`\
3+
\ to `InferenceNamedWriteablesProvider`"
4+
area: Machine Learning
5+
type: bug
6+
issues: []

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

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@
7474
import org.elasticsearch.xpack.inference.services.custom.response.SparseEmbeddingResponseParser;
7575
import org.elasticsearch.xpack.inference.services.deepseek.DeepSeekChatCompletionModel;
7676
import org.elasticsearch.xpack.inference.services.elastic.completion.ElasticInferenceServiceCompletionServiceSettings;
77+
import org.elasticsearch.xpack.inference.services.elastic.densetextembeddings.ElasticInferenceServiceDenseTextEmbeddingsServiceSettings;
7778
import org.elasticsearch.xpack.inference.services.elastic.rerank.ElasticInferenceServiceRerankServiceSettings;
7879
import org.elasticsearch.xpack.inference.services.elastic.sparseembeddings.ElasticInferenceServiceSparseEmbeddingsServiceSettings;
7980
import org.elasticsearch.xpack.inference.services.elasticsearch.CustomElandInternalServiceSettings;
@@ -879,5 +880,14 @@ private static void addElasticNamedWriteables(List<NamedWriteableRegistry.Entry>
879880
ElasticInferenceServiceRerankServiceSettings::new
880881
)
881882
);
883+
884+
// Dense Text Embeddings
885+
namedWriteables.add(
886+
new NamedWriteableRegistry.Entry(
887+
ServiceSettings.class,
888+
ElasticInferenceServiceDenseTextEmbeddingsServiceSettings.NAME,
889+
ElasticInferenceServiceDenseTextEmbeddingsServiceSettings::new
890+
)
891+
);
882892
}
883893
}

0 commit comments

Comments
 (0)