Skip to content

Commit 21400d2

Browse files
committed
Mirror upstream elastic#135940 as single snapshot commit for AI review
BASE=bcb861ae10000f94977487193d242c6f30443741 HEAD=eed749e3a6d09e03e0731f5c526fa49e15cd471a Branch=main
1 parent bcb861a commit 21400d2

File tree

54 files changed

+3171
-307
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

54 files changed

+3171
-307
lines changed

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

Lines changed: 103 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,6 @@ setup:
123123
- match: {hits.hits.0.fields.vector5.0: [1, 111, -13, 15, -128]}
124124
- match: {hits.hits.0.fields.vector6.0: [-1, 11, 0, 12, 111]}
125125

126-
127126
- match: {hits.hits.1._id: "2"}
128127
- match: {hits.hits.1.fields.name.0: "moose.jpg"}
129128

@@ -143,7 +142,6 @@ setup:
143142
- match: {hits.hits.1.fields.vector4.0: [-1, 50, -1, 1, 120]}
144143
- match: {hits.hits.1.fields.vector5.0: [1, 111, -13, 15, -128]}
145144

146-
147145
- match: {hits.hits.2._id: "3"}
148146
- match: {hits.hits.2.fields.name.0: "rabbit.jpg"}
149147

@@ -161,3 +159,106 @@ setup:
161159
- close_to: { hits.hits.2.fields.vector2.0.4: { value: -100.0, error: 0.001 } }
162160

163161
- match: {hits.hits.2.fields.vector3.0: [-1, 100, -13, 15, -128]}
162+
163+
---
164+
"dense_vector docvalues with bfloat16":
165+
- requires:
166+
cluster_features: [ "mapper.vectors.generic_vector_format" ]
167+
reason: Needs bfloat16 support
168+
- skip:
169+
awaits_fix: https://github.com/elastic/elasticsearch/issues/131109
170+
reason: Feature flag not enabled
171+
- do:
172+
indices.create:
173+
index: test-bfloat16
174+
body:
175+
mappings:
176+
properties:
177+
name:
178+
type: keyword
179+
vector7:
180+
type: dense_vector
181+
element_type: bfloat16
182+
dims: 5
183+
index: true
184+
vector8:
185+
type: dense_vector
186+
element_type: bfloat16
187+
dims: 5
188+
index: false
189+
190+
- do:
191+
index:
192+
index: test-bfloat16
193+
id: "1"
194+
body:
195+
name: cow.jpg
196+
vector7: [230.0, 300.33, -34.8988, 15.555, -200.0]
197+
vector8: [130.0, 115.0, -1.02, 15.555, -100.0]
198+
- do:
199+
index:
200+
index: test-bfloat16
201+
id: "2"
202+
body:
203+
name: moose.jpg
204+
vector7: [-0.5, 100.0, -13, 14.8, -156.0]
205+
- do:
206+
index:
207+
index: test-bfloat16
208+
id: "3"
209+
body:
210+
name: rabbit.jpg
211+
vector8: [130.0, 115.0, -1.02, 15.555, -100.0]
212+
213+
- do:
214+
indices.refresh: {}
215+
216+
- do:
217+
search:
218+
_source: false
219+
index: test-bfloat16
220+
body:
221+
docvalue_fields: [name, vector7, vector8]
222+
sort: name
223+
224+
- match: {hits.hits.0._id: "1"}
225+
- match: {hits.hits.0.fields.name.0: "cow.jpg"}
226+
227+
- length: {hits.hits.0.fields.vector7.0: 5}
228+
- length: {hits.hits.0.fields.vector8.0: 5}
229+
230+
- close_to: { hits.hits.0.fields.vector7.0.0: { value: 230.0, error: 0.1 } }
231+
- close_to: { hits.hits.0.fields.vector7.0.1: { value: 300.33, error: 0.1 } }
232+
- close_to: { hits.hits.0.fields.vector7.0.2: { value: -34.8988, error: 0.1 } }
233+
- close_to: { hits.hits.0.fields.vector7.0.3: { value: 15.555, error: 0.1 } }
234+
- close_to: { hits.hits.0.fields.vector7.0.4: { value: -200.0, error: 0.1 } }
235+
236+
- close_to: { hits.hits.0.fields.vector8.0.0: { value: 130.0, error: 0.1 } }
237+
- close_to: { hits.hits.0.fields.vector8.0.1: { value: 115.0, error: 0.1 } }
238+
- close_to: { hits.hits.0.fields.vector8.0.2: { value: -1.02, error: 0.1 } }
239+
- close_to: { hits.hits.0.fields.vector8.0.3: { value: 15.555, error: 0.1 } }
240+
- close_to: { hits.hits.0.fields.vector8.0.4: { value: -100.0, error: 0.1 } }
241+
242+
- match: {hits.hits.1._id: "2"}
243+
- match: {hits.hits.1.fields.name.0: "moose.jpg"}
244+
245+
- length: {hits.hits.1.fields.vector7.0: 5}
246+
- match: {hits.hits.1.fields.vector8: null}
247+
248+
- close_to: { hits.hits.1.fields.vector7.0.0: { value: -0.5, error: 0.1 } }
249+
- close_to: { hits.hits.1.fields.vector7.0.1: { value: 100.0, error: 0.1 } }
250+
- close_to: { hits.hits.1.fields.vector7.0.2: { value: -13, error: 0.1 } }
251+
- close_to: { hits.hits.1.fields.vector7.0.3: { value: 14.8, error: 0.1 } }
252+
- close_to: { hits.hits.1.fields.vector7.0.4: { value: -156.0, error: 0.1 } }
253+
254+
- match: {hits.hits.2._id: "3"}
255+
- match: {hits.hits.2.fields.name.0: "rabbit.jpg"}
256+
257+
- length: {hits.hits.2.fields.vector8.0: 5}
258+
- match: {hits.hits.2.fields.vector7: null}
259+
260+
- close_to: { hits.hits.2.fields.vector8.0.0: { value: 130.0, error: 0.1 } }
261+
- close_to: { hits.hits.2.fields.vector8.0.1: { value: 115.0, error: 0.1 } }
262+
- close_to: { hits.hits.2.fields.vector8.0.2: { value: -1.02, error: 0.1 } }
263+
- close_to: { hits.hits.2.fields.vector8.0.3: { value: 15.555, error: 0.1 } }
264+
- close_to: { hits.hits.2.fields.vector8.0.4: { value: -100.0, error: 0.1 } }

0 commit comments

Comments
 (0)