diff options
| author | Bjørn Christian Seime <bjorn.christian@seime.no> | 2024-04-26 07:56:30 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-26 07:56:30 +0200 |
| commit | 95cccd4698808d320f41461697f74a1c9d161bac (patch) | |
| tree | cd9336653627e4f3274a29f5f9da435a97b2150c /config-model/src/test/java/com/yahoo/vespa/model/container | |
| parent | face477e177b1fdd5462a5c961e2d00d43895b97 (diff) | |
| parent | a374e90b5f95b3f3c533a4d0302ac0e66c32668f (diff) | |
Merge pull request #31049 from vespa-engine/jobergum/add-prepend-embedder-support
add prepend support to embedder
Diffstat (limited to 'config-model/src/test/java/com/yahoo/vespa/model/container')
| -rw-r--r-- | config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java b/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java index fb1e176f707..6f629d99c92 100644 --- a/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java +++ b/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java @@ -88,6 +88,8 @@ public class EmbedderTestCase { var tokenizerCfg = assertHuggingfaceTokenizerComponentPresent(cluster); assertEquals("https://my/url/tokenizer.json", modelReference(tokenizerCfg.model().get(0), "path").url().orElseThrow().value()); assertEquals(-1, tokenizerCfg.maxLength()); + assertEquals("Represent this sentence for searching relevant passages:", embedderCfg.prependQuery()); + assertEquals("passage:", embedderCfg.prependDocument()); } @Test @@ -101,6 +103,8 @@ public class EmbedderTestCase { var tokenizerCfg = assertHuggingfaceTokenizerComponentPresent(cluster); assertEquals("https://data.vespa.oath.cloud/onnx_models/multilingual-e5-base/tokenizer.json", modelReference(tokenizerCfg.model().get(0), "path").url().orElseThrow().value()); assertEquals(-1, tokenizerCfg.maxLength()); + assertEquals("Represent this sentence for searching relevant passages:", embedderCfg.prependQuery()); + assertEquals("passage:", embedderCfg.prependDocument()); } @Test |