Skip to content

Commit 352c0d8

Browse files
committed
Update TitanEmbeddingBedrockApi for TitanEmbeddingResponse
- Update the fields successCount, failureCount and results Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
1 parent 3409c1f commit 352c0d8

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/titan/api/TitanEmbeddingBedrockApi.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,10 @@ public TitanEmbeddingRequest build() {
177177
public record TitanEmbeddingResponse(
178178
@JsonProperty("embedding") float[] embedding,
179179
@JsonProperty("inputTextTokenCount") Integer inputTextTokenCount,
180+
@JsonProperty("successCount") Integer successCount,
181+
@JsonProperty("failureCount") Integer failureCount,
180182
@JsonProperty("embeddingsByType") Map<String, Object> embeddingsByType,
183+
@JsonProperty("results") Object results,
181184
@JsonProperty("message") Object message) {
182185

183186

0 commit comments

Comments
 (0)