diff --git a/pgvector/psycopg2/__init__.py b/pgvector/psycopg2/__init__.py
index 9d0473a..ebab3d1 100644
--- a/pgvector/psycopg2/__init__.py
+++ b/pgvector/psycopg2/__init__.py
@@ -1,8 +1,9 @@
import psycopg2
+from .bit import register_bit_info
from .halfvec import register_halfvec_info
from .sparsevec import register_sparsevec_info
from .vector import register_vector_info
-from ..utils import SparseVector
+from ..utils import Bit, SparseVector
__all__ = ['register_vector']
@@ -16,6 +17,8 @@ def register_vector(conn_or_curs=None):
except psycopg2.errors.UndefinedObject:
raise psycopg2.ProgrammingError('vector type not found in the database')
+ register_bit_info()
+
try:
cur.execute('SELECT NULL::halfvec')
register_halfvec_info(cur.description[0][1])
diff --git a/pgvector/psycopg2/bit.py b/pgvector/psycopg2/bit.py
new file mode 100644
index 0000000..5b10d6d
--- /dev/null
+++ b/pgvector/psycopg2/bit.py
@@ -0,0 +1,14 @@
+from psycopg2.extensions import adapt, register_adapter
+from ..utils import Bit
+
+
+class BitAdapter:
+ def __init__(self, value):
+ self._value = value
+
+ def getquoted(self):
+ return adapt(Bit.to_db(self._value)).getquoted()
+
+
+def register_bit_info():
+ register_adapter(Bit, BitAdapter)
diff --git a/tests/test_psycopg2.py b/tests/test_psycopg2.py
index cc7e5c0..cf0a2b1 100644
--- a/tests/test_psycopg2.py
+++ b/tests/test_psycopg2.py
@@ -1,5 +1,5 @@
import numpy as np
-from pgvector.psycopg2 import register_vector, SparseVector
+from pgvector.psycopg2 import register_vector, Bit, SparseVector
import psycopg2
conn = psycopg2.connect(dbname='pgvector_python_test')
@@ -37,7 +37,7 @@ def test_halfvec(self):
assert res[1][0] is None
def test_bit(self):
- embedding = '101'
+ embedding = Bit('101')
cur.execute('INSERT INTO psycopg2_items (binary_embedding) VALUES (%s), (NULL)', (embedding,))
cur.execute('SELECT binary_embedding FROM psycopg2_items ORDER BY id')
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy