From 4ebc66656a894c394ae54e95521b5b287814f883 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alex=20Gr=C3=B6nholm?= <alex.gronholm@nextday.fi>
Date: Thu, 22 Feb 2024 10:47:20 +0200
Subject: [PATCH] Fixed pgvector extension not working

---
 CHANGES.rst            | 4 ++++
 src/sqlacodegen/cli.py | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/CHANGES.rst b/CHANGES.rst
index 4588f4f..fb07b5e 100644
--- 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
index adac632..755f542 100644
--- 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