diff --git a/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/titan/api/TitanEmbeddingBedrockApi.java b/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/titan/api/TitanEmbeddingBedrockApi.java index a96455a5b..f2a16579d 100644 --- a/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/titan/api/TitanEmbeddingBedrockApi.java +++ b/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/titan/api/TitanEmbeddingBedrockApi.java @@ -177,7 +177,10 @@ public class TitanEmbeddingBedrockApi extends public record TitanEmbeddingResponse( @JsonProperty("embedding") float[] embedding, @JsonProperty("inputTextTokenCount") Integer inputTextTokenCount, + @JsonProperty("successCount") Integer successCount, + @JsonProperty("failureCount") Integer failureCount, @JsonProperty("embeddingsByType") Map embeddingsByType, + @JsonProperty("results") Object results, @JsonProperty("message") Object message) {