pgvector 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 215f975cafa2f782f3777ec65dd2fea587b420286f7f5ccf63b6b68376756dfd
-  data.tar.gz: e1979a1aa4fd4157cb04ae73c0c62b378046ca97154a25d4355e827be5abc53b
+  metadata.gz: 7485ea4be0d5be0177a972db911c696daf3438a661ddac61b08f4e8b2da3ac51
+  data.tar.gz: 2532ef79f5db88aecb681d9455e38f3e5fc1d30bde015d0a1e9daaa9fe82635e
 SHA512:
-  metadata.gz: 1e54d7c41f8750b99262021e402f329252b428678ee99510dd52a1a113aeffdf35628a23b6ca9408ce55350a237cb2d224e66dfe7aafefd645612a29b74529ff
-  data.tar.gz: c8a014bb3708690cf3e463954aa7a846ce79075da039e23b3f94644e50e3c28d05376539699fd8ffe3c3b93d54eecb393fed472bddb60619ba897a5204bc511a
+  metadata.gz: be40e4c3e16dd904a200115794a8ffaa850b40c5055330bd873ec7a707164a53b29d22040defbb4dd8a9cff597d4e1ad5c659d37a580ccc201ce30a9eb17fef9
+  data.tar.gz: f5d36289b043d987920911ab08d85d7d1066039f84dcc2a24436701c06b246adcb8fd3c32e3f76f3e1604001403574c14818f91a64357ce4ebd675458e57184c

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.2.2 (2023-10-03)
+
+- Added `nearest_neighbors` method to datasets with Sequel
+
 ## 0.2.1 (2023-06-04)
 
 - Added support for Sequel

data/README.md

@@ -29,6 +29,12 @@ Or check out some examples:
 
 ## pg
 
+Enable the extension
+
+```ruby
+conn.exec("CREATE EXTENSION IF NOT EXISTS vector")
+```
+
 Register the vector type with your connection
 
 ```ruby
@@ -52,6 +58,12 @@ conn.exec_params("SELECT * FROM items ORDER BY embedding <-> $1 LIMIT 5", [embed
 
 ## Sequel
 
+Enable the extension
+
+```ruby
+DB.run("CREATE EXTENSION IF NOT EXISTS vector")
+```
+
 Create a table
 
 ```ruby

data/lib/pgvector/version.rb

@@ -1,3 +1,3 @@
 module Pgvector
-  VERSION = "0.2.1"
+  VERSION = "0.2.2"
 end

data/lib/sequel/plugins/pgvector.rb

@@ -8,12 +8,10 @@ module Sequel
         end
       end
 
-      module ClassMethods
-        attr_accessor :vector_columns
-
+      module DatasetMethods
         def nearest_neighbors(column, value, distance:)
           value = ::Pgvector.encode(value) unless value.is_a?(String)
-          quoted_column = dataset.quote_identifier(column)
+          quoted_column = quote_identifier(column)
           distance = distance.to_s
 
           operator =
@@ -41,6 +39,12 @@ module Sequel
             .exclude(column => nil)
             .order(Sequel.lit(order, value))
         end
+      end
+
+      module ClassMethods
+        attr_accessor :vector_columns
+
+        Sequel::Plugins.def_dataset_methods(self, :nearest_neighbors)
 
         Plugins.inherited_instance_variables(self, :@vector_columns => :dup)
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pgvector
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-06-05 00:00:00.000000000 Z
+date: 2023-10-04 00:00:00.000000000 Z
 dependencies: []
 description:
 email: andrew@ankane.org