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/psycopg/register.py | 37 ++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 .venv/lib/python3.12/site-packages/pgvector/psycopg/register.py (limited to '.venv/lib/python3.12/site-packages/pgvector/psycopg/register.py') diff --git a/.venv/lib/python3.12/site-packages/pgvector/psycopg/register.py b/.venv/lib/python3.12/site-packages/pgvector/psycopg/register.py new file mode 100644 index 00000000..b93fd3ee --- /dev/null +++ b/.venv/lib/python3.12/site-packages/pgvector/psycopg/register.py @@ -0,0 +1,37 @@ +from psycopg.types import TypeInfo +from .bit import register_bit_info +from .halfvec import register_halfvec_info +from .sparsevec import register_sparsevec_info +from .vector import register_vector_info + + +def register_vector(context): + info = TypeInfo.fetch(context, 'vector') + register_vector_info(context, info) + + info = TypeInfo.fetch(context, 'bit') + register_bit_info(context, info) + + info = TypeInfo.fetch(context, 'halfvec') + if info is not None: + register_halfvec_info(context, info) + + info = TypeInfo.fetch(context, 'sparsevec') + if info is not None: + register_sparsevec_info(context, info) + + +async def register_vector_async(context): + info = await TypeInfo.fetch(context, 'vector') + register_vector_info(context, info) + + info = await TypeInfo.fetch(context, 'bit') + register_bit_info(context, info) + + info = await TypeInfo.fetch(context, 'halfvec') + if info is not None: + register_halfvec_info(context, info) + + info = await TypeInfo.fetch(context, 'sparsevec') + if info is not None: + register_sparsevec_info(context, info) -- cgit v1.2.3