feat: Add new fields to conversation api

Signed-off-by: Javier Aliaga <javier@diagrid.io>

Author: javier-aliaga

sdk/src/main/java/io/dapr/client/DaprClientImpl.java

@@ -42,7 +42,10 @@
 import io.dapr.client.domain.ConversationResponseAlpha2;
 import io.dapr.client.domain.ConversationResultAlpha2;
 import io.dapr.client.domain.ConversationResultChoices;
+import io.dapr.client.domain.ConversationResultCompletionUsage;
+import io.dapr.client.domain.ConversationResultCompletionUsageDetails;
 import io.dapr.client.domain.ConversationResultMessage;
+import io.dapr.client.domain.ConversationResultPromptUsageDetails;
 import io.dapr.client.domain.ConversationToolCalls;
 import io.dapr.client.domain.ConversationToolCallsOfFunction;
 import io.dapr.client.domain.ConversationTools;
@@ -1793,6 +1796,7 @@ public Mono<ConversationResponseAlpha2> converseAlpha2(ConversationRequestAlpha2
       DaprAiProtos.ConversationResponseAlpha2 conversationResponse = conversationResponseMono.block();
 
       assert conversationResponse != null;
+
       List<ConversationResultAlpha2> results = buildConversationResults(conversationResponse.getOutputsList());
       return Mono.just(new ConversationResponseAlpha2(conversationResponse.getContextId(), results));
     } catch (Exception ex) {
@@ -1857,6 +1861,33 @@ private DaprAiProtos.ConversationRequestAlpha2 buildConversationRequestProto(Con
 
       builder.addInputs(inputBuilder.build());
     }
+
+    if (request.getResponseFormat() != null) {
+      Map<String, Value> responseParams = request.getResponseFormat()
+          .entrySet().stream()
+          .collect(Collectors.toMap(
+              Map.Entry::getKey,
+              e -> {
+                try {
+                  return ProtobufValueHelper.toProtobufValue(e.getValue());
+                } catch (IOException ex) {
+                  throw new RuntimeException(ex);
+                }
+              }
+          ));
+
+      builder.setResponseFormat(Struct.newBuilder().putAllFields(responseParams).build());
+    }
+
+    if (request.getPromptCacheRetention() != null) {
+      Duration javaDuration = request.getPromptCacheRetention();
+      builder.setPromptCacheRetention(
+          com.google.protobuf.Duration.newBuilder()
+              .setSeconds(javaDuration.getSeconds())
+              .setNanos(javaDuration.getNano())
+              .build()
+      );
+    }
 
     return builder.build();
   }
@@ -1974,14 +2005,38 @@ private List<ConversationResultAlpha2> buildConversationResults(
       for (DaprAiProtos.ConversationResultChoices protoChoice : protoResult.getChoicesList()) {
         ConversationResultMessage message = buildConversationResultMessage(protoChoice);
         choices.add(new ConversationResultChoices(protoChoice.getFinishReason(), protoChoice.getIndex(), message));
-      }  
+      }
 
-      results.add(new ConversationResultAlpha2(choices));
+      results.add(new ConversationResultAlpha2(
+          choices,
+          protoResult.getModel(),
+          getConversationResultCompletionUsage(protoResult))
+      );
     }
 
     return results;
   }
 
+  private static ConversationResultCompletionUsage getConversationResultCompletionUsage(
+      DaprAiProtos.ConversationResultAlpha2 protoResult) {
+    var usage = new ConversationResultCompletionUsage(
+        protoResult.getUsage().getCompletionTokens(),
+        protoResult.getUsage().getPromptTokens(),
+        protoResult.getUsage().getTotalTokens());
+
+    usage.setCompletionTokenDetails(new ConversationResultCompletionUsageDetails(
+        protoResult.getUsage().getCompletionTokensDetails().getAcceptedPredictionTokens(),
+        protoResult.getUsage().getCompletionTokensDetails().getAudioTokens(),
+        protoResult.getUsage().getCompletionTokensDetails().getReasoningTokens(),
+        protoResult.getUsage().getCompletionTokensDetails().getRejectedPredictionTokens()));
+
+    usage.setPromptTokenDetails(new ConversationResultPromptUsageDetails(
+        protoResult.getUsage().getPromptTokensDetails().getAudioTokens(),
+        protoResult.getUsage().getPromptTokensDetails().getCachedTokens()
+    ));
+    return usage;
+  }
+
   private ConversationResultMessage buildConversationResultMessage(DaprAiProtos.ConversationResultChoices protoChoice) {
     if (!protoChoice.hasMessage()) {
       return null;

sdk/src/main/java/io/dapr/client/DaprPreviewClient.java

@@ -17,20 +17,15 @@
 import io.dapr.client.domain.BulkPublishRequest;
 import io.dapr.client.domain.BulkPublishResponse;
 import io.dapr.client.domain.BulkPublishResponseFailedEntry;
-import io.dapr.client.domain.CloudEvent;
 import io.dapr.client.domain.ConversationRequest;
 import io.dapr.client.domain.ConversationRequestAlpha2;
 import io.dapr.client.domain.ConversationResponse;
 import io.dapr.client.domain.ConversationResponseAlpha2;
 import io.dapr.client.domain.DecryptRequestAlpha1;
-import io.dapr.client.domain.DeleteJobRequest;
 import io.dapr.client.domain.EncryptRequestAlpha1;
-import io.dapr.client.domain.GetJobRequest;
-import io.dapr.client.domain.GetJobResponse;
 import io.dapr.client.domain.LockRequest;
 import io.dapr.client.domain.QueryStateRequest;
 import io.dapr.client.domain.QueryStateResponse;
-import io.dapr.client.domain.ScheduleJobRequest;
 import io.dapr.client.domain.UnlockRequest;
 import io.dapr.client.domain.UnlockResponseStatus;
 import io.dapr.client.domain.query.Query;

sdk/src/main/java/io/dapr/client/domain/ConversationOutput.java

@@ -13,7 +13,6 @@
 
 package io.dapr.client.domain;
 
-import java.util.Collections;
 import java.util.Map;
 
 /**

sdk/src/main/java/io/dapr/client/domain/ConversationRequestAlpha2.java

@@ -13,6 +13,7 @@
 
 package io.dapr.client.domain;
 
+import java.time.Duration;
 import java.util.List;
 import java.util.Map;
 
@@ -31,6 +32,8 @@ public class ConversationRequestAlpha2 {
   private String toolChoice;
   private Map<String, Object> parameters;
   private Map<String, String> metadata;
+  private Map<String, Object> responseFormat;
+  private Duration promptCacheRetention;
 
   /**
    * Constructs a ConversationRequestAlpha2 with a component name and conversation inputs.
@@ -206,4 +209,22 @@ public ConversationRequestAlpha2 setMetadata(Map<String, String> metadata) {
     this.metadata = metadata;
     return this;
   }
+
+  public Map<String, Object> getResponseFormat() {
+    return responseFormat;
+  }
+
+  public ConversationRequestAlpha2 setResponseFormat(Map<String, Object> responseFormat) {
+    this.responseFormat = responseFormat;
+    return this;
+  }
+
+  public Duration getPromptCacheRetention() {
+    return promptCacheRetention;
+  }
+
+  public ConversationRequestAlpha2 setPromptCacheRetention(Duration promptCacheRetention) {
+    this.promptCacheRetention = promptCacheRetention;
+    return this;
+  }
 }

sdk/src/main/java/io/dapr/client/domain/ConversationResponse.java

@@ -13,7 +13,6 @@
 
 package io.dapr.client.domain;
 
-import java.util.Collections;
 import java.util.List;
 
 /**

sdk/src/main/java/io/dapr/client/domain/ConversationResultAlpha2.java

@@ -21,14 +21,22 @@
 public class ConversationResultAlpha2 {
 
   private final List<ConversationResultChoices> choices;
+  private final String model;
+  private final ConversationResultCompletionUsage usage;
 
   /**
    * Constructor.
    *
    * @param choices the list of conversation result choices.
+   * @param model the model used for the conversation.
+   * @param usage the usage of the model.
    */
-  public ConversationResultAlpha2(List<ConversationResultChoices> choices) {
+  public ConversationResultAlpha2(List<ConversationResultChoices> choices,
+                                  String model,
+                                  ConversationResultCompletionUsage usage) {
     this.choices = List.copyOf(choices);
+    this.model = model;
+    this.usage = usage;
   }
 
   /**
@@ -39,4 +47,22 @@ public ConversationResultAlpha2(List<ConversationResultChoices> choices) {
   public List<ConversationResultChoices> getChoices() {
     return choices;
   }
+
+  /**
+   * Gets the model used for the conversation.
+   *
+   * @return the model used for the conversation.
+   */
+  public String getModel() {
+    return model;
+  }
+
+  /**
+   * Gets the usage of the model.
+   *
+   * @return the usage of the model.
+   */
+  public ConversationResultCompletionUsage getUsage() {
+    return usage;
+  }
 }

sdk/src/main/java/io/dapr/client/domain/ConversationResultCompletionUsage.java

@@ -0,0 +1,99 @@
+/*
+ * Copyright 2026 The Dapr Authors
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package io.dapr.client.domain;
+
+public class ConversationResultCompletionUsage {
+  private final long completionTokens;
+  private final long promptTokens;
+  private final long totalTokens;
+
+  private ConversationResultCompletionUsageDetails completionTokenDetails;
+  private ConversationResultPromptUsageDetails promptTokenDetails;
+
+  /**
+   * Constructor.
+   *
+   * @param completionTokens completion tokens used.
+   * @param promptTokens     prompt tokens used.
+   * @param totalTokens      total tokens used.
+   */
+  public ConversationResultCompletionUsage(long completionTokens, long promptTokens, long totalTokens) {
+    this.completionTokens = completionTokens;
+    this.promptTokens = promptTokens;
+    this.totalTokens = totalTokens;
+  }
+
+  /**
+   * Completion tokens used.
+   *
+   * @return completion tokens used.
+   */
+  public long getCompletionTokens() {
+    return completionTokens;
+  }
+
+  /**
+   * Prompt tokens used.
+   *
+   * @return prompt tokens used.
+   */
+  public long getPromptTokens() {
+    return promptTokens;
+  }
+
+  /**
+   * Total tokens used.
+   *
+   * @return total tokens used.
+   */
+  public long getTotalTokens() {
+    return totalTokens;
+  }
+
+  /**
+   * Completion token details.
+   *
+   * @return the completionTokenDetails
+   */
+  public ConversationResultCompletionUsageDetails getCompletionTokenDetails() {
+    return completionTokenDetails;
+  }
+
+  /**
+   * Completion token details.
+   *
+   * @param completionTokenDetails the completionTokenDetails to set
+   */
+  public void setCompletionTokenDetails(ConversationResultCompletionUsageDetails completionTokenDetails) {
+    this.completionTokenDetails = completionTokenDetails;
+  }
+
+  /**
+   * Prompt token details.
+   *
+   * @return the promptTokenDetails
+   */
+  public ConversationResultPromptUsageDetails getPromptTokenDetails() {
+    return promptTokenDetails;
+  }
+
+  /**
+   * Prompt token details.
+   *
+   * @param promptTokenDetails the promptTokenDetails to set
+   */
+  public void setPromptTokenDetails(ConversationResultPromptUsageDetails promptTokenDetails) {
+    this.promptTokenDetails = promptTokenDetails;
+  }
+}

sdk/src/main/java/io/dapr/client/domain/ConversationResultCompletionUsageDetails.java

@@ -0,0 +1,73 @@
+/*
+ * Copyright 2026 The Dapr Authors
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package io.dapr.client.domain;
+
+public class ConversationResultCompletionUsageDetails {
+  private final long acceptedPredictionTokens;
+  private final long audioTokens;
+  private final long reasoningTokens;
+  private final long rejectedPredictionTokens;
+
+  /**
+   * Constructor.
+   *
+   * @param acceptedPredictionTokens accepted prediction tokens used.
+   * @param audioTokens              audio tokens used.
+   * @param reasoningTokens          reasoning tokens used.
+   * @param rejectedPredictionTokens rejected prediction tokens used.
+   */
+  public ConversationResultCompletionUsageDetails(long acceptedPredictionTokens, long audioTokens,
+                                                  long reasoningTokens, long rejectedPredictionTokens) {
+    this.acceptedPredictionTokens = acceptedPredictionTokens;
+    this.audioTokens = audioTokens;
+    this.reasoningTokens = reasoningTokens;
+    this.rejectedPredictionTokens = rejectedPredictionTokens;
+  }
+
+  /**
+   * Accepted prediction tokens used.
+   *
+   * @return accepted prediction tokens used.
+   */
+  public long getAcceptedPredictionTokens() {
+    return acceptedPredictionTokens;
+  }
+
+  /**
+   * Audio tokens used.
+   *
+   * @return audio tokens used.
+   */
+  public long getAudioTokens() {
+    return audioTokens;
+  }
+
+  /**
+   * Reasoning tokens used.
+   *
+   * @return reasoning tokens used.
+   */
+  public long getReasoningTokens() {
+    return reasoningTokens;
+  }
+
+  /**
+   * Rejected prediction tokens used.
+   *
+   * @return rejected prediction tokens used.
+   */
+  public long getRejectedPredictionTokens() {
+    return rejectedPredictionTokens;
+  }
+}