/
rnekrasov
/
pgvector-python
Обзор
Документация
Войти
/
rnekrasov
/
pgvector-python
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
pgvector/psycopg/vector.py
55 строк
1 KB
Andrew Kane
Use consistent style [skip ci]
06 апр 2024, 22:14
06 апр 2024, 22:14
70fc5df
Код
Авторство
О чём код?
from psycopg.adapt import Loader, Dumper from psycopg.pq import Format from ..utils import from_db, from_db_binary, to_db, to_db_binary class VectorDumper(Dumper): format = Format.TEXT def dump(self, obj): return to_db(obj).encode('utf8') class VectorBinaryDumper(VectorDumper): format = Format.BINARY def dump(self, obj): return to_db_binary(obj) class VectorLoader(Loader): format = Format.TEXT def load(self, data): if isinstance(data, memoryview): data = bytes(data) return from_db(data.decode('utf8')) class VectorBinaryLoader(VectorLoader): format = Format.BINARY def load(self, data): if isinstance(data, memoryview): data = bytes(data) return from_db_binary(data) def register_vector_info(context, info): if info is None: raise psycopg.ProgrammingError('vector type not found in the database') info.register(context) # add oid to anonymous class for set_types text_dumper = type('', (VectorDumper,), {'oid': info.oid}) binary_dumper = type('', (VectorBinaryDumper,), {'oid': info.oid}) adapters = context.adapters adapters.register_dumper('numpy.ndarray', text_dumper) adapters.register_dumper('numpy.ndarray', binary_dumper) adapters.register_loader(info.oid, VectorLoader) adapters.register_loader(info.oid, VectorBinaryLoader)