Skip to content

Commit 70d25f0

Browse files
committed
Update __init__.py files for easier import
1 parent 2382be2 commit 70d25f0

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

libs/aws/langchain_aws/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from langchain_aws.chat_models import BedrockChat, ChatBedrock
1+
from langchain_aws.chat_models import BedrockChat, ChatBedrock, ChatBedrockConverse
22
from langchain_aws.embeddings import BedrockEmbeddings
33
from langchain_aws.graphs import NeptuneAnalyticsGraph, NeptuneGraph
44
from langchain_aws.llms import Bedrock, BedrockLLM, SagemakerEndpoint
@@ -13,6 +13,7 @@
1313
"BedrockLLM",
1414
"BedrockChat",
1515
"ChatBedrock",
16+
"ChatBedrockConverse",
1617
"SagemakerEndpoint",
1718
"AmazonKendraRetriever",
1819
"AmazonKnowledgeBasesRetriever",
Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
from langchain_aws.chat_models.bedrock import BedrockChat, ChatBedrock
2+
from langchain_aws.chat_models.bedrock_converse import ChatBedrockConverse
23

3-
__all__ = ["BedrockChat", "ChatBedrock"]
4+
__all__ = ["BedrockChat", "ChatBedrock", "ChatBedrockConverse"]

0 commit comments

Comments
 (0)