File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments