diff --git a/libs/aws/langchain_aws/__init__.py b/libs/aws/langchain_aws/__init__.py index 68921e44..a19759fe 100644 --- a/libs/aws/langchain_aws/__init__.py +++ b/libs/aws/langchain_aws/__init__.py @@ -6,6 +6,7 @@ AmazonKendraRetriever, AmazonKnowledgeBasesRetriever, ) +from langchain_aws.vectorstores.documentdb import DocumentDBVectorSearch __all__ = [ "Bedrock", @@ -13,6 +14,7 @@ "BedrockLLM", "BedrockChat", "ChatBedrock", + "DocumentDBVectorSearch", "SagemakerEndpoint", "AmazonKendraRetriever", "AmazonKnowledgeBasesRetriever", diff --git a/libs/aws/langchain_aws/vectorstores/documentdb_new.py b/libs/aws/langchain_aws/vectorstores/documentdb.py similarity index 100% rename from libs/aws/langchain_aws/vectorstores/documentdb_new.py rename to libs/aws/langchain_aws/vectorstores/documentdb.py