Skip to content

Commit dce8dd8

Browse files
committed
Update the feature flag & cluster feature names
1 parent 9db655c commit dce8dd8

File tree

8 files changed

+12
-17
lines changed

8 files changed

+12
-17
lines changed

docs/changelog/135940.yaml

Lines changed: 0 additions & 5 deletions
This file was deleted.

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.vectors/200_dense_vector_docvalue_fields.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ setup:
163163
---
164164
"dense_vector docvalues with bfloat16":
165165
- requires:
166-
cluster_features: [ "mapper.vectors.hnsw_bfloat16_on_disk_rescoring" ]
166+
cluster_features: [ "mapper.vectors.generic_vector_format" ]
167167
reason: Needs bfloat16 support
168168
- do:
169169
indices.create:

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.vectors/41_knn_search_bbq_hnsw.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ setup:
340340
---
341341
"Test index configured rescore vector with on-disk rescoring":
342342
- requires:
343-
cluster_features: ["mapper.vectors.hnsw_bfloat16_on_disk_rescoring"]
343+
cluster_features: ["mapper.vectors.generic_vector_format"]
344344
reason: Needs on_disk_rescoring feature
345345
- skip:
346346
features: "headers"

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.vectors/41_knn_search_bbq_hnsw_bfloat16.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
setup:
22
- requires:
3-
cluster_features: "mapper.vectors.hnsw_bfloat16_on_disk_rescoring"
3+
cluster_features: "mapper.vectors.generic_vector_format"
44
reason: 'bfloat16 needs to be supported'
55
- do:
66
indices.create:

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.vectors/42_knn_search_bbq_flat_bfloat16.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
setup:
22
- requires:
3-
cluster_features: "mapper.vectors.hnsw_bfloat16_on_disk_rescoring"
3+
cluster_features: "mapper.vectors.generic_vector_format"
44
reason: 'bfloat16 needs to be supported'
55
- do:
66
indices.create:

server/src/main/java/org/elasticsearch/index/codec/vectors/es93/ES93GenericFlatVectorsFormat.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525

2626
public class ES93GenericFlatVectorsFormat extends AbstractFlatVectorsFormat {
2727

28-
public static final FeatureFlag ES93_VECTOR_FORMATS = new FeatureFlag("es93_vector_formats");
28+
public static final FeatureFlag GENERIC_VECTOR_FORMAT = new FeatureFlag("generic_vector_format");
2929

3030
static final String NAME = "ES93GenericFlatVectorsFormat";
3131
static final String VECTOR_FORMAT_INFO_EXTENSION = "vfi";

server/src/main/java/org/elasticsearch/index/mapper/MapperFeatures.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ public class MapperFeatures implements FeatureSpecification {
6161
);
6262
static final NodeFeature EXCLUDE_VECTORS_DOCVALUE_BUGFIX = new NodeFeature("mapper.exclude_vectors_docvalue_bugfix");
6363
static final NodeFeature BASE64_DENSE_VECTORS = new NodeFeature("mapper.base64_dense_vectors");
64-
public static final NodeFeature HNSW_BFLOAT16_ON_DISK_RESCORING = new NodeFeature("mapper.vectors.hnsw_bfloat16_on_disk_rescoring");
64+
public static final NodeFeature GENERIC_VECTOR_FORMAT = new NodeFeature("mapper.vectors.generic_vector_format");
6565

6666
@Override
6767
public Set<NodeFeature> getTestFeatures() {
@@ -106,9 +106,9 @@ public Set<NodeFeature> getTestFeatures() {
106106
EXCLUDE_VECTORS_DOCVALUE_BUGFIX,
107107
BASE64_DENSE_VECTORS
108108
);
109-
if (ES93GenericFlatVectorsFormat.ES93_VECTOR_FORMATS.isEnabled()) {
109+
if (ES93GenericFlatVectorsFormat.GENERIC_VECTOR_FORMAT.isEnabled()) {
110110
features = new HashSet<>(features);
111-
features.add(HNSW_BFLOAT16_ON_DISK_RESCORING);
111+
features.add(GENERIC_VECTOR_FORMAT);
112112
}
113113
return features;
114114
}

server/src/main/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldMapper.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ public static class Builder extends FieldMapper.Builder {
242242
private final Parameter<ElementType> elementType = new Parameter<>("element_type", false, () -> ElementType.FLOAT, (n, c, o) -> {
243243
ElementType elementType = namesToElementType.get((String) o);
244244
if (elementType == null
245-
|| (elementType == ElementType.BFLOAT16 && ES93GenericFlatVectorsFormat.ES93_VECTOR_FORMATS.isEnabled() == false)) {
245+
|| (elementType == ElementType.BFLOAT16 && ES93GenericFlatVectorsFormat.GENERIC_VECTOR_FORMAT.isEnabled() == false)) {
246246
throw new MapperParsingException("invalid element_type [" + o + "]; available types are " + namesToElementType.keySet());
247247
}
248248
return elementType;
@@ -2149,7 +2149,7 @@ public static class HnswIndexOptions extends DenseVectorIndexOptions {
21492149

21502150
@Override
21512151
public KnnVectorsFormat getVectorsFormat(ElementType elementType) {
2152-
if (ES93GenericFlatVectorsFormat.ES93_VECTOR_FORMATS.isEnabled()) {
2152+
if (ES93GenericFlatVectorsFormat.GENERIC_VECTOR_FORMAT.isEnabled()) {
21532153
return new ES93HnswVectorsFormat(m, efConstruction, elementType);
21542154
} else {
21552155
if (elementType == ElementType.BIT) {
@@ -2231,7 +2231,7 @@ public BBQHnswIndexOptions(int m, int efConstruction, boolean onDiskRescore, Res
22312231
@Override
22322232
KnnVectorsFormat getVectorsFormat(ElementType elementType) {
22332233
assert elementType == ElementType.FLOAT || elementType == ElementType.BFLOAT16;
2234-
return ES93GenericFlatVectorsFormat.ES93_VECTOR_FORMATS.isEnabled()
2234+
return ES93GenericFlatVectorsFormat.GENERIC_VECTOR_FORMAT.isEnabled()
22352235
? new ES93HnswBinaryQuantizedVectorsFormat(m, efConstruction, elementType, onDiskRescore)
22362236
: new ES818HnswBinaryQuantizedVectorsFormat(m, efConstruction);
22372237
}
@@ -2299,7 +2299,7 @@ static class BBQFlatIndexOptions extends QuantizedIndexOptions {
22992299
@Override
23002300
KnnVectorsFormat getVectorsFormat(ElementType elementType) {
23012301
assert elementType == ElementType.FLOAT || elementType == ElementType.BFLOAT16;
2302-
return ES93GenericFlatVectorsFormat.ES93_VECTOR_FORMATS.isEnabled()
2302+
return ES93GenericFlatVectorsFormat.GENERIC_VECTOR_FORMAT.isEnabled()
23032303
? new ES93BinaryQuantizedVectorsFormat(elementType, false)
23042304
: new ES818BinaryQuantizedVectorsFormat();
23052305
}

0 commit comments

Comments
 (0)