forgotten addDeltas

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/model/OpenAiChatCompletionOutput.java

@@ -34,6 +34,7 @@ public class OpenAiChatCompletionOutput extends OpenAiCompletionOutput
    * @param delta the delta to add.
    */
   public void addDelta(OpenAiDeltaChatCompletion delta) {
+    super.addDelta(delta);
 
     if (delta.getSystemFingerprint() != null) {
       systemFingerprint = delta.getSystemFingerprint();

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/model/OpenAiCompletionOutput.java

@@ -43,4 +43,25 @@ public class OpenAiCompletionOutput {
   @JsonProperty("prompt_filter_results")
   @Getter(onMethod_ = @Nullable)
   private List<OpenAiPromptFilterResult> promptFilterResults;
+
+  void addDelta(OpenAiDeltaChatCompletion delta) {
+    created = delta.getCreated();
+    id = delta.getId();
+    model = delta.getModel();
+    object = delta.getObject();
+
+    if (delta.getUsage() != null) {
+      if (usage == null) {
+        usage = new OpenAiUsage();
+      }
+      usage.addDelta(delta.getUsage());
+    }
+
+    if (delta.getPromptFilterResults() != null) {
+      if (promptFilterResults == null) {
+        promptFilterResults = delta.getPromptFilterResults();
+      }
+      // prompt_filter_results is only present once in the first delta
+    }
+  }
 }

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/model/OpenAiUsage.java

@@ -27,4 +27,12 @@ public class OpenAiUsage {
   @JsonProperty("total_tokens")
   @Getter(onMethod_ = @Nonnull)
   private Integer totalTokens;
+
+  void addDelta(OpenAiUsage delta) {
+    if (delta.getCompletionTokens() != null) {
+      completionTokens = delta.getCompletionTokens();
+    }
+    promptTokens = delta.getPromptTokens();
+    totalTokens = delta.getTotalTokens();
+  }
 }