jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: aws components and model lists (#4560)

Browse source 
* models updated

removed image models,
It was removed earlier but some PR Messed up the list again.

* Update aws_constants.py

image embedd removed
This commit is contained in:
Edwin Jose 2024-11-13 08:07:22 -05:00 • committed by GitHub
commit 5bb588c5cb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 31 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

37
src/backend/base/langflow/base/models/aws_constants.py
View file

@ -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 = [

4
src/backend/base/langflow/components/embeddings/amazon_bedrock.py
View file

@ -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(