Pass destination

This allows embedders to switch on it to enable bucket testing
and similar.

1 files changed, 10 insertions, 4 deletions

diff --git a/linguistics/src/main/java/com/yahoo/language/process/Embedder.java b/linguistics/src/main/java/com/yahoo/language/process/Embedder.java
index 56c401a7c61..1f4473220d7 100644
--- a/linguistics/src/main/java/com/yahoo/language/process/Embedder.java
+++ b/linguistics/src/main/java/com/yahoo/language/process/Embedder.java
@@ -22,10 +22,13 @@ public interface Embedder {
      *
      * @param text the text to embed
      * @param language the language of the text, or UNKNOWN to use language independent embedding
+     * @param destination the name of the recipient of this tensor, either a query feature name
+     *                    ("query(feature)"), or a schema and field name concatenated by a dot ("schema.field").
+     *                    This is useful for embedder components that alters behavior depending on the receiver.
      * @return the text embedded as a list of token ids
      * @throws IllegalArgumentException if the language is not supported by this embedder
      */
-    List<Integer> embed(String text, Language language);
+    List<Integer> embed(String text, Language language, String destination);
 
     /**
      * Converts text into tokens in a tensor.
@@ -33,21 +36,24 @@ public interface Embedder {
      *
      * @param text the text to embed
      * @param language the language of the text, or UNKNOWN to use language independent embedding
+     * @param destination the name of the recipient of this tensor, either a query feature name
+     *                    ("query(feature)"), or a schema and field name concatenated by a dot ("schema.field").
+     *                    This is useful for embedder components that alters behavior depending on the receiver.
      * @param tensorType the type of the tensor to be returned
      * @return the tensor embedding of the text, as the spoecified tensor type
      * @throws IllegalArgumentException if the language or tensor type is not supported by this embedder
      */
-    Tensor embed(String text, Language language, TensorType tensorType);
+    Tensor embed(String text, Language language, String destination, TensorType tensorType);
 
     class FailingEmbedder implements Embedder {
 
         @Override
-        public List<Integer> embed(String text, Language language) {
+        public List<Integer> embed(String text, Language language, String destination) {
             throw new IllegalStateException("No embedder has been configured");
         }
 
         @Override
-        public Tensor embed(String text, Language language, TensorType tensorType) {
+        public Tensor embed(String text, Language language, String destination, TensorType tensorType) {
             throw new IllegalStateException("No embedder has been configured");
         }