diff --git a/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/cohere/api/CohereChatBedrockApi.java b/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/cohere/api/CohereChatBedrockApi.java index 1e77a7451..9feb62eeb 100644 --- a/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/cohere/api/CohereChatBedrockApi.java +++ b/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/cohere/api/CohereChatBedrockApi.java @@ -322,6 +322,7 @@ public class CohereChatBedrockApi extends @JsonInclude(Include.NON_NULL) public record Generation( @JsonProperty("id") String id, + @JsonProperty("event_type") String event_type, @JsonProperty("likelihood") Float likelihood, @JsonProperty("token_likelihoods") List tokenLikelihoods, @JsonProperty("finish_reason") FinishReason finishReason, 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 9b725e325..01968c81c 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 @@ -17,6 +17,7 @@ package org.springframework.ai.bedrock.titan.api; import java.time.Duration; import java.util.List; +import java.util.Map; import com.fasterxml.jackson.annotation.JsonInclude; import com.fasterxml.jackson.annotation.JsonInclude.Include; @@ -26,6 +27,7 @@ import software.amazon.awssdk.auth.credentials.AwsCredentialsProvider; import software.amazon.awssdk.regions.Region; import org.springframework.ai.bedrock.api.AbstractBedrockApi; +import org.springframework.ai.bedrock.titan.api.TitanEmbeddingBedrockApi.TitanEmbeddingModel; import org.springframework.ai.bedrock.titan.api.TitanEmbeddingBedrockApi.TitanEmbeddingRequest; import org.springframework.ai.bedrock.titan.api.TitanEmbeddingBedrockApi.TitanEmbeddingResponse; import org.springframework.util.Assert; @@ -139,7 +141,10 @@ public class TitanEmbeddingBedrockApi extends public record TitanEmbeddingResponse( @JsonProperty("embedding") float[] embedding, @JsonProperty("inputTextTokenCount") Integer inputTextTokenCount, + @JsonProperty("embeddingsByType") Map embeddingsByType, @JsonProperty("message") Object message) { + + } /**