Fixed error with asyncpg and pgvector < 0.7 - fixes #83

ankane · ankane · commit 4672a4b8a83b · 2024-07-17T10:37:37.000-07:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 0.3.2 (unreleased)
+
+- Fixed error with asyncpg and pgvector < 0.7
+
 ## 0.3.1 (2024-07-10)
 
 - Fixed error parsing zero sparse vectors
diff --git a/pgvector/asyncpg/register.py b/pgvector/asyncpg/register.py
@@ -9,16 +9,20 @@ async def register_vector(conn):
         format='binary'
     )
 
-    await conn.set_type_codec(
-        'halfvec',
-        encoder=HalfVector._to_db_binary,
-        decoder=HalfVector._from_db_binary,
-        format='binary'
-    )
+    try:
+        await conn.set_type_codec(
+            'halfvec',
+            encoder=HalfVector._to_db_binary,
+            decoder=HalfVector._from_db_binary,
+            format='binary'
+        )
 
-    await conn.set_type_codec(
-        'sparsevec',
-        encoder=SparseVector._to_db_binary,
-        decoder=SparseVector._from_db_binary,
-        format='binary'
-    )
+        await conn.set_type_codec(
+            'sparsevec',
+            encoder=SparseVector._to_db_binary,
+            decoder=SparseVector._from_db_binary,
+            format='binary'
+        )
+    except ValueError as e:
+        if not str(e).startswith('unknown type:'):
+            raise e
