creating the new DemoChatBedrock POC.

Author: Vishal Patil

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",

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"]