Registering Vector Type with SQLAlchemy Registries (#45)

KellyRousselHoomano · web-flow · commit 4e02d6f7d62e · 2023-11-22T10:16:26.000-08:00
diff --git a/pgvector/sqlalchemy/__init__.py b/pgvector/sqlalchemy/__init__.py
@@ -1,6 +1,6 @@
 from sqlalchemy.types import UserDefinedType, Float
 from ..utils import from_db, to_db
-
+from sqlalchemy.dialects import postgresql
 __all__ = ['Vector']
 
 
@@ -35,3 +35,6 @@ def max_inner_product(self, other):
 
         def cosine_distance(self, other):
             return self.op('<=>', return_type=Float)(other)
+
+# Register Vector type to PostgreSQL's reflection subsystem
+postgresql.base.ischema_names['vector'] = Vector
