Skip to content

Commit f779591

Browse files
committed
Improved register_vector for pg8000 [skip ci]
1 parent 3c1dd2d commit f779591

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

pgvector/pg8000/register.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,19 +11,19 @@ def register_vector(conn: Connection) -> None:
1111
if 'vector' not in type_info:
1212
raise RuntimeError('vector type not found in the database')
1313

14-
conn.register_out_adapter(Vector, Vector._to_db)
14+
conn.register_out_adapter(Vector, lambda v: v.to_text())
1515
conn.register_in_adapter(type_info['vector'], Vector.from_text)
1616

1717
try:
1818
import numpy as np
19-
conn.register_out_adapter(np.ndarray, Vector._to_db)
19+
conn.register_out_adapter(np.ndarray, lambda v: Vector(v).to_text())
2020
except ImportError:
2121
pass
2222

2323
if 'halfvec' in type_info:
24-
conn.register_out_adapter(HalfVector, HalfVector._to_db)
24+
conn.register_out_adapter(HalfVector, lambda v: v.to_text())
2525
conn.register_in_adapter(type_info['halfvec'], HalfVector.from_text)
2626

2727
if 'sparsevec' in type_info:
28-
conn.register_out_adapter(SparseVector, SparseVector._to_db)
28+
conn.register_out_adapter(SparseVector, lambda v: v.to_text())
2929
conn.register_in_adapter(type_info['sparsevec'], SparseVector.from_text)

0 commit comments

Comments
 (0)