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

Add AWS Bedrock Amazon Titan Text Premier model support

Browse Source
This commit is contained in:
wmz7year
2024-05-08 08:52:30 +08:00
committed by Christian Tzolov
parent 3497a1ec7b
commit 25f91c3297
3 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -275,7 +275,12 @@ public class TitanChatBedrockApi extends
/**
* amazon.titan-text-express-v1
*/
TITAN_TEXT_EXPRESS_V1("amazon.titan-text-express-v1");
TITAN_TEXT_EXPRESS_V1("amazon.titan-text-express-v1"),
/**
* amazon.titan-text-premier-v1:0
*/
TITAN_TEXT_PREMIER_V1("amazon.titan-text-premier-v1:0");
private final String id;

2
models/spring-ai-bedrock/src/test/java/org/springframework/ai/bedrock/titan/BedrockTitanChatClientIT.java
View File

@@ -205,7 +205,7 @@ class BedrockTitanChatClientIT {
@Bean
public TitanChatBedrockApi titanApi() {
return new TitanChatBedrockApi(TitanChatModel.TITAN_TEXT_EXPRESS_V1.id(),
return new TitanChatBedrockApi(TitanChatModel.TITAN_TEXT_PREMIER_V1.id(),
EnvironmentVariableCredentialsProvider.create(), Region.US_EAST_1.id(), new ObjectMapper(),
Duration.ofMinutes(2));
}

2
spring-ai-docs/src/main/antora/modules/ROOT/pages/api/chat/bedrock/bedrock-titan.adoc
View File

@@ -80,7 +80,7 @@ The prefix `spring.ai.bedrock.titan.chat` is the property prefix that configures
|====
Look at the https://github.com/spring-projects/spring-ai/blob/4839a6175cd1ec89498b97d3efb6647022c3c7cb/models/spring-ai-bedrock/src/main/java/org/springframework/ai/bedrock/titan/api/TitanChatBedrockApi.java#L220[TitanChatBedrockApi#TitanChatModel] for other model IDs.
Supported values are: `amazon.titan-text-lite-v1` and `amazon.titan-text-express-v1`.
Supported values are: `amazon.titan-text-lite-v1`, `amazon.titan-text-express-v1` and `amazon.titan-text-premier-v1:0`.
Model ID values can also be found in the https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids-arns.html[AWS Bedrock documentation for base model IDs].
TIP: All properties prefixed with `spring.ai.bedrock.titan.chat.options` can be overridden at runtime by adding a request specific <<chat-options>> to the `Prompt` call.