Add multiple embeddings for the same embedder in tests

This commit is contained in:
ManyTheFish 2025-02-12 16:13:34 +01:00
parent 8419ed52a1
commit 1caad4c4b0

View File

@ -99,10 +99,10 @@ async fn add_remove_embeddings() {
snapshot!(code, @"202 Accepted"); snapshot!(code, @"202 Accepted");
server.wait_task(response.uid()).await.succeeded(); server.wait_task(response.uid()).await.succeeded();
// 2 embedded documents for 4 embeddings in total // 2 embedded documents for 5 embeddings in total
let documents = json!([ let documents = json!([
{"id": 0, "name": "kefir", "_vectors": { "manual": [0, 0, 0], "handcrafted": [0, 0, 0] }}, {"id": 0, "name": "kefir", "_vectors": { "manual": [0, 0, 0], "handcrafted": [0, 0, 0] }},
{"id": 1, "name": "echo", "_vectors": { "manual": [1, 1, 1], "handcrafted": [1, 1, 1] }}, {"id": 1, "name": "echo", "_vectors": { "manual": [1, 1, 1], "handcrafted": [[1, 1, 1], [2, 2, 2]] }},
]); ]);
let (response, code) = index.add_documents(documents, None).await; let (response, code) = index.add_documents(documents, None).await;
@ -114,7 +114,7 @@ async fn add_remove_embeddings() {
{ {
"numberOfDocuments": 2, "numberOfDocuments": 2,
"isIndexing": false, "isIndexing": false,
"numberOfEmbeddings": 4, "numberOfEmbeddings": 5,
"numberOfEmbeddedDocuments": 2, "numberOfEmbeddedDocuments": 2,
"fieldDistribution": { "fieldDistribution": {
"id": 2, "id": 2,
@ -217,10 +217,10 @@ async fn add_remove_embedded_documents() {
snapshot!(code, @"202 Accepted"); snapshot!(code, @"202 Accepted");
server.wait_task(response.uid()).await.succeeded(); server.wait_task(response.uid()).await.succeeded();
// 2 embedded documents for 4 embeddings in total // 2 embedded documents for 5 embeddings in total
let documents = json!([ let documents = json!([
{"id": 0, "name": "kefir", "_vectors": { "manual": [0, 0, 0], "handcrafted": [0, 0, 0] }}, {"id": 0, "name": "kefir", "_vectors": { "manual": [0, 0, 0], "handcrafted": [0, 0, 0] }},
{"id": 1, "name": "echo", "_vectors": { "manual": [1, 1, 1], "handcrafted": [1, 1, 1] }}, {"id": 1, "name": "echo", "_vectors": { "manual": [1, 1, 1], "handcrafted": [[1, 1, 1], [2, 2, 2]] }},
]); ]);
let (response, code) = index.add_documents(documents, None).await; let (response, code) = index.add_documents(documents, None).await;
@ -232,7 +232,7 @@ async fn add_remove_embedded_documents() {
{ {
"numberOfDocuments": 2, "numberOfDocuments": 2,
"isIndexing": false, "isIndexing": false,
"numberOfEmbeddings": 4, "numberOfEmbeddings": 5,
"numberOfEmbeddedDocuments": 2, "numberOfEmbeddedDocuments": 2,
"fieldDistribution": { "fieldDistribution": {
"id": 2, "id": 2,
@ -241,7 +241,7 @@ async fn add_remove_embedded_documents() {
} }
"###); "###);
// delete one embedded document, remaining 1 embedded documents for 2 embeddings in total // delete one embedded document, remaining 1 embedded documents for 3 embeddings in total
let (response, code) = index.delete_document(0).await; let (response, code) = index.delete_document(0).await;
snapshot!(code, @"202 Accepted"); snapshot!(code, @"202 Accepted");
index.wait_task(response.uid()).await.succeeded(); index.wait_task(response.uid()).await.succeeded();
@ -251,7 +251,7 @@ async fn add_remove_embedded_documents() {
{ {
"numberOfDocuments": 1, "numberOfDocuments": 1,
"isIndexing": false, "isIndexing": false,
"numberOfEmbeddings": 2, "numberOfEmbeddings": 3,
"numberOfEmbeddedDocuments": 1, "numberOfEmbeddedDocuments": 1,
"fieldDistribution": { "fieldDistribution": {
"id": 1, "id": 1,