|
18 | 18 | import com.meilisearch.sdk.Index;
|
19 | 19 | import com.meilisearch.sdk.model.EmbedderSource;
|
20 | 20 | import com.meilisearch.sdk.model.Embedder;
|
| 21 | +import com.meilisearch.sdk.model.EmbedderDistribution; |
21 | 22 | import com.meilisearch.sdk.model.FacetSortValue;
|
22 | 23 | import com.meilisearch.sdk.model.Faceting;
|
23 | 24 | import com.meilisearch.sdk.model.LocalizedAttribute;
|
@@ -1504,7 +1505,7 @@ public void testUpdateEmbeddersSettings() throws Exception {
|
1504 | 1505 | .setSource(EmbedderSource.HUGGING_FACE)
|
1505 | 1506 | .setModel("sentence-transformers/all-MiniLM-L6-v2")
|
1506 | 1507 | .setRevision("main")
|
1507 |
| - .setDistribution("uniform"); |
| 1508 | + .setDistribution(EmbedderDistribution.uniform()); |
1508 | 1509 |
|
1509 | 1510 | // Test REST embedder with request and response
|
1510 | 1511 | Map<String, Object> request = new HashMap<>();
|
@@ -1569,7 +1570,8 @@ public void testUpdateEmbeddersSettings() throws Exception {
|
1569 | 1570 | assertThat(retrievedHf.getSource(), is(equalTo(EmbedderSource.HUGGING_FACE)));
|
1570 | 1571 | assertThat(retrievedHf.getModel(), is(equalTo("sentence-transformers/all-MiniLM-L6-v2")));
|
1571 | 1572 | assertThat(retrievedHf.getRevision(), is(equalTo("main")));
|
1572 |
| - assertThat(retrievedHf.getDistribution(), is(equalTo("uniform"))); |
| 1573 | + assertThat(retrievedHf.getDistribution().getMean(), is(equalTo(0.5))); |
| 1574 | + assertThat(retrievedHf.getDistribution().getSigma(), is(equalTo(0.5))); |
1573 | 1575 |
|
1574 | 1576 | // Check REST embedder
|
1575 | 1577 | Embedder retrievedRest = updatedEmbedders.get("rest");
|
|
0 commit comments