Reduced allocations [skip ci]

Author: ankane

pgvector/halfvec.py

@@ -61,7 +61,7 @@ def to_binary(self) -> bytes:
 
     @classmethod
     def from_text(cls, value: str) -> HalfVector:
-        return cls([float(v) for v in value[1:-1].split(',')])
+        return cls(cls._list_from_text(value))
 
     @classmethod
     def from_binary(cls, value: bytes) -> HalfVector:
@@ -109,6 +109,10 @@ def _from_db(cls, value: str | HalfVector | None) -> list[float] | None:
             return value
 
         if isinstance(value, str):
-            value = cls.from_text(value)
+            return cls._list_from_text(value)
 
         return value.to_list()
+
+    @classmethod
+    def _list_from_text(cls, value: str) -> list[float]:
+        return [float(v) for v in value[1:-1].split(',')]

pgvector/vector.py

@@ -59,7 +59,7 @@ def to_binary(self) -> bytes:
 
     @classmethod
     def from_text(cls, value: str) -> Vector:
-        return cls([float(v) for v in value[1:-1].split(',')])
+        return cls(cls._list_from_text(value))
 
     @classmethod
     def from_binary(cls, value: bytes) -> Vector:
@@ -107,6 +107,10 @@ def _from_db(cls, value: str | Vector | None) -> list[float] | None:
             return value
 
         if isinstance(value, str):
-            value = cls.from_text(value)
+            return cls._list_from_text(value)
 
         return value.to_list()
+
+    @classmethod
+    def _list_from_text(cls, value: str) -> list[float]:
+        return [float(v) for v in value[1:-1].split(',')]