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/vector.py | 26 ++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 .venv/lib/python3.12/site-packages/pgvector/psycopg2/vector.py (limited to '.venv/lib/python3.12/site-packages/pgvector/psycopg2/vector.py') diff --git a/.venv/lib/python3.12/site-packages/pgvector/psycopg2/vector.py b/.venv/lib/python3.12/site-packages/pgvector/psycopg2/vector.py new file mode 100644 index 00000000..9861f01f --- /dev/null +++ b/.venv/lib/python3.12/site-packages/pgvector/psycopg2/vector.py @@ -0,0 +1,26 @@ +import numpy as np +from psycopg2.extensions import adapt, new_array_type, new_type, register_adapter, register_type +from ..utils import Vector + + +class VectorAdapter: + def __init__(self, value): + self._value = value + + def getquoted(self): + return adapt(Vector._to_db(self._value)).getquoted() + + +def cast_vector(value, cur): + return Vector._from_db(value) + + +def register_vector_info(oid, array_oid, scope): + vector = new_type((oid,), 'VECTOR', cast_vector) + register_type(vector, scope) + + if array_oid is not None: + vectorarray = new_array_type((array_oid,), 'VECTORARRAY', vector) + register_type(vectorarray, scope) + + register_adapter(np.ndarray, VectorAdapter) -- cgit v1.2.3