diff options
Diffstat (limited to '.venv/lib/python3.12/site-packages/pgvector/psycopg2/sparsevec.py')
-rw-r--r-- | .venv/lib/python3.12/site-packages/pgvector/psycopg2/sparsevec.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/pgvector/psycopg2/sparsevec.py b/.venv/lib/python3.12/site-packages/pgvector/psycopg2/sparsevec.py new file mode 100644 index 00000000..a542807b --- /dev/null +++ b/.venv/lib/python3.12/site-packages/pgvector/psycopg2/sparsevec.py @@ -0,0 +1,25 @@ +from psycopg2.extensions import adapt, new_array_type, new_type, register_adapter, register_type +from ..utils import SparseVector + + +class SparsevecAdapter: + def __init__(self, value): + self._value = value + + def getquoted(self): + return adapt(SparseVector._to_db(self._value)).getquoted() + + +def cast_sparsevec(value, cur): + return SparseVector._from_db(value) + + +def register_sparsevec_info(oid, array_oid, scope): + sparsevec = new_type((oid,), 'SPARSEVEC', cast_sparsevec) + register_type(sparsevec, scope) + + if array_oid is not None: + sparsevecarray = new_array_type((array_oid,), 'SPARSEVECARRAY', sparsevec) + register_type(sparsevecarray, scope) + + register_adapter(SparseVector, SparsevecAdapter) |