From 352c0d83be14db6fba83b2dcf7f3774fca9c1d94 Mon Sep 17 00:00:00 2001
From: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
Date: Thu, 10 Apr 2025 19:40:22 +0100
Subject: [PATCH] Update TitanEmbeddingBedrockApi for TitanEmbeddingResponse

 - Update the fields successCount, failureCount and results

Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
---
 .../ai/bedrock/titan/api/TitanEmbeddingBedrockApi.java         | 3 +++
 1 file changed, 3 insertions(+)

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<String, Object> embeddingsByType,
+			@JsonProperty("results") Object results,
 			@JsonProperty("message") Object message) {