|
33 | 33 | import org.elasticsearch.index.IndexVersions; |
34 | 34 | import org.elasticsearch.index.codec.CodecService; |
35 | 35 | import org.elasticsearch.index.codec.LegacyPerFieldMapperCodec; |
36 | | -import org.elasticsearch.index.codec.PerFieldMapperCodec; |
37 | 36 | import org.elasticsearch.index.codec.vectors.BFloat16; |
38 | 37 | import org.elasticsearch.index.codec.vectors.diskbbq.es94.ES940DiskBBQVectorsFormat; |
39 | 38 | import org.elasticsearch.index.codec.vectors.es93.ES93HnswBinaryQuantizedVectorsFormat; |
@@ -1974,17 +1973,11 @@ public void testKnnVectorsFormat() throws IOException { |
1974 | 1973 | ); |
1975 | 1974 | CodecService codecService = new CodecService(mapperService, BigArrays.NON_RECYCLING_INSTANCE, null); |
1976 | 1975 | Codec codec = codecService.codec("default"); |
1977 | | - KnnVectorsFormat knnVectorsFormat; |
1978 | | - if (CodecService.ZSTD_STORED_FIELDS_FEATURE_FLAG) { |
1979 | | - assertThat(codec, instanceOf(PerFieldMapperCodec.class)); |
1980 | | - knnVectorsFormat = ((PerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
1981 | | - } else { |
1982 | | - if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
1983 | | - codec = deduplicateFieldInfosCodec.delegate(); |
1984 | | - } |
1985 | | - assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
1986 | | - knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
| 1976 | + if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
| 1977 | + codec = deduplicateFieldInfosCodec.delegate(); |
1987 | 1978 | } |
| 1979 | + assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
| 1980 | + KnnVectorsFormat knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
1988 | 1981 | String expectedString = "ES93HnswVectorsFormat(name=ES93HnswVectorsFormat, maxConn=" |
1989 | 1982 | + (setM ? m : DEFAULT_MAX_CONN) |
1990 | 1983 | + ", beamWidth=" |
@@ -2049,17 +2042,11 @@ public void testKnnQuantizedFlatVectorsFormat() throws IOException { |
2049 | 2042 | ); |
2050 | 2043 | CodecService codecService = new CodecService(mapperService, BigArrays.NON_RECYCLING_INSTANCE, null); |
2051 | 2044 | Codec codec = codecService.codec("default"); |
2052 | | - KnnVectorsFormat knnVectorsFormat; |
2053 | | - if (CodecService.ZSTD_STORED_FIELDS_FEATURE_FLAG) { |
2054 | | - assertThat(codec, instanceOf(PerFieldMapperCodec.class)); |
2055 | | - knnVectorsFormat = ((PerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2056 | | - } else { |
2057 | | - if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
2058 | | - codec = deduplicateFieldInfosCodec.delegate(); |
2059 | | - } |
2060 | | - assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
2061 | | - knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
| 2045 | + if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
| 2046 | + codec = deduplicateFieldInfosCodec.delegate(); |
2062 | 2047 | } |
| 2048 | + assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
| 2049 | + KnnVectorsFormat knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2063 | 2050 | VectorScorerFactory factory = VectorScorerFactory.instance().orElse(null); |
2064 | 2051 | String encoding = quantizedFlatFormat.equals("int4_flat") ? "PACKED_NIBBLE" : "SEVEN_BIT"; |
2065 | 2052 | assertThat( |
@@ -2100,17 +2087,11 @@ public void testKnnQuantizedHNSWVectorsFormat() throws IOException { |
2100 | 2087 | ); |
2101 | 2088 | CodecService codecService = new CodecService(mapperService, BigArrays.NON_RECYCLING_INSTANCE, null); |
2102 | 2089 | Codec codec = codecService.codec("default"); |
2103 | | - KnnVectorsFormat knnVectorsFormat; |
2104 | | - if (CodecService.ZSTD_STORED_FIELDS_FEATURE_FLAG) { |
2105 | | - assertThat(codec, instanceOf(PerFieldMapperCodec.class)); |
2106 | | - knnVectorsFormat = ((PerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2107 | | - } else { |
2108 | | - if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
2109 | | - codec = deduplicateFieldInfosCodec.delegate(); |
2110 | | - } |
2111 | | - assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
2112 | | - knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
| 2090 | + if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
| 2091 | + codec = deduplicateFieldInfosCodec.delegate(); |
2113 | 2092 | } |
| 2093 | + assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
| 2094 | + KnnVectorsFormat knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2114 | 2095 | VectorScorerFactory factory = VectorScorerFactory.instance().orElse(null); |
2115 | 2096 | assertThat( |
2116 | 2097 | knnVectorsFormat, |
@@ -2157,17 +2138,11 @@ public void testKnnBBQHNSWVectorsFormat() throws IOException { |
2157 | 2138 | ); |
2158 | 2139 | CodecService codecService = new CodecService(mapperService, BigArrays.NON_RECYCLING_INSTANCE, null); |
2159 | 2140 | Codec codec = codecService.codec("default"); |
2160 | | - KnnVectorsFormat knnVectorsFormat; |
2161 | | - if (CodecService.ZSTD_STORED_FIELDS_FEATURE_FLAG) { |
2162 | | - assertThat(codec, instanceOf(PerFieldMapperCodec.class)); |
2163 | | - knnVectorsFormat = ((PerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2164 | | - } else { |
2165 | | - if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
2166 | | - codec = deduplicateFieldInfosCodec.delegate(); |
2167 | | - } |
2168 | | - assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
2169 | | - knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
| 2141 | + if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
| 2142 | + codec = deduplicateFieldInfosCodec.delegate(); |
2170 | 2143 | } |
| 2144 | + assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
| 2145 | + KnnVectorsFormat knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2171 | 2146 | String expectedPrefix = "ES93HnswBinaryQuantizedVectorsFormat(name=ES93HnswBinaryQuantizedVectorsFormat, maxConn=" |
2172 | 2147 | + m |
2173 | 2148 | + ", beamWidth=" |
@@ -2213,17 +2188,11 @@ public void testKnnHalfByteQuantizedHNSWVectorsFormat() throws IOException { |
2213 | 2188 | })); |
2214 | 2189 | CodecService codecService = new CodecService(mapperService, BigArrays.NON_RECYCLING_INSTANCE, null); |
2215 | 2190 | Codec codec = codecService.codec("default"); |
2216 | | - KnnVectorsFormat knnVectorsFormat; |
2217 | | - if (CodecService.ZSTD_STORED_FIELDS_FEATURE_FLAG) { |
2218 | | - assertThat(codec, instanceOf(PerFieldMapperCodec.class)); |
2219 | | - knnVectorsFormat = ((PerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2220 | | - } else { |
2221 | | - if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
2222 | | - codec = deduplicateFieldInfosCodec.delegate(); |
2223 | | - } |
2224 | | - assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
2225 | | - knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
| 2191 | + if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) { |
| 2192 | + codec = deduplicateFieldInfosCodec.delegate(); |
2226 | 2193 | } |
| 2194 | + assertThat(codec, instanceOf(LegacyPerFieldMapperCodec.class)); |
| 2195 | + KnnVectorsFormat knnVectorsFormat = ((LegacyPerFieldMapperCodec) codec).getKnnVectorsFormatForField("field"); |
2227 | 2196 | VectorScorerFactory factory = VectorScorerFactory.instance().orElse(null); |
2228 | 2197 | assertThat( |
2229 | 2198 | knnVectorsFormat, |
|
0 commit comments