Skip to content

Commit

Permalink
chore(bedrock): support missing model versions in Amazon Bedrock model (
Browse files Browse the repository at this point in the history
#30324)

### Issue # (if applicable)

N/A

### Reason for this change



Compare with [the guide](https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html) and add missing models.


### Description of changes


Added the following models.

- Amazon: 
   - amazon.titan-text-lite-v1
   - amazon.titan-text-premier-v1:0
   - amazon.titan-embed-text-v2:0
- Cohere:
   - cohere.command-r-v1:0
   - cohere.command-r-plus-v1:0
- Meta
   - meta.llama3-8b-instruct-v1:0
   - meta.llama3-70b-instruct-v1:0
- Mistral AI
   - mistral.mistral-large-2402-v1:0

### Checklist
- [x] My code adheres to the [CONTRIBUTING GUIDE](https://github.com/aws/aws-cdk/blob/main/CONTRIBUTING.md) and [DESIGN GUIDELINES](https://github.com/aws/aws-cdk/blob/main/docs/DESIGN_GUIDELINES.md)
  • Loading branch information
yasomaru authored May 27, 2024
1 parent b17dfe7 commit c03cdab
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions packages/aws-cdk-lib/aws-bedrock/lib/foundation-model.ts
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,12 @@ export class FoundationModelIdentifier {
/** Base model "amazon.titan-text-express-v1". */
public static readonly AMAZON_TITAN_TEXT_G1_EXPRESS_V1 = new FoundationModelIdentifier('amazon.titan-text-express-v1');

/** Base model "amazon.titan-text-lite-v1". */
public static readonly AMAZON_TITAN_TEXT_G1_LITE_V1 = new FoundationModelIdentifier('amazon.titan-text-lite-v1');

/** Base model "amazon.titan-text-premier-v1:0". */
public static readonly AMAZON_TITAN_TEXT_PREMIER_V1 = new FoundationModelIdentifier('amazon.titan-text-premier-v1:0');

/** Base model "amazon.titan-embed-text-v1". */
public static readonly AMAZON_TITAN_EMBEDDINGS_G1_TEXT_V1 = new FoundationModelIdentifier('amazon.titan-embed-text-v1');

Expand All @@ -32,6 +38,9 @@ export class FoundationModelIdentifier {
/** Base model "amazon.titan-embed-g1-text-02". */
public static readonly AMAZON_TITAN_EMBED_G1_TEXT_02 = new FoundationModelIdentifier('amazon.titan-embed-g1-text-02');

/** Base model "amazon.titan-embed-text-v2:0". */
public static readonly AMAZON_TITAN_EMBED_TEXT_V2_0 = new FoundationModelIdentifier('amazon.titan-embed-text-v2:0');

/** Base model "amazon.titan-image-generator-v1". */
public static readonly AMAZON_TITAN_IMAGE_GENERATOR_G1_V1 = new FoundationModelIdentifier('amazon.titan-image-generator-v1');

Expand Down Expand Up @@ -125,6 +134,12 @@ export class FoundationModelIdentifier {
/** Base model "cohere.command-light-text-v14:7:4k". */
public static readonly COHERE_COMMAND_LIGHT_TEXT_V14_7_4K = new FoundationModelIdentifier('cohere.command-light-text-v14:7:4k');

/** Base model "cohere.command-r-v1:0". */
public static readonly COHERE_COMMAND_R_V1 = new FoundationModelIdentifier('ccohere.command-r-v1:0');

/** Base model "cohere.command-r-v1:0". */
public static readonly COHERE_COMMAND_R_PLUS_V1 = new FoundationModelIdentifier('cohere.command-r-plus-v1:0');

/** Base model "cohere.embed-english-v3". */
public static readonly COHERE_EMBED_ENGLISH_V3 = new FoundationModelIdentifier('cohere.embed-english-v3');

Expand Down Expand Up @@ -155,12 +170,21 @@ export class FoundationModelIdentifier {
/** Base model "meta.llama2-70b-chat-v1:0:4k". */
public static readonly META_LLAMA_2_70B_CHAT_V1_0_4K = new FoundationModelIdentifier('meta.llama2-70b-chat-v1:0:4k');

/** Base model "meta.llama3-8b-instruct-v1:0". */
public static readonly META_LLAMA_3_8B_INSTRUCT_V1 = new FoundationModelIdentifier('meta.llama3-8b-instruct-v1:0');

/** Base model "meta.llama3-70b-instruct-v1:0". */
public static readonly META_LLAMA_3_70_INSTRUCT_V1 = new FoundationModelIdentifier('meta.llama3-70b-instruct-v1:0');

/** Base model "mistral.mistral-7b-instruct-v0:2". */
public static readonly MISTRAL_MISTRAL_7B_INSTRUCT_V0_2 = new FoundationModelIdentifier('mistral.mistral-7b-instruct-v0:2');

/** Base model "mistral.mixtral-8x7b-instruct-v0:1". */
public static readonly MISTRAL_MIXTRAL_8X7B_INSTRUCT_V0_1 = new FoundationModelIdentifier('mistral.mixtral-8x7b-instruct-v0:1');

/** Base model "mistral.mistral-large-2402-v1:0". */
public static readonly MISTRAL_LARGE_V0_1 = new FoundationModelIdentifier('mistral.mistral-large-2402-v1:0');

/**
* Base model "stability.stable-diffusion-xl".
* @deprecated use latest version of the model
Expand Down

0 comments on commit c03cdab

Please sign in to comment.