riveter-sdk 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 867ffe82023d427f6584d46ad9d0e0e1799bbab22ac173a66eb542859d13fd7d
-  data.tar.gz: 6906da4985f0795a6074d4b3891f02cb27ccd02009165a13e9a550bb962449a6
+  metadata.gz: c11eb2e15de73602df663800326a931a9300daf4ccc615a761e761e1ed776a71
+  data.tar.gz: '092c593d7aac42080efd1ea487822e36eae8218f2095f3cbb706950048fb011b'
 SHA512:
-  metadata.gz: 2663f0f05a85eb76a748775bae906427513d103d259ff9bab9cb7a8661ad8237bfbb65e3ce35c1cf94a39215749decd733640a8e739423a1358890f88c889bad
-  data.tar.gz: 45ef4e0f5b33e003ca0bc9916d0105f23d15a7fbd03280f5cd5ab84001884e86202673e48632f3e8100eee0e5fde2f8b58a8f17852aa7df0081f4c3c6375870d
+  metadata.gz: 60c2971793274fdaa1ed1ed545ac3c0ce389ece4f408eee7e79131b3d0d4f59d2988460a1f90e432cb7d48ad4bb440b909b40fec7f5873ce4d75c445d7dd8184
+  data.tar.gz: 586e72eb77cdf4fd1818cfa75285a02bf9df60ef8331f4fd0542baf1a636be79e92fbca855957cc9741e3b01148417ed6dcb15333a5e0940e4099870eb7ad4ae

data/lib/riveter/models.rb

@@ -120,6 +120,36 @@ module Riveter
     end
   end
 
+  # A row as compact: true answers it: the values alone. #cells is each output column's value
+  # (nil when the cell has none). Address the row by #position elsewhere.
+  class CompactDatasetRow < Model
+    field :position, :input, :cells
+  end
+
+  # One of a column's most common values (a string is cut at 200 characters) and its row count.
+  class DatasetColumnValueCount < Model
+    field :value, :count
+  end
+
+  # One column's counts over every row: #states (output columns only: cells by state, pending /
+  # stale / failed / not_found / ok / manual), #filled, #distinct_values, the most common #values.
+  class DatasetColumnSummary < Model
+    field :name, :kind, :states, :filled, :distinct_values
+
+    def values
+      (@raw["values"] || []).map { |item| DatasetColumnValueCount.new(item) }
+    end
+  end
+
+  # Per-column counts over every row of a dataset (Datasets#summary).
+  class DatasetSummary < Model
+    field :row_count
+
+    def columns
+      (@raw["columns"] || []).map { |item| DatasetColumnSummary.new(item) }
+    end
+  end
+
   # A run on a dataset as the dataset lists it; #type is dataset_build or enrichment.
   class DatasetRunSummary < Model
     field :id, :type, :status, :created_at, :finished_at

data/lib/riveter/page.rb

@@ -41,18 +41,20 @@ module Riveter
   end
 
   # One page of dataset rows (cursor-paginated). #each iterates the page;
-  # #auto_paging_each walks every following page too.
+  # #auto_paging_each walks every following page too. Rows are DatasetRow, or
+  # CompactDatasetRow for a compact: true read (row_class).
   class DatasetRowsPage
     include Enumerable
 
     attr_reader :rows, :next_cursor, :row_count
 
-    def initialize(data, fetch_page)
+    def initialize(data, fetch_page, row_class: DatasetRow)
       data = data.is_a?(Hash) ? data : {}
-      @rows = (data["rows"] || []).map { |item| DatasetRow.new(item) }
+      @rows = (data["rows"] || []).map { |item| row_class.new(item) }
       @next_cursor = data["next_cursor"]
       @row_count = data["row_count"]
       @fetch_page = fetch_page
+      @row_class = row_class
     end
 
     def each(&block)
@@ -66,7 +68,7 @@ module Riveter
     def next_page
       raise Error, "No next page" unless next_page?
 
-      DatasetRowsPage.new(@fetch_page.call(next_cursor), @fetch_page)
+      DatasetRowsPage.new(@fetch_page.call(next_cursor), @fetch_page, row_class: @row_class)
     end
 
     def auto_paging_each(&block)

data/lib/riveter/resources.rb

@@ -135,12 +135,28 @@ module Riveter
       end
 
       # A page of rows in position order (`limit:`, `cursor:`, `updated_since:`);
-      # #auto_paging_each walks every following page. (operationId: listDatasetRows)
-      def list_rows(dataset_id, **params)
+      # #auto_paging_each walks every following page. `columns:` (an Array of names) keeps only
+      # those columns; `compact: true` answers CompactDatasetRow (the values alone), so a
+      # 1,000-row page of one column is one small call. (operationId: listDatasetRows)
+      def list_rows(dataset_id, columns: nil, compact: false, **params)
+        query = params.merge(columns: columns && Array(columns).join(","), compact: compact ? "true" : nil)
         fetch_page = lambda do |cursor|
-          @client.request(:get, "/datasets/#{dataset_id}/rows", query: params.merge(cursor: cursor))
+          @client.request(:get, "/datasets/#{dataset_id}/rows", query: query.merge(cursor: cursor))
         end
-        DatasetRowsPage.new(fetch_page.call(params[:cursor]), fetch_page)
+        row_class = compact ? CompactDatasetRow : DatasetRow
+        DatasetRowsPage.new(fetch_page.call(params[:cursor]), fetch_page, row_class: row_class)
+      end
+
+      # Per-column counts over every row, no rows read: an output column's cells by state and its
+      # most common values; an input column's filled rows and values. `columns:` picks the columns,
+      # `top_values:` how many values per column (1–100, default 20). (operationId: getDatasetSummary)
+      def summary(dataset_id, columns: nil, top_values: nil)
+        DatasetSummary.new(
+          @client.request(
+            :get, "/datasets/#{dataset_id}/summary",
+            query: { columns: columns && Array(columns).join(","), top_values: top_values }
+          )
+        )
       end
 
       # Append rows (≤ 100 per call) keyed by input column name; duplicates on the index

data/lib/riveter/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Riveter
-  VERSION = "1.0.1"
+  VERSION = "1.1.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: riveter-sdk
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Riveter
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-10-27 00:00:00.000000000 Z
+date: 2026-10-30 00:00:00.000000000 Z
 dependencies: []
 description: Client for the Riveter API — enrich data, build datasets, scrape pages,
   and run web searches. https://docs.riveterhq.com