From 9d544142a8c615aad5af1054261117db100fdcac Mon Sep 17 00:00:00 2001 From: Alon Parag Date: Sat, 4 May 2024 10:15:50 +0200 Subject: [PATCH 1/2] removing reference to langhcain_community --- libs/aws/langchain_aws/vectorstores/documentdb.py | 4 ++-- .../tests/integration_tests/vectorstores/test_documentdb.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/aws/langchain_aws/vectorstores/documentdb.py b/libs/aws/langchain_aws/vectorstores/documentdb.py index a6ea3943..bbf38969 100644 --- 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("") diff --git a/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py b/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py index 77afc9e1..b0a4a4c1 100644 --- 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: From 2fbb124c93ddf0ba66527a11cc59333ba829fec9 Mon Sep 17 00:00:00 2001 From: Alon Parag Date: Sat, 4 May 2024 10:18:47 +0200 Subject: [PATCH 2/2] removing reference to langhcain_community --- libs/aws/langchain_aws/vectorstores/__init__.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/libs/aws/langchain_aws/vectorstores/__init__.py b/libs/aws/langchain_aws/vectorstores/__init__.py index 0549a570..020a087f 100644 --- 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