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

Adjust Bedrok reponse API models

Browse Source
This commit is contained in:
Christian Tzolov
2024-10-08 13:50:27 +02:00
parent d4ab868921
commit d50e7e2a8d
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/cohere/api/CohereChatBedrockApi.java
View File

@@ -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<TokenLikelihood> tokenLikelihoods,
@JsonProperty("finish_reason") FinishReason finishReason,

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

@@ -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<String, Object> embeddingsByType,
@JsonProperty("message") Object message) {
}
/**