Skip to content

Commit 15f1ac7

Browse files
committed
Simplified Psycopg 2 code
1 parent 135cd53 commit 15f1ac7

3 files changed

Lines changed: 9 additions & 3 deletions

File tree

pgvector/psycopg2/halfvec.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,9 @@ def getquoted(self) -> bytes:
1111

1212

1313
def cast_halfvec(value: str | None, cur: cursor) -> HalfVector | None:
14-
return HalfVector._from_db(value)
14+
if value is None:
15+
return None
16+
return HalfVector.from_text(value)
1517

1618

1719
def register_halfvec_info(oid: int, array_oid: int | None, scope: connection | cursor | None) -> None:

pgvector/psycopg2/sparsevec.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,9 @@ def getquoted(self) -> bytes:
1111

1212

1313
def cast_sparsevec(value: str | None, cur: cursor) -> SparseVector | None:
14-
return SparseVector._from_db(value)
14+
if value is None:
15+
return None
16+
return SparseVector.from_text(value)
1517

1618

1719
def register_sparsevec_info(oid: int, array_oid: int | None, scope: connection | cursor | None) -> None:

pgvector/psycopg2/vector.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,9 @@ def getquoted(self) -> bytes:
1111

1212

1313
def cast_vector(value: str | None, cur: cursor) -> Vector | None:
14-
return Vector._from_db(value)
14+
if value is None:
15+
return None
16+
return Vector.from_text(value)
1517

1618

1719
def register_vector_info(oid: int, array_oid: int | None, scope: connection | cursor | None) -> None:

0 commit comments

Comments
 (0)