Removed _to_db_binary methods [skip ci]

Author: ankane

pgvector/asyncpg/register.py

@@ -6,7 +6,7 @@ async def register_vector(conn: Connection, schema: str = 'public') -> None:
     await conn.set_type_codec(
         'vector',
         schema=schema,
-        encoder=Vector._to_db_binary,
+        encoder=lambda v: (v if isinstance(v, Vector) else Vector(v)).to_binary(),
         decoder=Vector.from_binary,
         format='binary'
     )
@@ -15,15 +15,15 @@ async def register_vector(conn: Connection, schema: str = 'public') -> None:
         await conn.set_type_codec(
             'halfvec',
             schema=schema,
-            encoder=HalfVector._to_db_binary,
+            encoder=lambda v: (v if isinstance(v, HalfVector) else HalfVector(v)).to_binary(),
             decoder=HalfVector.from_binary,
             format='binary'
         )
 
         await conn.set_type_codec(
             'sparsevec',
             schema=schema,
-            encoder=SparseVector._to_db_binary,
+            encoder=lambda v: (v if isinstance(v, SparseVector) else SparseVector(v)).to_binary(),
             decoder=SparseVector.from_binary,
             format='binary'
         )

pgvector/halfvec.py

@@ -93,13 +93,6 @@ def _to_db(cls, value: object) -> str | None:
 
         return value.to_text()
 
-    @classmethod
-    def _to_db_binary(cls, value: object) -> bytes:
-        if not isinstance(value, cls):
-            value = cls(value)  # type: ignore
-
-        return value.to_binary()
-
     @classmethod
     def _from_db(cls, value: str | HalfVector | None) -> list[float] | None:
         if value is None:

pgvector/sparsevec.py

@@ -158,13 +158,6 @@ def _to_db(cls, value: object) -> str | None:
 
         return value.to_text()
 
-    @classmethod
-    def _to_db_binary(cls, value: object) -> bytes:
-        if not isinstance(value, cls):
-            value = cls(value)
-
-        return value.to_binary()
-
     @classmethod
     def _from_db(cls, value: str | SparseVector | None) -> SparseVector | None:
         if value is None or isinstance(value, SparseVector):

pgvector/vector.py

@@ -91,13 +91,6 @@ def _to_db(cls, value: object) -> str | None:
 
         return value.to_text()
 
-    @classmethod
-    def _to_db_binary(cls, value: object) -> bytes:
-        if not isinstance(value, cls):
-            value = cls(value)  # type: ignore
-
-        return value.to_binary()
-
     @classmethod
     def _from_db(cls, value: str | Vector | None) -> list[float] | None:
         if value is None: