Support direct tensor rendering

1 files changed, 12 insertions, 3 deletions

diff --git a/vespaclient-java/src/main/java/com/yahoo/vespaget/ClientParameters.java b/vespaclient-java/src/main/java/com/yahoo/vespaget/ClientParameters.java
index 7e464431f9a..91837cb4b09 100644
--- a/vespaclient-java/src/main/java/com/yahoo/vespaget/ClientParameters.java
+++ b/vespaclient-java/src/main/java/com/yahoo/vespaget/ClientParameters.java
@@ -39,13 +39,15 @@ public class ClientParameters {
     public final boolean jsonOutput;
     // Output JSON tensors in short form
     public final boolean tensorShortForm;
-
+    // Output JSON tensorvalues directly
+    public final boolean tensorDirectValues;
 
     private ClientParameters(
             boolean help, Iterator<String> documentIds, boolean printIdsOnly,
             String fieldSet, String route, String cluster, String configId,
             boolean showDocSize, double timeout, boolean noRetry, int traceLevel,
-            DocumentProtocol.Priority priority, boolean jsonOutput, boolean tensorShortForm) {
+            DocumentProtocol.Priority priority, boolean jsonOutput, boolean tensorShortForm,
+            boolean tensorDirectValues) {
 
         this.help = help;
         this.documentIds = documentIds;
@@ -61,6 +63,7 @@ public class ClientParameters {
         this.priority = priority;
         this.jsonOutput = jsonOutput;
         this.tensorShortForm = tensorShortForm;
+        this.tensorDirectValues = tensorDirectValues;
     }
 
     public static class Builder {
@@ -78,6 +81,7 @@ public class ClientParameters {
         private DocumentProtocol.Priority priority;
         private boolean jsonOutput;
         private boolean tensorShortForm;
+        private boolean tensorDirectValues;
 
         public Builder setHelp(boolean help) {
             this.help = help;
@@ -149,10 +153,15 @@ public class ClientParameters {
             return this;
         }
 
+        public Builder setTensorDirectValues(boolean tensorDirectValues) {
+            this.tensorDirectValues = tensorDirectValues;
+            return this;
+        }
+
         public ClientParameters build() {
             return new ClientParameters(
                     help, documentIds, printIdsOnly, fieldSet, route, cluster, configId,
-                    showDocSize, timeout, noRetry, traceLevel, priority, jsonOutput, tensorShortForm);
+                    showDocSize, timeout, noRetry, traceLevel, priority, jsonOutput, tensorShortForm, tensorDirectValues);
         }
     }