async support

langchain-ai · Nov 21, 2024 · 6cb5b0a · 6cb5b0a
1 parent ff71063
commit 6cb5b0a
Show file tree

Hide file tree

Showing 46 changed files with 7,164 additions and 2,363 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,48 @@
 )
 
 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",