support for stream and regular

Author: Vaijanath Rao

src/main/java/de/kherud/llama/LlamaOutput.java

@@ -3,6 +3,7 @@
 import org.jetbrains.annotations.NotNull;
 
 import java.nio.charset.StandardCharsets;
+import java.util.Collections;
 import java.util.Map;
 
 /**
@@ -24,16 +25,72 @@ public final class LlamaOutput {
     @NotNull
     public final Map<String, Float> probabilities;
 
+    /**
+     * Indicates whether this is the final output in the sequence
+     */
     final boolean stop;
 
+    /**
+     * Create a new LlamaOutput from raw bytes with probability information.
+     *
+     * @param generated The raw bytes of generated text
+     * @param probabilities Mapping of tokens to their probabilities
+     * @param stop Whether this is the final output
+     */
     LlamaOutput(byte[] generated, @NotNull Map<String, Float> probabilities, boolean stop) {
         this.text = new String(generated, StandardCharsets.UTF_8);
         this.probabilities = probabilities;
         this.stop = stop;
     }
 
+    /**
+     * Create a new LlamaOutput from a string with probability information.
+     *
+     * @param text The generated text as a string
+     * @param probabilities Mapping of tokens to their probabilities
+     * @param stop Whether this is the final output
+     */
+    public LlamaOutput(@NotNull String text, @NotNull Map<String, Float> probabilities, boolean stop) {
+        this.text = text;
+        this.probabilities = probabilities;
+        this.stop = stop;
+    }
+
+    /**
+     * Create a new LlamaOutput from a string with no probability information.
+     * This is useful for chat completions where token probabilities might not be available.
+     *
+     * @param text The generated text as a string
+     * @param stop Whether this is the final output
+     */
+    public LlamaOutput(@NotNull String text, boolean stop) {
+        this.text = text;
+        this.probabilities = Collections.emptyMap();
+        this.stop = stop;
+    }
+
+    /**
+     * Create a new LlamaOutput by wrapping a StreamingOutput object.
+     * 
+     * @param output The streaming output to wrap
+     */
+    public LlamaOutput(@NotNull StreamingOutput output) {
+        this.text = output.text;
+        this.probabilities = output.probabilities != null ? output.probabilities : Collections.emptyMap();
+        this.stop = output.isFinal;
+    }
+
+    /**
+     * Returns whether this is the final output in a sequence
+     * 
+     * @return true if this is the final output, false otherwise
+     */
+    public boolean isStop() {
+        return stop;
+    }
+
     @Override
     public String toString() {
         return text;
     }
-}
+}