Skip to content

Commit 2b45433

Browse files
committed
Fixing flakiness in CSV tests.
1 parent a2822e3 commit 2b45433

File tree

1 file changed

+19
-22
lines changed

1 file changed

+19
-22
lines changed

x-pack/plugin/esql/qa/testFixtures/src/main/resources/text-embedding.csv-spec

Lines changed: 19 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -35,16 +35,15 @@ required_capability: semantic_text_field_caps
3535
FROM semantic_text METADATA _score
3636
| EVAL query_embedding = TEXT_EMBEDDING("be excellent to each other", "test_dense_inference")
3737
| WHERE KNN(semantic_text_dense_field, query_embedding)
38-
| KEEP semantic_text_field, query_embedding, _score
39-
| EVAL _score = ROUND(_score, 4)
4038
| SORT _score DESC
4139
| LIMIT 10
40+
| KEEP semantic_text_field, query_embedding
4241
;
4342

44-
semantic_text_field:text | query_embedding:dense_vector | _score:double
45-
be excellent to each other | [45.0, 55.0, 54.0] | 1.0
46-
live long and prosper | [45.0, 55.0, 54.0] | 0.0295
47-
all we have to decide is what to do with the time that is given to us | [45.0, 55.0, 54.0] | 0.0214
43+
semantic_text_field:text | query_embedding:dense_vector
44+
be excellent to each other | [45.0, 55.0, 54.0]
45+
live long and prosper | [45.0, 55.0, 54.0]
46+
all we have to decide is what to do with the time that is given to us | [45.0, 55.0, 54.0]
4847
;
4948

5049
text_embedding with knn (inline) on semantic_text_dense_field
@@ -55,16 +54,15 @@ required_capability: semantic_text_field_caps
5554

5655
FROM semantic_text METADATA _score
5756
| WHERE KNN(semantic_text_dense_field, TEXT_EMBEDDING("be excellent to each other", "test_dense_inference"))
58-
| KEEP semantic_text_field, _score
59-
| EVAL _score = ROUND(_score, 4)
6057
| SORT _score DESC
6158
| LIMIT 10
59+
| KEEP semantic_text_field
6260
;
6361

64-
semantic_text_field:text | _score:double
65-
be excellent to each other | 1.0
66-
live long and prosper | 0.0295
67-
all we have to decide is what to do with the time that is given to us | 0.0214
62+
semantic_text_field:text
63+
be excellent to each other
64+
live long and prosper
65+
all we have to decide is what to do with the time that is given to us
6866
;
6967

7068

@@ -78,17 +76,16 @@ required_capability: semantic_text_field_caps
7876
FROM semantic_text METADATA _score
7977
| FORK (EVAL query_embedding = TEXT_EMBEDDING("be excellent to each other", "test_dense_inference") | WHERE KNN(semantic_text_dense_field, query_embedding))
8078
(EVAL query_embedding = TEXT_EMBEDDING("live long and prosper", "test_dense_inference") | WHERE KNN(semantic_text_dense_field, query_embedding))
81-
| KEEP semantic_text_field, query_embedding, _score, _fork
82-
| EVAL _score = ROUND(_score, 4)
83-
| SORT _score DESC
79+
| SORT _score DESC, _fork ASC
8480
| LIMIT 10
81+
| KEEP semantic_text_field, query_embedding, _fork
8582
;
8683

87-
semantic_text_field:text | query_embedding:dense_vector | _fork:keyword | _score:double
88-
be excellent to each other | [45.0, 55.0, 54.0] | fork1 | 1.0
89-
live long and prosper | [50.0, 57.0, 56.0] | fork2 | 1.0
90-
be excellent to each other | [50.0, 57.0, 56.0] | fork2 | 0.0295
91-
live long and prosper | [45.0, 55.0, 54.0] | fork1 | 0.0295
92-
all we have to decide is what to do with the time that is given to us | [45.0, 55.0, 54.0] | fork1 | 0.0214
93-
all we have to decide is what to do with the time that is given to us | [50.0, 57.0, 56.0] | fork2 | 0.0109
84+
semantic_text_field:text | query_embedding:dense_vector | _fork:keyword
85+
be excellent to each other | [45.0, 55.0, 54.0] | fork1
86+
live long and prosper | [50.0, 57.0, 56.0] | fork2
87+
live long and prosper | [45.0, 55.0, 54.0] | fork1
88+
be excellent to each other | [50.0, 57.0, 56.0] | fork2
89+
all we have to decide is what to do with the time that is given to us | [45.0, 55.0, 54.0] | fork1
90+
all we have to decide is what to do with the time that is given to us | [50.0, 57.0, 56.0] | fork2
9491
;

0 commit comments

Comments
 (0)