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,6 +42,7 @@
 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.ConversationResultMessage;
 import io.dapr.client.domain.ConversationToolCalls;
 import io.dapr.client.domain.ConversationToolCallsOfFunction;
@@ -1793,6 +1794,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 +1859,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,9 +2003,16 @@ 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(),
+          new ConversationResultCompletionUsage(
+              protoResult.getUsage().getCompletionTokens(),
+              protoResult.getUsage().getPromptTokens(),
+              protoResult.getUsage().getTotalTokens()))
+      );
     }
 
     return results;

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,60 @@
+/*
+ * 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;
+
+  /**
+   * 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;
+  }
+}

sdk/src/test/java/io/dapr/client/DaprPreviewClientGrpcTest.java

@@ -17,6 +17,8 @@
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.google.protobuf.ByteString;
+import com.google.protobuf.Struct;
+import com.google.protobuf.Value;
 import io.dapr.client.domain.AssistantMessage;
 import io.dapr.client.domain.BulkPublishEntry;
 import io.dapr.client.domain.BulkPublishRequest;
@@ -43,6 +45,7 @@
 import io.dapr.client.domain.QueryStateRequest;
 import io.dapr.client.domain.QueryStateResponse;
 import io.dapr.client.domain.SystemMessage;
+import io.dapr.client.domain.TestData;
 import io.dapr.client.domain.ToolMessage;
 import io.dapr.client.domain.UnlockResponseStatus;
 import io.dapr.client.domain.UserMessage;
@@ -75,6 +78,7 @@
 
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
+import java.time.Duration;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
@@ -1061,6 +1065,9 @@ public void converseAlpha2ComplexRequestTest() {
     Map<String, Object> parameters = new HashMap<>();
     parameters.put("max_tokens", "1000");
 
+    var responseFormat = new HashMap<String, Object>();
+    responseFormat.put("temperature", 0.7);
+    responseFormat.put("data", new TestData("Peter", 40));
     ConversationRequestAlpha2 request = new ConversationRequestAlpha2("openai", List.of(input));
     request.setContextId("test-context");
     request.setTemperature(0.7);
@@ -1069,11 +1076,19 @@ public void converseAlpha2ComplexRequestTest() {
     request.setToolChoice("auto");
     request.setMetadata(metadata);
     request.setParameters(parameters);
+    request.setPromptCacheRetention(Duration.ofDays(1));
+    request.setResponseFormat(responseFormat);
 
     // Mock response with tool calls
     DaprAiProtos.ConversationResponseAlpha2 grpcResponse = DaprAiProtos.ConversationResponseAlpha2.newBuilder()
         .setContextId("test-context")
         .addOutputs(DaprAiProtos.ConversationResultAlpha2.newBuilder()
+            .setModel("gpt-3.5-turbo")
+            .setUsage(DaprAiProtos.ConversationResultAlpha2CompletionUsage.newBuilder()
+                .setPromptTokens(100)
+                .setCompletionTokens(100)
+                .setTotalTokens(200)
+                .build())
             .addChoices(DaprAiProtos.ConversationResultChoices.newBuilder()
                 .setFinishReason("tool_calls")
                 .setIndex(0)
@@ -1108,6 +1123,11 @@ public void converseAlpha2ComplexRequestTest() {
     assertEquals("tool_calls", choice.getFinishReason());
     assertEquals("I'll help you get the weather information.", choice.getMessage().getContent());
     assertEquals(1, choice.getMessage().getToolCalls().size());
+    assertEquals("gpt-3.5-turbo", response.getOutputs().get(0).getModel());
+    assertEquals(100, response.getOutputs().get(0).getUsage().getCompletionTokens());
+    assertEquals(100, response.getOutputs().get(0).getUsage().getPromptTokens());
+    assertEquals(200, response.getOutputs().get(0).getUsage().getTotalTokens());
+
 
     ConversationToolCalls toolCall = choice.getMessage().getToolCalls().get(0);
     assertEquals("call_123", toolCall.getId());
@@ -1128,6 +1148,13 @@ public void converseAlpha2ComplexRequestTest() {
     assertEquals("value1", capturedRequest.getMetadataMap().get("key1"));
     assertEquals(1, capturedRequest.getToolsCount());
     assertEquals("get_weather", capturedRequest.getTools(0).getFunction().getName());
+    assertEquals(Struct.newBuilder()
+            .putFields("temperature", Value.newBuilder().setNumberValue(0.7).build())
+            .putFields("data", Value.newBuilder().setStringValue("TestData{name='Peter', age=40}").build())
+            .build(),
+        capturedRequest.getResponseFormat());
+    assertEquals(Duration.ofDays(1).getSeconds(), capturedRequest.getPromptCacheRetention().getSeconds());
+    assertEquals(0, capturedRequest.getPromptCacheRetention().getNanos());
   }
 
   @Test

sdk/src/test/java/io/dapr/client/domain/TestData.java

@@ -0,0 +1,27 @@
+package io.dapr.client.domain;
+
+public class TestData {
+  private final String name;
+  private final int age;
+
+  public TestData(String name, int age) {
+    this.name = name;
+    this.age = age;
+  }
+
+  public int getAge() {
+    return age;
+  }
+
+  public String getName() {
+    return name;
+  }
+
+  @Override
+  public String toString() {
+    return "TestData{" +
+        "name='" + name + "'" +
+        ", age=" + age +
+        "}";
+  }
+}