Skip to content

Commit 3dc1fd5

Browse files
authored
Set correct type identification during deserialization (elastic#138847)
1 parent 3d691a5 commit 3dc1fd5

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

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
@@ -79,6 +79,7 @@
7979
import org.elasticsearch.xpack.inference.services.custom.response.TextEmbeddingResponseParser;
8080
import org.elasticsearch.xpack.inference.services.deepseek.DeepSeekChatCompletionModel;
8181
import org.elasticsearch.xpack.inference.services.elastic.completion.ElasticInferenceServiceCompletionServiceSettings;
82+
import org.elasticsearch.xpack.inference.services.elastic.densetextembeddings.ElasticInferenceServiceDenseTextEmbeddingsServiceSettings;
8283
import org.elasticsearch.xpack.inference.services.elastic.rerank.ElasticInferenceServiceRerankServiceSettings;
8384
import org.elasticsearch.xpack.inference.services.elastic.sparseembeddings.ElasticInferenceServiceSparseEmbeddingsServiceSettings;
8485
import org.elasticsearch.xpack.inference.services.elasticsearch.CustomElandInternalServiceSettings;
@@ -857,5 +858,14 @@ private static void addElasticNamedWriteables(List<NamedWriteableRegistry.Entry>
857858
ElasticInferenceServiceRerankServiceSettings::new
858859
)
859860
);
861+
862+
// Dense Text Embeddings
863+
namedWriteables.add(
864+
new NamedWriteableRegistry.Entry(
865+
ServiceSettings.class,
866+
ElasticInferenceServiceDenseTextEmbeddingsServiceSettings.NAME,
867+
ElasticInferenceServiceDenseTextEmbeddingsServiceSettings::new
868+
)
869+
);
860870
}
861871
}

0 commit comments

Comments
 (0)