async support (#53)

* async support * format * fixed unit test failures * sort imports * more code reformat * typing fixes * typing fixes * one more reformat * typing fixes * docs * use sed to change get_messages() to messages
langchain-ai · Nov 27, 2024 · 9c3bccd · 9c3bccd
1 parent ff71063
commit 9c3bccd
Show file tree

Hide file tree

Showing 46 changed files with 7,334 additions and 2,374 deletions.
diff --git a/libs/elasticsearch/Makefile b/libs/elasticsearch/Makefile
@@ -46,6 +46,12 @@ spell_fix:
 check_imports: $(shell find langchain_elasticsearch -name '*.py')
 	poetry run python ./scripts/check_imports.py $^
 
+run_unasync:
+	poetry run python ./scripts/run_unasync.py
+
+run_unasync_check:
+	poetry run python ./scripts/run_unasync.py --check
+
 ######################
 # HELP
 ######################

diff --git a/libs/elasticsearch/langchain_elasticsearch/__init__.py b/libs/elasticsearch/langchain_elasticsearch/__init__.py
@@ -1,4 +1,9 @@
 from elasticsearch.helpers.vectorstore import (
+    AsyncBM25Strategy,
+    AsyncDenseVectorScriptScoreStrategy,
+    AsyncDenseVectorStrategy,
+    AsyncRetrievalStrategy,
+    AsyncSparseVectorStrategy,
     BM25Strategy,
     DenseVectorScriptScoreStrategy,
     DenseVectorStrategy,
@@ -8,28 +13,51 @@
 )
 
 from langchain_elasticsearch.cache import (
+    AsyncElasticsearchCache,
+    AsyncElasticsearchEmbeddingsCache,
     ElasticsearchCache,
     ElasticsearchEmbeddingsCache,
 )
-from langchain_elasticsearch.chat_history import ElasticsearchChatMessageHistory
-from langchain_elasticsearch.embeddings import ElasticsearchEmbeddings
-from langchain_elasticsearch.retrievers import ElasticsearchRetriever
+from langchain_elasticsearch.chat_history import (
+    AsyncElasticsearchChatMessageHistory,
+    ElasticsearchChatMessageHistory,
+)
+from langchain_elasticsearch.embeddings import (
+    AsyncElasticsearchEmbeddings,
+    ElasticsearchEmbeddings,
+)
+from langchain_elasticsearch.retrievers import (
+    AsyncElasticsearchRetriever,
+    ElasticsearchRetriever,
+)
 from langchain_elasticsearch.vectorstores import (
     ApproxRetrievalStrategy,
+    AsyncElasticsearchStore,
     BM25RetrievalStrategy,
     ElasticsearchStore,
     ExactRetrievalStrategy,
     SparseRetrievalStrategy,
 )
 
 __all__ = [
+    "AsyncElasticsearchCache",
+    "AsyncElasticsearchChatMessageHistory",
+    "AsyncElasticsearchEmbeddings",
+    "AsyncElasticsearchEmbeddingsCache",
+    "AsyncElasticsearchRetriever",
+    "AsyncElasticsearchStore",
     "ElasticsearchCache",
     "ElasticsearchChatMessageHistory",
     "ElasticsearchEmbeddings",
     "ElasticsearchEmbeddingsCache",
     "ElasticsearchRetriever",
     "ElasticsearchStore",
     # retrieval strategies
+    "AsyncBM25Strategy",
+    "AsyncDenseVectorScriptScoreStrategy",
+    "AsyncDenseVectorStrategy",
+    "AsyncRetrievalStrategy",
+    "AsyncSparseVectorStrategy",
     "BM25Strategy",
     "DenseVectorScriptScoreStrategy",
     "DenseVectorStrategy",