diff --git a/src/backend/base/langflow/base/models/aws_constants.py b/src/backend/base/langflow/base/models/aws_constants.py index 6a2d39e90..23fd90dce 100644 --- a/src/backend/base/langflow/base/models/aws_constants.py +++ b/src/backend/base/langflow/base/models/aws_constants.py @@ -1,35 +1,56 @@ AWS_MODEL_IDs = [ + # Amazon Titan Models "amazon.titan-text-express-v1", "amazon.titan-text-lite-v1", "amazon.titan-text-premier-v1:0", - "amazon.titan-embed-text-v1", - "amazon.titan-embed-text-v2:0", - "amazon.titan-embed-image-v1", - "amazon.titan-image-generator-v1", + # Anthropic Models "anthropic.claude-v2", "anthropic.claude-v2:1", "anthropic.claude-3-sonnet-20240229-v1:0", + "anthropic.claude-3-5-sonnet-20240620-v1:0", + "anthropic.claude-3-5-sonnet-20241022-v2:0", "anthropic.claude-3-haiku-20240307-v1:0", + "anthropic.claude-3-5-haiku-20241022-v1:0", "anthropic.claude-3-opus-20240229-v1:0", "anthropic.claude-instant-v1", + # AI21 Labs Models + "ai21.jamba-instruct-v1:0", "ai21.j2-mid-v1", "ai21.j2-ultra-v1", + "ai21.jamba-1-5-large-v1:0", + "ai21.jamba-1-5-mini-v1:0", + # Cohere Models "cohere.command-text-v14", "cohere.command-light-text-v14", "cohere.command-r-v1:0", "cohere.command-r-plus-v1:0", - "cohere.embed-english-v3", - "cohere.embed-multilingual-v3", + # Meta Models "meta.llama2-13b-chat-v1", "meta.llama2-70b-chat-v1", "meta.llama3-8b-instruct-v1:0", "meta.llama3-70b-instruct-v1:0", + "meta.llama3-1-8b-instruct-v1:0", + "meta.llama3-1-70b-instruct-v1:0", + "meta.llama3-1-405b-instruct-v1:0", + "meta.llama3-2-1b-instruct-v1:0", + "meta.llama3-2-3b-instruct-v1:0", + "meta.llama3-2-11b-instruct-v1:0", + "meta.llama3-2-90b-instruct-v1:0", + # Mistral AI Models "mistral.mistral-7b-instruct-v0:2", "mistral.mixtral-8x7b-instruct-v0:1", "mistral.mistral-large-2402-v1:0", + "mistral.mistral-large-2407-v1:0", "mistral.mistral-small-2402-v1:0", - "stability.stable-diffusion-xl-v0", - "stability.stable-diffusion-xl-v1", +] + +AWS_EMBEDDING_MODEL_IDS = [ + # Amazon Titan Embedding Models + "amazon.titan-embed-text-v1", + "amazon.titan-embed-text-v2:0", + # Cohere Embedding Models + "cohere.embed-english-v3", + "cohere.embed-multilingual-v3", ] AWS_REGIONS = [ diff --git a/src/backend/base/langflow/components/embeddings/amazon_bedrock.py b/src/backend/base/langflow/components/embeddings/amazon_bedrock.py index ab368e5c1..caeafc91a 100644 --- a/src/backend/base/langflow/components/embeddings/amazon_bedrock.py +++ b/src/backend/base/langflow/components/embeddings/amazon_bedrock.py @@ -1,4 +1,4 @@ -from langflow.base.models.aws_constants import AWS_REGIONS +from langflow.base.models.aws_constants import AWS_EMBEDDING_MODEL_IDS, AWS_REGIONS from langflow.base.models.model import LCModelComponent from langflow.field_typing import Embeddings from langflow.inputs import SecretStrInput @@ -15,7 +15,7 @@ class AmazonBedrockEmbeddingsComponent(LCModelComponent): DropdownInput( name="model_id", display_name="Model Id", - options=["amazon.titan-embed-text-v1"], + options=AWS_EMBEDDING_MODEL_IDS, value="amazon.titan-embed-text-v1", ), SecretStrInput(