Skip to content

Commit 5f3f9d5

Browse files
committed
Add basic csv tests.
1 parent db6fa4c commit 5f3f9d5

File tree

2 files changed

+38
-0
lines changed

2 files changed

+38
-0
lines changed

x-pack/plugin/esql/qa/testFixtures/src/main/java/org/elasticsearch/xpack/esql/CsvTestsDataLoader.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,6 +393,10 @@ public static void createInferenceEndpoints(RestClient client) throws IOExceptio
393393
createSparseEmbeddingInferenceEndpoint(client);
394394
}
395395

396+
if (clusterHasDenseEmbeddingInferenceEndpoint(client) == false) {
397+
createDenseEmbeddingInferenceEndpoint(client);
398+
}
399+
396400
if (clusterHasRerankInferenceEndpoint(client) == false) {
397401
createRerankInferenceEndpoint(client);
398402
}
@@ -404,6 +408,7 @@ public static void createInferenceEndpoints(RestClient client) throws IOExceptio
404408

405409
public static void deleteInferenceEndpoints(RestClient client) throws IOException {
406410
deleteSparseEmbeddingInferenceEndpoint(client);
411+
deleteDenseEmbeddingInferenceEndpoint(client);
407412
deleteRerankInferenceEndpoint(client);
408413
deleteCompletionInferenceEndpoint(client);
409414
}
@@ -427,6 +432,24 @@ public static boolean clusterHasSparseEmbeddingInferenceEndpoint(RestClient clie
427432
return clusterHasInferenceEndpoint(client, TaskType.SPARSE_EMBEDDING, "test_sparse_inference");
428433
}
429434

435+
public static void createDenseEmbeddingInferenceEndpoint(RestClient client) throws IOException {
436+
createInferenceEndpoint(client, TaskType.TEXT_EMBEDDING, "test_dense_inference", """
437+
{
438+
"service": "text_embedding_test_service",
439+
"service_settings": { "model": "my_model", "api_key": "abc64", "dimensions": 10 },
440+
"task_settings": { }
441+
}
442+
""");
443+
}
444+
445+
public static void deleteDenseEmbeddingInferenceEndpoint(RestClient client) throws IOException {
446+
deleteInferenceEndpoint(client, "test_dense_inference");
447+
}
448+
449+
public static boolean clusterHasDenseEmbeddingInferenceEndpoint(RestClient client) throws IOException {
450+
return clusterHasInferenceEndpoint(client, TaskType.TEXT_EMBEDDING, "test_dense_inference");
451+
}
452+
430453
public static void createRerankInferenceEndpoint(RestClient client) throws IOException {
431454
createInferenceEndpoint(client, TaskType.RERANK, "test_reranker", """
432455
{
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Note:
2+
// The "test_completion" service returns the prompt in uppercase, making the output easy to guess.
3+
4+
5+
completion using a ROW source operator
6+
required_capability: text_embedding_function
7+
required_capability: dense_vector_field_type
8+
9+
ROW input="Who is Victor Hugo?"
10+
| EVAL embedding = TEXT_EMBEDDING("Who is Victor Hugo?", "test_dense_inference")
11+
;
12+
13+
input:keyword | embedding:dense_vector
14+
Who is Victor Hugo? | [56.0, 50.0, 48.0, 50.0, 54.0, 52.0, 49.0, 51.0, 51.0, 56.0]
15+
;

0 commit comments

Comments
 (0)