Pass destination

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

2 files changed, 14 insertions, 8 deletions

diff --git a/linguistics/abi-spec.json b/linguistics/abi-spec.json
index d257c451739..5865c28bbb6 100644
--- a/linguistics/abi-spec.json
+++ b/linguistics/abi-spec.json
@@ -338,8 +338,8 @@
     ],
     "methods": [
       "public void <init>()",
-      "public java.util.List embed(java.lang.String, com.yahoo.language.Language)",
-      "public com.yahoo.tensor.Tensor embed(java.lang.String, com.yahoo.language.Language, com.yahoo.tensor.TensorType)"
+      "public java.util.List embed(java.lang.String, com.yahoo.language.Language, java.lang.String)",
+      "public com.yahoo.tensor.Tensor embed(java.lang.String, com.yahoo.language.Language, java.lang.String, com.yahoo.tensor.TensorType)"
     ],
     "fields": []
   },
@@ -352,8 +352,8 @@
       "abstract"
     ],
     "methods": [
-      "public abstract java.util.List embed(java.lang.String, com.yahoo.language.Language)",
-      "public abstract com.yahoo.tensor.Tensor embed(java.lang.String, com.yahoo.language.Language, com.yahoo.tensor.TensorType)"
+      "public abstract java.util.List embed(java.lang.String, com.yahoo.language.Language, java.lang.String)",
+      "public abstract com.yahoo.tensor.Tensor embed(java.lang.String, com.yahoo.language.Language, java.lang.String, com.yahoo.tensor.TensorType)"
     ],
     "fields": [
       "public static final com.yahoo.language.process.Embedder throwsOnUse"
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");
         }