activerecord-turbopuffer-adapter 0.1.9 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fde17eff7db3a885b55a20550170a189c505ef52afc6bbdfded1f9b5dfcc9a76
-  data.tar.gz: 4ee793a06d003d5d69c6d3748c7a74331d6bd4c5db1df6c4ec79fba22f408fea
+  metadata.gz: ae2b8fa993fe940121966a4d3c8c9b95a43afc484348f1598ed76bb24ad0c021
+  data.tar.gz: 02b90f77da9ec0f5133fc021fa516b64516a79eea2b6d63d0bacf6b7feb97399
 SHA512:
-  metadata.gz: b716b0601df52af79a1d6edbba112630504f356fb2e6c7b91ad00ac6e41485809b43849eb58d8399f0238190db6ea3f5bd3b0b490f0658b17e2d138132f24b7c
-  data.tar.gz: 7b79a130b932801f07c98ec849cb9838c8c86246c432464b64aa2c9c079d8432146bb189986886169f614f395fb5e1c00a7b5ccc56a76424d4fbe1ca3ef6b290
+  metadata.gz: fe02be8410aae63cc2479a3cb9ec52e1b54ffae8f34e21b33a63168f3572edeef99d5ccce08a675ce30adc227f967e519f4dd66cedb160a4f1b0dc4299a79c1b
+  data.tar.gz: ba93057de23dd67659cdbfd4d0a8e822fddd7ff162129932b995061c8c8e8d19cb10ae832c08e41639b2edf619d5119239b7f01043d42404bd7aaaac1f0ce37e

data/README.md

@@ -109,6 +109,7 @@ Document.where(tags: nil)
 Document.where(scores: 90..)
 
 Document.order(:title).limit(20)
+Document.order(:category, views: :desc)
 Document.order(:title).limit(20).offset(20)
 Document.group(:title).count
 Document.count

data/lib/active_record/connection_adapters/turbopuffer_adapter.rb

@@ -284,7 +284,7 @@ module ActiveRecord
 
         tpuf_query_args[:offset] = query.offset if query.offset
 
-        tpuf_query_args[:rank_by] = query.rank_by.first if query.rank_by.present?
+        tpuf_query_args[:rank_by] = query.rank_by if query.rank_by
         tpuf_query_args[:filters] = query.filters if query.filters
         consistency = consistency_for(query)
         tpuf_query_args[:consistency] = consistency if consistency

data/lib/arel/visitors/turbopuffer.rb

@@ -101,11 +101,18 @@ module Arel::Visitors
       end
     end
 
-    def visit_Arel_Nodes_SelectStatement(o)
-      # https://turbopuffer.com/docs/query#ordering-by-attributes: "Ordering by
-      # multiple attributes isn't yet implemented."
-      raise NotImplementedError, "one ranking per query, sort in Ruby after loading" if o.orders.size > 1
+    def ranking(orders)
+      return if orders.empty?
+      return visit(orders.first) if orders.size == 1
+
+      if orders.any? { |order| order.is_a?(Arel::Nodes::RankByNode) }
+        raise NotImplementedError, "a ranking cannot be combined with other orderings, sort in Ruby after loading"
+      end
 
+      orders.map { |order| visit(order) }
+    end
+
+    def visit_Arel_Nodes_SelectStatement(o)
       core = o.cores.last
 
       raise NotImplementedError, "distinct is not implemented yet" if core.set_quantifier
@@ -125,7 +132,7 @@ module Arel::Visitors
         namespace:          visit(core.source),
         filters:            conjoin(filters),
         top_k:              o.limit && visit(o.limit),
-        rank_by:            o.orders.map { |ord| visit(ord) },
+        rank_by:            ranking(o.orders),
         include_attributes: attributes.flat_map { |p| visit(p) },
         group_by:           core.groups.map { |g| visit(g) },
         aggregate_by:       aggregate && visit(aggregate),

data/lib/turbopuffer/active_record/version.rb

@@ -1,5 +1,5 @@
 module Turbopuffer
   module ActiveRecord
-    VERSION = "0.1.9"
+    VERSION = "0.1.10"
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-turbopuffer-adapter
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors:
 - Richard Monette
 bindir: bin
 cert_chain: []
-date: 2026-10-11 00:00:00.000000000 Z
+date: 2026-10-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord