diff --git a/src/backend/base/langflow/base/models/model_input_constants.py b/src/backend/base/langflow/base/models/model_input_constants.py index 83099d44e..ae9b79133 100644 --- a/src/backend/base/langflow/base/models/model_input_constants.py +++ b/src/backend/base/langflow/base/models/model_input_constants.py @@ -1,4 +1,5 @@ from langflow.base.models.model import LCModelComponent +from langflow.components.models.amazon_bedrock import AmazonBedrockComponent from langflow.components.models.anthropic import AnthropicModelComponent from langflow.components.models.azure_openai import AzureChatOpenAIComponent from langflow.components.models.groq import GroqModel @@ -29,7 +30,7 @@ AZURE_INPUTS = get_filtered_inputs(AzureChatOpenAIComponent) GROQ_INPUTS = get_filtered_inputs(GroqModel) ANTHROPIC_INPUTS = get_filtered_inputs(AnthropicModelComponent) NVIDIA_INPUTS = get_filtered_inputs(NVIDIAModelComponent) - +AMAZON_BEDROCK_INPUTS = get_filtered_inputs(AmazonBedrockComponent) OPENAI_FIELDS = {input_.name: input_ for input_ in OPENAI_INPUTS} @@ -38,8 +39,9 @@ AZURE_FIELDS = create_input_fields_dict(AZURE_INPUTS, "azure") GROQ_FIELDS = create_input_fields_dict(GROQ_INPUTS, "groq") ANTHROPIC_FIELDS = create_input_fields_dict(ANTHROPIC_INPUTS, "anthropic") NVIDIA_FIELDS = create_input_fields_dict(NVIDIA_INPUTS, "nvidia") +AMAZON_BEDROCK_FIELDS = create_input_fields_dict(AMAZON_BEDROCK_INPUTS, "amazon_bedrock") -MODEL_PROVIDERS = ["Azure OpenAI", "OpenAI", "Groq", "Anthropic", "NVIDIA"] +MODEL_PROVIDERS = ["Azure OpenAI", "OpenAI", "Groq", "Anthropic", "NVIDIA", "Amazon Bedrock"] MODEL_PROVIDERS_DICT = { "Azure OpenAI": { @@ -67,5 +69,11 @@ MODEL_PROVIDERS_DICT = { "prefix": "nvidia_", "component_class": NVIDIAModelComponent(), }, + "Amazon Bedrock": { + "fields": AMAZON_BEDROCK_FIELDS, + "inputs": AMAZON_BEDROCK_INPUTS, + "prefix": "amazon_bedrock_", + "component_class": AmazonBedrockComponent(), + }, } ALL_PROVIDER_FIELDS: list[str] = [field for provider in MODEL_PROVIDERS_DICT.values() for field in provider["fields"]]