neighbor 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1154f0138248270d73d1681962c7bfecc8f81e074104a24071d32d76d5405705
-  data.tar.gz: 23c823cc022efcfebf9533d0388dd3abaec0fca61cc7e0911a2972dfb3bf429b
+  metadata.gz: 0d009ddc4383b98d64d9e9a5b4e5371261af7abde22470a7f3fa9b021f9040b4
+  data.tar.gz: b83f4e909fa613e82e64687d5d6014be9c28bd314c52780aa04a312f9dbcc3be
 SHA512:
-  metadata.gz: 67cc38e53cb43b8775561dd234cf073d2d45115b1c517aa32f890ab9f3d059106915158ee8bc08d2af9091e2107273ba9a65da6c66a1646afc419b62a8a669e0
-  data.tar.gz: 63cf2a9765043884726a59a6c058730d004be2b15bed0565be70579ef1717b322bb424e1311ced9a5685c8f45b7521333b0c66cca42c83cd52cb94cd13a7a5fd
+  metadata.gz: 8ba200a4bdf3004dbb4b033e12bd2b9c5fd0be91dd01bd35e11ceeb862483e038affa3781c62307729cd790ae7ed5832cbb28aba1edce240250fd4e2931033cb
+  data.tar.gz: e8924a3d4bae4d872ba6d2ce4f6afe0dca62984156210d05e51c847e2d3f3be0dc258e78160f504ee7e9c0650078628bd471174b07f08d65b3079efcca72e155

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## 1.2.0 (2026-06-04)
+
+- Added `threshold` option
+
+## 1.1.1 (2026-05-14)
+
+- Fixed `sqlite3 is not part of the bundle` error
+
 ## 1.1.0 (2026-05-14)
 
 - Added experimental support for SQLite with no extension

data/README.md

@@ -5,9 +5,9 @@ Nearest neighbor search for Rails
 Supports:
 
 - Postgres (pgvector and cube)
-- MariaDB 11.8
-- MySQL 9 (searching requires HeatWave) - experimental
-- SQLite - experimental
+- MariaDB 11.8+
+- MySQL 9.7+ (searching requires HeatWave)
+- SQLite
 
 Also available for [Redis](https://github.com/ankane/neighbor-redis) and [S3 Vectors](https://github.com/ankane/neighbor-s3)
 
@@ -91,6 +91,12 @@ nearest_item = item.nearest_neighbors(:embedding, distance: "euclidean").first
 nearest_item.neighbor_distance
 ```
 
+Get the nearest neighbors within a distance threshold
+
+```ruby
+item.nearest_neighbors(:embedding, distance: "euclidean", threshold: 0.5).first(5)
+```
+
 See the additional docs for:
 
 - [pgvector](#pgvector)
@@ -923,12 +929,14 @@ bundle exec rake test:postgresql
 
 # SQLite
 bundle exec rake test:sqlite
+bundle exec rake test:sqlitevec
+bundle exec rake test:vec1
 
 # MariaDB
 docker run -e MARIADB_ALLOW_EMPTY_ROOT_PASSWORD=1 -e MARIADB_DATABASE=neighbor_test -p 3307:3306 mariadb:11.8
 bundle exec rake test:mariadb
 
 # MySQL
-docker run -e MYSQL_ALLOW_EMPTY_PASSWORD=1 -e MYSQL_DATABASE=neighbor_test -p 3306:3306 mysql:9
+docker run -e MYSQL_ALLOW_EMPTY_PASSWORD=1 -e MYSQL_DATABASE=neighbor_test -p 3306:3306 mysql:9.7
 bundle exec rake test:mysql
 ```

data/lib/neighbor/model.rb

@@ -60,7 +60,7 @@ module Neighbor
           end
         end
 
-        scope :nearest_neighbors, ->(attribute_name, vector, distance:, precision: nil) {
+        scope :nearest_neighbors, ->(attribute_name, vector, distance:, threshold: nil, precision: nil) {
           attribute_name = attribute_name.to_sym
           options = neighbor_attributes[attribute_name]
           raise ArgumentError, "Invalid attribute" unless options
@@ -137,9 +137,16 @@ module Neighbor
 
           # for select, use column_names instead of * to account for ignored columns
           select_columns = select_values.any? ? [] : column_names
-          select(*select_columns, "#{neighbor_distance} AS neighbor_distance")
+          result = select(*select_columns, "#{neighbor_distance} AS neighbor_distance")
             .where.not(attribute_name => nil)
             .reorder(Arel.sql(order))
+
+          if threshold
+            op = distance == "inner_product" ? ">=" : "<="
+            result = result.where("#{neighbor_distance} #{op} ?", threshold)
+          end
+
+          result
         }
 
         def nearest_neighbors(attribute_name, **options)

data/lib/neighbor/version.rb

@@ -1,3 +1,3 @@
 module Neighbor
-  VERSION = "1.1.0"
+  VERSION = "1.2.0"
 end

data/lib/neighbor.rb

@@ -30,7 +30,12 @@ ActiveSupport.on_load(:active_record) do
   end
 
   Neighbor::MySQL.initialize!
-  Neighbor::SQLite.initialize_adapter!
+
+  begin
+    Neighbor::SQLite.initialize_adapter!
+  rescue Gem::LoadError
+    # tries to load sqlite3 gem, which may not be available
+  end
 end
 
 require_relative "neighbor/railtie" if defined?(Rails::Railtie)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: neighbor
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Andrew Kane