pgvector 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 07a80636c13841d2fa97f8b740feb095c1c2f94fa3691b374d1472bba918bddf
-  data.tar.gz: 50c9e67781fbbe23fa3dbe2f87790025f18d3686e412b46f62d1cb0d2995ecb3
+  metadata.gz: 3d3ef6a53417383ff7f8a2a514df78513dfe9029e524f0f624df8828fb99dba0
+  data.tar.gz: f326a21d3942079cdadc22d9a61367e7a6651ae37c5eb07a3955182f9f569ad0
 SHA512:
-  metadata.gz: f0aa5b733e1bc4022c2052be6b04aa0656bf306f588baf2907827674aeadc7dc52507895043049246634999dea37de629c19a164bd289fbd979eb628aaab0c33
-  data.tar.gz: f52e2e9c1c074c4f809fe78d6926bcb630694d79199e288ebb33d136a05191c4b2fff6038f8d7fc56d4291f949caa1b536d942aa7d0912be2daccf5a7ee23f83
+  metadata.gz: fa4d6519685d179e3d5712cbcf1e6989ca4f98e5b0902f5061eca2be55451162f199fcc75f19841dbd8ada8c18de69ac6fa39cad545d4b5a6c542e9cdd0109ea
+  data.tar.gz: fd71245492b2ff8a06a0af60dbd12e57e44025f2662d4ea77eb7176aaaa4c4cb3b39c1e159edce92944da5b54fddbb19d1b504acd76a12c4efdb9a0fb6e797da

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## 0.3.1 (2024-07-10)
+
+- Added support for `bit` type to pg
+- Added extension for Sequel
+
 ## 0.3.0 (2024-06-25)
 
 - Added support for `halfvec` and `sparsevec` types

data/lib/pgvector/bit.rb

@@ -0,0 +1,28 @@
+module Pgvector
+  class Bit
+    def initialize(data)
+      if data.is_a?(Array)
+        @data = data.map { |v| v ? "1" : "0" }.join
+      else
+        @data = data.to_str
+      end
+    end
+
+    def self.from_text(string)
+      Bit.new(string)
+    end
+
+    def self.from_binary(string)
+      length = string[..3].unpack1("l>")
+      Bit.new(string[4..].unpack("B*").join[...length])
+    end
+
+    def to_s
+      @data
+    end
+
+    def to_a
+      @data.each_char.map { |v| v != "0" }
+    end
+  end
+end

data/lib/pgvector/pg.rb

@@ -9,6 +9,9 @@ module Pgvector
       # no binary decoder for halfvec since unpack does not have directive for half-precision
       registry.register_type(0, "halfvec", nil, TextDecoder::Halfvec)
 
+      registry.register_type(0, "bit", nil, TextDecoder::Bit)
+      registry.register_type(1, "bit", nil, BinaryDecoder::Bit)
+
       registry.register_type(0, "sparsevec", nil, TextDecoder::Sparsevec)
       registry.register_type(1, "sparsevec", nil, BinaryDecoder::Sparsevec)
     end
@@ -20,6 +23,12 @@ module Pgvector
         end
       end
 
+      class Bit < ::PG::SimpleDecoder
+        def decode(string, tuple = nil, field = nil)
+          ::Pgvector::Bit.from_binary(string).to_s
+        end
+      end
+
       class Sparsevec < ::PG::SimpleDecoder
         def decode(string, tuple = nil, field = nil)
           SparseVector.from_binary(string)
@@ -40,6 +49,12 @@ module Pgvector
         end
       end
 
+      class Bit < ::PG::SimpleDecoder
+        def decode(string, tuple = nil, field = nil)
+          ::Pgvector::Bit.from_text(string).to_s
+        end
+      end
+
       class Sparsevec < ::PG::SimpleDecoder
         def decode(string, tuple = nil, field = nil)
           SparseVector.from_text(string)

data/lib/pgvector/version.rb

@@ -1,3 +1,3 @@
 module Pgvector
-  VERSION = "0.3.0"
+  VERSION = "0.3.1"
 end

data/lib/pgvector.rb

@@ -1,4 +1,5 @@
 # modules
+require_relative "pgvector/bit"
 require_relative "pgvector/half_vector"
 require_relative "pgvector/sparse_vector"
 require_relative "pgvector/vector"
@@ -8,7 +9,7 @@ module Pgvector
   autoload :PG, "pgvector/pg"
 
   def self.encode(data)
-    if data.is_a?(SparseVector)
+    if data.is_a?(Vector) || data.is_a?(HalfVector) || data.is_a?(SparseVector)
       data.to_s
     else
       Vector.new(data).to_s

data/lib/sequel/extensions/pgvector.rb

@@ -0,0 +1,5 @@
+require_relative "../plugins/pgvector"
+
+module Sequel
+  Dataset.register_extension(:pgvector, Plugins::Pgvector::DatasetMethods)
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pgvector
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-06-26 00:00:00.000000000 Z
+date: 2024-07-11 00:00:00.000000000 Z
 dependencies: []
 description:
 email: andrew@ankane.org
@@ -20,11 +20,13 @@ files:
 - LICENSE.txt
 - README.md
 - lib/pgvector.rb
+- lib/pgvector/bit.rb
 - lib/pgvector/half_vector.rb
 - lib/pgvector/pg.rb
 - lib/pgvector/sparse_vector.rb
 - lib/pgvector/vector.rb
 - lib/pgvector/version.rb
+- lib/sequel/extensions/pgvector.rb
 - lib/sequel/plugins/pgvector.rb
 homepage: https://github.com/pgvector/pgvector-ruby
 licenses: