From 4a52a71956a8d46fcb7294ac71734504bb09bcc2 Mon Sep 17 00:00:00 2001 From: S. Solomon Darnell Date: Fri, 28 Mar 2025 21:52:21 -0500 Subject: two version of R2R are here --- .../site-packages/pgvector/psycopg2/halfvec.py | 25 ++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 .venv/lib/python3.12/site-packages/pgvector/psycopg2/halfvec.py (limited to '.venv/lib/python3.12/site-packages/pgvector/psycopg2/halfvec.py') diff --git a/.venv/lib/python3.12/site-packages/pgvector/psycopg2/halfvec.py b/.venv/lib/python3.12/site-packages/pgvector/psycopg2/halfvec.py new file mode 100644 index 00000000..b50e89b6 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/pgvector/psycopg2/halfvec.py @@ -0,0 +1,25 @@ +from psycopg2.extensions import adapt, new_array_type, new_type, register_adapter, register_type +from ..utils import HalfVector + + +class HalfvecAdapter: + def __init__(self, value): + self._value = value + + def getquoted(self): + return adapt(HalfVector._to_db(self._value)).getquoted() + + +def cast_halfvec(value, cur): + return HalfVector._from_db(value) + + +def register_halfvec_info(oid, array_oid, scope): + halfvec = new_type((oid,), 'HALFVEC', cast_halfvec) + register_type(halfvec, scope) + + if array_oid is not None: + halfvecarray = new_array_type((array_oid,), 'HALFVECARRAY', halfvec) + register_type(halfvecarray, scope) + + register_adapter(HalfVector, HalfvecAdapter) -- cgit v1.2.3