Merge pull request #8 from 3coins/add-bedrock-llms

Add bedrock llms
langchain-ai · Apr 4, 2024 · 15b5895 · 15b5895
2 parents 2a803a4 + 4eae07c
commit 15b5895
Show file tree

Hide file tree

Showing 10 changed files with 2,238 additions and 2 deletions.
diff --git a/libs/aws/langchain_aws/__init__.py b/libs/aws/langchain_aws/__init__.py
@@ -1,10 +1,15 @@
-from langchain_aws.llms import SagemakerEndpoint
+from langchain_aws.chat_models import BedrockChat, ChatBedrock
+from langchain_aws.llms import Bedrock, BedrockLLM, SagemakerEndpoint
 from langchain_aws.retrievers import (
     AmazonKendraRetriever,
     AmazonKnowledgeBasesRetriever,
 )
 
 __all__ = [
+    "Bedrock",
+    "BedrockLLM",
+    "BedrockChat",
+    "ChatBedrock",
     "SagemakerEndpoint",
     "AmazonKendraRetriever",
     "AmazonKnowledgeBasesRetriever",

diff --git a/libs/aws/langchain_aws/chat_models/__init__.py b/libs/aws/langchain_aws/chat_models/__init__.py
@@ -0,0 +1,3 @@
+from langchain_aws.chat_models.bedrock import BedrockChat, ChatBedrock
+
+__all__ = ["BedrockChat", "ChatBedrock"]