add elasticknn to init (#7284)

1 year ago · 681f2678a3
parent c23e16c459
commit 681f2678a3
1 changed files with 5 additions and 1 deletions
--- a/langchain/vectorstores/init.py
+++ b/langchain/vectorstores/init.py
@ -15,7 +15,10 @@ from langchain.vectorstores.clarifai import Clarifai
 from langchain.vectorstores.clickhouse import Clickhouse, ClickhouseSettings
 from langchain.vectorstores.deeplake import DeepLake
 from langchain.vectorstores.docarray import DocArrayHnswSearch, DocArrayInMemorySearch
-from langchain.vectorstores.elastic_vector_search import ElasticVectorSearch
+from langchain.vectorstores.elastic_vector_search import (
+    ElasticKnnSearch,
+    ElasticVectorSearch,
+)
 from langchain.vectorstores.faiss import FAISS
 from langchain.vectorstores.hologres import Hologres
 from langchain.vectorstores.lancedb import LanceDB
@ -57,6 +60,7 @@ __all__ = [
    "DocArrayHnswSearch",
    "DocArrayInMemorySearch",
    "ElasticVectorSearch",
+    "ElasticKnnSearch",
    "FAISS",
    "PGEmbedding",
    "Hologres",