Merge pull request #3 from Alonoparag/async-vectorstore

Async vectorstore
langchain-ai · May 4, 2024 · 7c1bfa9 · 7c1bfa9
2 parents 144b277 + 2fbb124
commit 7c1bfa9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/libs/aws/langchain_aws/vectorstores/__init__.py b/libs/aws/langchain_aws/vectorstores/__init__.py
@@ -18,6 +18,4 @@
 
     Embeddings, Document
 """  # noqa: E501
-from langchain_aws.vectorstores.documentdb import (
-    DocumentDBVectorSearch,  # noqa: F401
-)
+from langchain_aws.vectorstores.documentdb import DocumentDBVectorSearch  # noqa: F401
diff --git a/libs/aws/langchain_aws/vectorstores/documentdb.py b/libs/aws/langchain_aws/vectorstores/documentdb.py
@@ -54,8 +54,8 @@ class DocumentDBVectorSearch(VectorStore):
     Example:
         . code-block:: python
 
-            from langchain_community.vectorstores import DocumentDBVectorSearch
-            from langchain_community.embeddings.openai import OpenAIEmbeddings
+            from langchain_aws.vectorstores import DocumentDBVectorSearch
+            from langchain_aws.embeddings.openai import OpenAIEmbeddings
             from pymongo import MongoClient
 
             mongo_client = MongoClient("<YOUR-CONNECTION-STRING>")

diff --git a/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py b/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py
@@ -59,7 +59,7 @@ def embedding_openai() -> Any:
 """
 This is how to run the integration tests:
 
-cd libs/community
+cd libs/aws
 make test TEST_FILE=tests/integration_tests/vectorstores/test_documentdb.py
 
 NOTE: You will first need to follow the contributor setup steps: