Fixed pgvector extension not working

agronholm · agronholm · commit 4ebc66656a89 · 2024-02-22T10:47:49.000+02:00
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,6 +1,10 @@
 Version history
 ===============
 
+**UNRELEASED**
+
+- Fixed pgvector support not working
+
 **3.0.0rc4**
 
 - Dropped support for Python 3.7
diff --git a/src/sqlacodegen/cli.py b/src/sqlacodegen/cli.py
@@ -19,7 +19,7 @@
     geoalchemy2 = None
 
 try:
-    import pgvector
+    import pgvector.sqlalchemy
 except ImportError:
     pgvector = None
 
