@@ -393,6 +393,10 @@ public static void createInferenceEndpoints(RestClient client) throws IOExceptio
393393 createSparseEmbeddingInferenceEndpoint (client );
394394 }
395395
396+ if (clusterHasDenseEmbeddingInferenceEndpoint (client ) == false ) {
397+ createDenseEmbeddingInferenceEndpoint (client );
398+ }
399+
396400 if (clusterHasRerankInferenceEndpoint (client ) == false ) {
397401 createRerankInferenceEndpoint (client );
398402 }
@@ -404,6 +408,7 @@ public static void createInferenceEndpoints(RestClient client) throws IOExceptio
404408
405409 public static void deleteInferenceEndpoints (RestClient client ) throws IOException {
406410 deleteSparseEmbeddingInferenceEndpoint (client );
411+ deleteDenseEmbeddingInferenceEndpoint (client );
407412 deleteRerankInferenceEndpoint (client );
408413 deleteCompletionInferenceEndpoint (client );
409414 }
@@ -427,6 +432,24 @@ public static boolean clusterHasSparseEmbeddingInferenceEndpoint(RestClient clie
427432 return clusterHasInferenceEndpoint (client , TaskType .SPARSE_EMBEDDING , "test_sparse_inference" );
428433 }
429434
435+ public static void createDenseEmbeddingInferenceEndpoint (RestClient client ) throws IOException {
436+ createInferenceEndpoint (client , TaskType .TEXT_EMBEDDING , "test_dense_inference" , """
437+ {
438+ "service": "text_embedding_test_service",
439+ "service_settings": { "model": "my_model", "api_key": "abc64", "dimensions": 10 },
440+ "task_settings": { }
441+ }
442+ """ );
443+ }
444+
445+ public static void deleteDenseEmbeddingInferenceEndpoint (RestClient client ) throws IOException {
446+ deleteInferenceEndpoint (client , "test_dense_inference" );
447+ }
448+
449+ public static boolean clusterHasDenseEmbeddingInferenceEndpoint (RestClient client ) throws IOException {
450+ return clusterHasInferenceEndpoint (client , TaskType .TEXT_EMBEDDING , "test_dense_inference" );
451+ }
452+
430453 public static void createRerankInferenceEndpoint (RestClient client ) throws IOException {
431454 createInferenceEndpoint (client , TaskType .RERANK , "test_reranker" , """
432455 {
0 commit comments