Added support for the pgvector extension

Author: agronholm

CHANGES.rst

@@ -4,6 +4,7 @@ Version history
 **UNRELEASED**
 
 - Dropped support for Python 3.7
+- Added support for the ``pgvector`` extension (with help from KellyRousselHoomano)
 
 **3.0.0rc3**
 

README.rst

@@ -45,6 +45,11 @@ To include support for the PostgreSQL ``GEOMETRY``, ``GEOGRAPHY``, and ``RASTER`
 (which should be considered as tested only under a few environments) specify the
 ``geoalchemy2`` extra:
 
+To include support for the PostgreSQL ``PGVECTOR`` extension type, specify the
+``pgvector`` extra::
+
+    pip install sqlacodegen[pgvector]
+
 .. code-block:: bash
 
     pip install sqlacodegen[geoalchemy2]

pyproject.toml

@@ -51,6 +51,7 @@ sqlmodel = [
 ]
 citext = ["sqlalchemy-citext >= 1.7.0"]
 geoalchemy2 = ["geoalchemy2 >= 0.11.1"]
+pgvector = ["pgvector >= 0.2.4"]
 
 [project.entry-points."sqlacodegen.generators"]
 tables = "sqlacodegen.generators:TablesGenerator"

src/sqlacodegen/cli.py

@@ -18,6 +18,11 @@
 except ImportError:
     geoalchemy2 = None
 
+try:
+    import pgvector
+except ImportError:
+    pgvector = None
+
 if sys.version_info < (3, 10):
     from importlib_metadata import entry_points, version
 else:
@@ -61,10 +66,13 @@ def main() -> None:
         return
 
     if citext:
-        print(f"Using sqlalchemy-citext {citext.__version__}")
+        print(f"Using sqlalchemy-citext {version('citext')}")
 
     if geoalchemy2:
-        print(f"Using geoalchemy2 {geoalchemy2.__version__}")
+        print(f"Using geoalchemy2 {version('geoalchemy2')}")
+
+    if pgvector:
+        print(f"Using pgvector {version('pgvector')}")
 
     # Use reflection to fill in the metadata
     engine = create_engine(args.url)