From 1e5aa0c2cf89107e7e44270b51c452fc228ca21e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 21 Aug 2024 23:09:12 -0300 Subject: [PATCH] feat: Add import check for langchain_aws module in AmazonBedrockComponent (#3492) --- .../base/langflow/components/models/AmazonBedrockModel.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/models/AmazonBedrockModel.py b/src/backend/base/langflow/components/models/AmazonBedrockModel.py index b34f7ce7f..8f39bfa08 100644 --- a/src/backend/base/langflow/components/models/AmazonBedrockModel.py +++ b/src/backend/base/langflow/components/models/AmazonBedrockModel.py @@ -1,5 +1,3 @@ -from langchain_aws import ChatBedrock - from langflow.base.models.model import LCModelComponent from langflow.field_typing import LanguageModel from langflow.inputs import MessageTextInput, SecretStrInput @@ -60,6 +58,10 @@ class AmazonBedrockComponent(LCModelComponent): ] def build_model(self) -> LanguageModel: # type: ignore[type-var] + try: + from langchain_aws import ChatBedrock + except ImportError: + raise ImportError("langchain_aws is not installed. Please install it with `pip install langchain_aws`.") if self.aws_access_key: import boto3 # type: ignore