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

feat: Add import check for langchain_aws module in AmazonBedrockComponent (#3492)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-08-21 23:09:12 -03:00 • committed by GitHub
commit 1e5aa0c2cf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/components/models/AmazonBedrockModel.py
View file

@ -1,5 +1,3 @@
from langchain_aws import ChatBedrock
from langflow.base.models.model import LCModelComponent from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel from langflow.field_typing import LanguageModel
from langflow.inputs import MessageTextInput, SecretStrInput from langflow.inputs import MessageTextInput, SecretStrInput
@ -60,6 +58,10 @@ class AmazonBedrockComponent(LCModelComponent):
] ]
def build_model(self) -> LanguageModel: # type: ignore[type-var] 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: if self.aws_access_key:
import boto3 # type: ignore import boto3 # type: ignore