Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

chore(bedrock): support missing model versions in Amazon Bedrock model #30324

Merged
merged 6 commits into from
May 27, 2024
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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