Skip to content

Commit 0cb587a

Browse files
committed
Refactor index names
1 parent e98bbcb commit 0cb587a

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/DenseVectorFieldTypeIT.java

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -24,28 +24,24 @@
2424

2525
import java.io.IOException;
2626
import java.util.ArrayList;
27+
import java.util.Arrays;
2728
import java.util.HashMap;
2829
import java.util.List;
2930
import java.util.Locale;
3031
import java.util.Map;
3132
import java.util.Set;
33+
import java.util.stream.Collectors;
3234

3335
import static org.elasticsearch.index.IndexSettings.INDEX_MAPPER_SOURCE_MODE_SETTING;
3436
import static org.elasticsearch.index.mapper.SourceFieldMapper.Mode.SYNTHETIC;
3537
import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertAcked;
3638

3739
public class DenseVectorFieldTypeIT extends AbstractEsqlIntegTestCase {
3840

39-
public static final Set<String> ALL_DENSE_VECTOR_INDEX_TYPES = Set.of(
40-
"int8_hnsw",
41-
"hnsw",
42-
"int4_hnsw",
43-
"bbq_hnsw",
44-
"int8_flat",
45-
"int4_flat",
46-
"bbq_flat",
47-
"flat"
48-
);
41+
public static final Set<String> ALL_DENSE_VECTOR_INDEX_TYPES = Arrays.stream(DenseVectorFieldMapper.VectorIndexType.values())
42+
.filter(DenseVectorFieldMapper.VectorIndexType::isEnabled)
43+
.map(v -> v.getName().toLowerCase(Locale.ROOT))
44+
.collect(Collectors.toSet());
4945

5046
public static final float DELTA = 1e-7F;
5147

0 commit comments

Comments
 (0)