WIPlesters/external-llms

1 files changed, 6 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/ai/vespa/llm/InferenceParameters.java b/vespajlib/src/main/java/ai/vespa/llm/InferenceParameters.java
index 2ef1b97a8b2..8c47ae92e38 100755
--- a/vespajlib/src/main/java/ai/vespa/llm/InferenceParameters.java
+++ b/vespajlib/src/main/java/ai/vespa/llm/InferenceParameters.java
@@ -4,7 +4,9 @@ import ai.vespa.llm.completion.Prompt;
 
 import java.util.Objects;
 import java.util.Optional;
+import java.util.function.Consumer;
 import java.util.function.Function;
+import java.util.function.Supplier;
 
 public class InferenceParameters {
 
@@ -44,5 +46,9 @@ public class InferenceParameters {
         }
     }
 
+    public void ifPresent(String option, Consumer<String> func) {
+        get(option).ifPresent(func);
+    }
+
 }