diff --git a/pyproject.toml b/pyproject.toml index 43407016..0fd96604 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -128,9 +128,10 @@ vectorstore-qdrant = ["qdrant-client>=1.0.0"] vectorstore-weaviate = ["weaviate-client>=4.0.0"] vectorstore-pinecone = ["pinecone-client>=3.0.0"] vectorstore-milvus = ["pymilvus>=2.0.0"] +vectorstore-pgvector = ["psycopg[binary]>=3.0.0", "psycopg2-binary>=2.9.0", "pgvector>=0.2.0"] vectorstore-all = [ - "semantica[vectorstore-qdrant,vectorstore-weaviate,vectorstore-pinecone,vectorstore-milvus]" + "semantica[vectorstore-qdrant,vectorstore-weaviate,vectorstore-pinecone,vectorstore-milvus,vectorstore-pgvector]" ] # ---- Infra / Queues / Workers ---- diff --git a/semantica/vector_store/__init__.py b/semantica/vector_store/__init__.py index 27f2b5d6..8b33c4be 100644 --- a/semantica/vector_store/__init__.py +++ b/semantica/vector_store/__init__.py @@ -147,6 +147,7 @@ from .methods import ( ) from .milvus_store import MilvusStore, MilvusClient, MilvusCollection, MilvusSearch from .namespace_manager import Namespace, NamespaceManager +from .pgvector_store import PgVectorStore from .pinecone_store import PineconeStore, PineconeClient, PineconeIndex, PineconeSearch from .qdrant_store import QdrantStore, QdrantClient, QdrantCollection, QdrantSearch from .registry import MethodRegistry, method_registry @@ -189,6 +190,8 @@ __all__ = [ "PineconeClient", "PineconeIndex", "PineconeSearch", + # PgVector + "PgVectorStore", # Hybrid search "HybridSearch", "MetadataFilter",