77
88package org .elasticsearch .xpack .inference ;
99
10- import org .apache .lucene .tests .util .LuceneTestCase ;
1110import org .elasticsearch .client .Request ;
1211import org .elasticsearch .common .Strings ;
1312import org .elasticsearch .inference .TaskType ;
1918
2019import static org .hamcrest .Matchers .containsString ;
2120
22- // Tests disabled in CI due to the models being too large to download. Can be enabled (commented out) for local testing
23- @ LuceneTestCase . AwaitsFix ( bugUrl = "https://github.com/elastic/elasticsearch/issues/105198" )
21+ // This test was previously disabled in CI due to the models being too large
22+ // See "https://github.com/elastic/elasticsearch/issues/105198".
2423public class TextEmbeddingCrudIT extends InferenceBaseRestTest {
2524
26- public void testPutE5Small_withNoModelVariant () throws IOException {
25+ public void testPutE5Small_withNoModelVariant () {
2726 {
2827 String inferenceEntityId = randomAlphaOfLength (10 ).toLowerCase ();
2928 expectThrows (
@@ -124,7 +123,7 @@ private String noModelIdVariantJsonEntity() {
124123 private String platformAgnosticModelVariantJsonEntity () {
125124 return """
126125 {
127- "service": "text_embedding ",
126+ "service": "elasticsearch ",
128127 "service_settings": {
129128 "num_allocations": 1,
130129 "num_threads": 1,
@@ -137,7 +136,7 @@ private String platformAgnosticModelVariantJsonEntity() {
137136 private String platformSpecificModelVariantJsonEntity () {
138137 return """
139138 {
140- "service": "text_embedding ",
139+ "service": "elasticsearch ",
141140 "service_settings": {
142141 "num_allocations": 1,
143142 "num_threads": 1,
@@ -150,7 +149,7 @@ private String platformSpecificModelVariantJsonEntity() {
150149 private String fakeModelVariantJsonEntity () {
151150 return """
152151 {
153- "service": "text_embedding ",
152+ "service": "elasticsearch ",
154153 "service_settings": {
155154 "num_allocations": 1,
156155 "num_threads": 1,
0 commit comments