38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
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)
|