CrateDB: Vector Store

langchain-ai · Oct 29, 2024 · 379ce72 · 379ce72
1 parent 94e5765
commit 379ce72
Show file tree

Hide file tree

Showing 9 changed files with 1,361 additions and 0 deletions.
diff --git a/libs/community/extended_testing_deps.txt b/libs/community/extended_testing_deps.txt
@@ -76,6 +76,7 @@ requests-toolbelt>=1.0.0,<2
 rspace_client>=2.5.0,<3
 scikit-learn>=1.2.2,<2
 simsimd>=5.0.0,<6
+sqlalchemy-cratedb>=0.40.0,<1
 sqlite-vss>=0.1.2,<0.2
 sqlite-vec>=0.1.0,<0.2
 sseclient-py>=1.8.0,<2

diff --git a/libs/community/langchain_community/vectorstores/__init__.py b/libs/community/langchain_community/vectorstores/__init__.py
@@ -92,6 +92,9 @@
     from langchain_community.vectorstores.couchbase import (
         CouchbaseVectorStore,
     )
+    from langchain_community.vectorstores.cratedb import (
+        CrateDBVectorSearch,
+    )
     from langchain_community.vectorstores.dashvector import (
         DashVector,
     )
@@ -334,6 +337,7 @@
     "Clickhouse",
     "ClickhouseSettings",
     "CouchbaseVectorStore",
+    "CrateDBVectorSearch",
     "DashVector",
     "DatabricksVectorSearch",
     "DeepLake",
@@ -438,6 +442,7 @@
     "Clickhouse": "langchain_community.vectorstores.clickhouse",
     "ClickhouseSettings": "langchain_community.vectorstores.clickhouse",
     "CouchbaseVectorStore": "langchain_community.vectorstores.couchbase",
+    "CrateDBVectorSearch": "langchain_community.vectorstores.cratedb",
     "DashVector": "langchain_community.vectorstores.dashvector",
     "DatabricksVectorSearch": "langchain_community.vectorstores.databricks_vector_search",  # noqa: E501
     "DeepLake": "langchain_community.vectorstores.deeplake",

diff --git a/libs/community/langchain_community/vectorstores/cratedb/__init__.py b/libs/community/langchain_community/vectorstores/cratedb/__init__.py
@@ -0,0 +1,7 @@
+from .base import CrateDBVectorSearch
+from .extended import CrateDBVectorSearchMultiCollection
+
+__all__ = [
+    "CrateDBVectorSearch",
+    "CrateDBVectorSearchMultiCollection",
+]