creating the new DemoChatBedrock POC.

langchain-ai · Jan 29, 2025 · 33664fb · 33664fb
1 parent ac7ec07
commit 33664fb
Show file tree

Hide file tree

Showing 8 changed files with 1,356 additions and 2 deletions.
diff --git a/libs/aws/langchain_aws/__init__.py b/libs/aws/langchain_aws/__init__.py
@@ -1,8 +1,9 @@
+from langchain_aws.chat_model_adapter import BedrockClaudeAdapter, BedrockLlamaAdapter, ModelAdapter
 from langchain_aws.chains import (
     create_neptune_opencypher_qa_chain,
     create_neptune_sparql_qa_chain,
 )
-from langchain_aws.chat_models import ChatBedrock, ChatBedrockConverse
+from langchain_aws.chat_models import ChatBedrock, ChatBedrockConverse, DemoChatBedrock
 from langchain_aws.embeddings import BedrockEmbeddings
 from langchain_aws.graphs import NeptuneAnalyticsGraph, NeptuneGraph
 from langchain_aws.llms import BedrockLLM, SagemakerEndpoint
@@ -20,6 +21,10 @@
     "BedrockLLM",
     "ChatBedrock",
     "ChatBedrockConverse",
+    "DemoChatBedrock",
+    "ModelAdapter",
+    "BedrockClaudeAdapter",
+    "BedrockLlamaAdapter",
     "SagemakerEndpoint",
     "AmazonKendraRetriever",
     "AmazonKnowledgeBasesRetriever",

diff --git a/libs/aws/langchain_aws/chat_model_adapter/__init__.py b/libs/aws/langchain_aws/chat_model_adapter/__init__.py
@@ -0,0 +1,5 @@
+from langchain_aws.chat_model_adapter.demo_chat_adapter import ModelAdapter
+from langchain_aws.chat_model_adapter.anthropic_adapter import BedrockClaudeAdapter
+from langchain_aws.chat_model_adapter.llama_adapter import BedrockLlamaAdapter
+
+__all__ = ["ModelAdapter", "BedrockClaudeAdapter", "BedrockLlamaAdapter"]