refactor: Update langchain-aws dependency to version 0.1.6

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 20:01:04 -03:00
commit 2b1d46e883
3 changed files with 25 additions and 8 deletions

View file

@ -1,4 +1,4 @@
from langchain_community.chat_models.bedrock import BedrockChat
from langchain_aws import ChatBedrock
from langflow.base.constants import STREAM_INFO_TEXT
from langflow.base.models.model import LCModelComponent
@ -87,7 +87,7 @@ class AmazonBedrockComponent(LCModelComponent):
cache = self.cache
stream = self.stream
try:
output = BedrockChat(
output = ChatBedrock(
credentials_profile_name=credentials_profile_name,
model_id=model_id,
region_name=region_name,