Skip to content

Commit 96068bb

Browse files
committed
Update text embedding CSV dataset to use dense_vector_text dataset.
1 parent 56fcea6 commit 96068bb

File tree

1 file changed

+19
-24
lines changed

1 file changed

+19
-24
lines changed

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

Lines changed: 19 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -26,44 +26,42 @@ Who is Victor Hugo? | [56.0, 50.0, 48.0]
2626
;
2727

2828

29-
text_embedding with knn on semantic_text_dense_field
29+
text_embedding with knn on dense vector field
3030
required_capability: text_embedding_function
3131
required_capability: dense_vector_field_type_released
3232
required_capability: knn_function_v5
33-
required_capability: semantic_text_field_caps
3433

3534
// tag::text-embedding-knn[]
36-
FROM semantic_text METADATA _score
35+
FROM dense_vector_text METADATA _score
3736
| EVAL query_embedding = TEXT_EMBEDDING("be excellent to each other", "test_dense_inference")
38-
| WHERE KNN(semantic_text_dense_field, query_embedding)
37+
| WHERE KNN(text_embedding_field, query_embedding)
3938
// end::text-embedding-knn[]
4039
| SORT _score DESC
4140
| LIMIT 10
42-
| KEEP semantic_text_field, query_embedding
41+
| KEEP text_field, query_embedding
4342
;
4443

45-
semantic_text_field:text | query_embedding:dense_vector
44+
text_field:text | query_embedding:dense_vector
4645
be excellent to each other | [45.0, 55.0, 54.0]
4746
live long and prosper | [45.0, 55.0, 54.0]
4847
all we have to decide is what to do with the time that is given to us | [45.0, 55.0, 54.0]
4948
;
5049

51-
text_embedding with knn (inline) on semantic_text_dense_field
50+
text_embedding with knn (inline) on dense vector field
5251
required_capability: text_embedding_function
5352
required_capability: dense_vector_field_type_released
5453
required_capability: knn_function_v5
55-
required_capability: semantic_text_field_caps
5654

5755
// tag::text-embedding-knn-inline[]
58-
FROM semantic_text METADATA _score
59-
| WHERE KNN(semantic_text_dense_field, TEXT_EMBEDDING("be excellent to each other", "test_dense_inference"))
56+
FROM dense_vector_text METADATA _score
57+
| WHERE KNN(text_embedding_field, TEXT_EMBEDDING("be excellent to each other", "test_dense_inference"))
6058
// end::text-embedding-knn-inline[]
6159
| SORT _score DESC
6260
| LIMIT 10
63-
| KEEP semantic_text_field
61+
| KEEP text_field
6462
;
6563

66-
semantic_text_field:text
64+
text_field:text
6765
be excellent to each other
6866
live long and prosper
6967
all we have to decide is what to do with the time that is given to us
@@ -73,17 +71,15 @@ text_embedding with multiple knn queries
7371
required_capability: text_embedding_function
7472
required_capability: dense_vector_field_type_released
7573
required_capability: knn_function_v5
76-
required_capability: fork_v9
77-
required_capability: semantic_text_field_caps
7874

79-
FROM semantic_text METADATA _score, _id
80-
| WHERE KNN(semantic_text_dense_field, TEXT_EMBEDDING("be excellent to each other", "test_dense_inference")) OR KNN(semantic_text_dense_field, TEXT_EMBEDDING("live long and prosper", "test_dense_inference"))
75+
FROM dense_vector_text METADATA _score, _id
76+
| WHERE KNN(text_embedding_field, TEXT_EMBEDDING("be excellent to each other", "test_dense_inference")) OR KNN(text_embedding_field, TEXT_EMBEDDING("live long and prosper", "test_dense_inference"))
8177
| SORT _score DESC, _id
8278
| LIMIT 10
83-
| KEEP semantic_text_field
79+
| KEEP text_field
8480
;
8581

86-
semantic_text_field:text
82+
text_field:text
8783
live long and prosper
8884
be excellent to each other
8985
all we have to decide is what to do with the time that is given to us
@@ -94,17 +90,16 @@ required_capability: text_embedding_function
9490
required_capability: dense_vector_field_type_released
9591
required_capability: knn_function_v5
9692
required_capability: fork_v9
97-
required_capability: semantic_text_field_caps
9893

99-
FROM semantic_text METADATA _score
100-
| FORK (EVAL query_embedding = TEXT_EMBEDDING("be excellent to each other", "test_dense_inference") | WHERE KNN(semantic_text_dense_field, query_embedding))
101-
(EVAL query_embedding = TEXT_EMBEDDING("live long and prosper", "test_dense_inference") | WHERE KNN(semantic_text_dense_field, query_embedding))
94+
FROM dense_vector_text METADATA _score
95+
| FORK (EVAL query_embedding = TEXT_EMBEDDING("be excellent to each other", "test_dense_inference") | WHERE KNN(text_embedding_field, query_embedding))
96+
(EVAL query_embedding = TEXT_EMBEDDING("live long and prosper", "test_dense_inference") | WHERE KNN(text_embedding_field, query_embedding))
10297
| SORT _score DESC, _fork ASC
10398
| LIMIT 10
104-
| KEEP semantic_text_field, query_embedding, _fork
99+
| KEEP text_field, query_embedding, _fork
105100
;
106101

107-
semantic_text_field:text | query_embedding:dense_vector | _fork:keyword
102+
text_field:text | query_embedding:dense_vector | _fork:keyword
108103
be excellent to each other | [45.0, 55.0, 54.0] | fork1
109104
live long and prosper | [50.0, 57.0, 56.0] | fork2
110105
live long and prosper | [45.0, 55.0, 54.0] | fork1

0 commit comments

Comments
 (0)