Restore 0.1.5 version from stash

This commit is contained in:
liaibo
2025-12-08 19:56:24 +08:00
parent de189e938d
commit 8db3f4e32d
8578 changed files with 2703426 additions and 217 deletions
@@ -0,0 +1,9 @@
from .register import register_vector
from ..utils import Vector, HalfVector, SparseVector
__all__ = [
'register_vector',
'Vector',
'HalfVector',
'SparseVector'
]
@@ -0,0 +1,31 @@
from ..utils import Vector, HalfVector, SparseVector
async def register_vector(conn, schema='public'):
await conn.set_type_codec(
'vector',
schema=schema,
encoder=Vector._to_db_binary,
decoder=Vector._from_db_binary,
format='binary'
)
try:
await conn.set_type_codec(
'halfvec',
schema=schema,
encoder=HalfVector._to_db_binary,
decoder=HalfVector._from_db_binary,
format='binary'
)
await conn.set_type_codec(
'sparsevec',
schema=schema,
encoder=SparseVector._to_db_binary,
decoder=SparseVector._from_db_binary,
format='binary'
)
except ValueError as e:
if not str(e).startswith('unknown type:'):
raise e