Simplified code

Author: ankane

pgvector/psycopg2/halfvec.py

@@ -3,11 +3,11 @@
 
 
 class HalfvecAdapter:
-    def __init__(self, value: object) -> None:
+    def __init__(self, value: HalfVector) -> None:
         self._value = value
 
     def getquoted(self) -> bytes:
-        return adapt(HalfVector._to_db(self._value)).getquoted()
+        return adapt(self._value.to_text()).getquoted()
 
 
 def cast_halfvec(value: str | None, cur: cursor) -> HalfVector | None:

pgvector/psycopg2/sparsevec.py

@@ -3,11 +3,11 @@
 
 
 class SparsevecAdapter:
-    def __init__(self, value: object) -> None:
+    def __init__(self, value: SparseVector) -> None:
         self._value = value
 
     def getquoted(self) -> bytes:
-        return adapt(SparseVector._to_db(self._value)).getquoted()
+        return adapt(self._value.to_text()).getquoted()
 
 
 def cast_sparsevec(value: str | None, cur: cursor) -> SparseVector | None:

pgvector/psycopg2/vector.py

@@ -1,13 +1,19 @@
 from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
+from typing import TYPE_CHECKING
 from .. import Vector
 
+if TYPE_CHECKING:
+    import numpy as np
+
 
 class VectorAdapter:
-    def __init__(self, value: object) -> None:
+    def __init__(self, value: Vector | np.ndarray) -> None:
+        if not isinstance(value, Vector):
+            value = Vector(value)
         self._value = value
 
     def getquoted(self) -> bytes:
-        return adapt(Vector._to_db(self._value)).getquoted()
+        return adapt(self._value.to_text()).getquoted()
 
 
 def cast_vector(value: str | None, cur: cursor) -> Vector | None: