Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update TitanEmbeddingBedrockApi for TitanEmbeddingResponse

Browse Source 
- Update the fields successCount, failureCount and results

Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
This commit is contained in:
Ilayaperumal Gopinathan
2025-04-10 19:40:22 +01:00
parent 3409c1fd87
commit 352c0d83be
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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) {