Skip to content

Commit 85d70b9

Browse files
committed
Improved type registration
1 parent 5cb3d50 commit 85d70b9

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

lib/pgvector/pg.rb

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,17 @@
33
module Pgvector
44
module PG
55
def self.register_vector(registry)
6-
registry.register_type(0, "vector", nil, TextDecoder::Vector)
7-
registry.register_type(1, "vector", nil, BinaryDecoder::Vector)
6+
registry.register_type(0, "vector", TextEncoder::Vector, TextDecoder::Vector)
7+
registry.register_type(1, "vector", BinaryEncoder::Vector, BinaryDecoder::Vector)
88

99
# no binary decoder for halfvec since unpack does not have directive for half-precision
10-
registry.register_type(0, "halfvec", nil, TextDecoder::Halfvec)
10+
registry.register_type(0, "halfvec", TextEncoder::Halfvec, TextDecoder::Halfvec)
1111

12-
registry.register_type(0, "bit", nil, TextDecoder::Bit)
12+
registry.register_type(0, "bit", TextEncoder::Bit, TextDecoder::Bit)
1313
registry.register_type(1, "bit", nil, BinaryDecoder::Bit)
1414

15-
registry.register_type(0, "sparsevec", nil, TextDecoder::Sparsevec)
16-
registry.register_type(1, "sparsevec", nil, BinaryDecoder::Sparsevec)
15+
registry.register_type(0, "sparsevec", TextEncoder::Sparsevec, TextDecoder::Sparsevec)
16+
registry.register_type(1, "sparsevec", BinaryEncoder::Sparsevec, BinaryDecoder::Sparsevec)
1717
end
1818

1919
module BinaryDecoder

0 commit comments

Comments
 (0)