completion-kit 0.28.10 → 0.28.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8eb41960ab1c047932e577095faa9bd23518a82b2a23ee30c84f862a32afc2d5
-  data.tar.gz: e4d577875390f29ca7b5ddd78415734fabe9d1b1de910a83ad925248d9a14809
+  metadata.gz: 6f1cd3f910ce4c3f45e96973a76be5d1bbc3e95ad5ac89eb5a685b8c128bd9e6
+  data.tar.gz: 3b464cf7ccf32f68f83f0356f11b77e8329260b1b72c56e8dea1748f760a89a4
 SHA512:
-  metadata.gz: 3e820588ee5b204628a8b317d4ce9d66f66053795d9159c6c90ea325c9ac45cf940cb01ba09878cc85cd298b452090f6309654afe3a4db946bff9212a88ee85f
-  data.tar.gz: 351ea8ff2c5012dfc5221d5dd4daf27e454b90d7c7cf63ac9839e76bc69e3f9e3b6e363331a257d9c65ba8bb73e375e1c9953cba36b875fbeb6f3c38583a1735
+  metadata.gz: 367e68a324e65f9c868e437a9c9d7ff0ec75e0da3cad77c3f717d2c970f024240a85d182df333735e0b8b455d1981949132f96c39298f7ae41b7f0e2a6d226cd
+  data.tar.gz: 149ec67cef9b3a3ae6bc1b071dc4c0e4f293b7dd2db12a7ab44a53338918ac8a199159f38610fc610bf4fb06d38c912c2186e4524cb3f62de721f002d5f91165

data/app/models/completion_kit/dataset.rb

@@ -19,9 +19,13 @@ module CompletionKit
       return 0 if csv_data.blank?
 
       require "csv"
-      ::CSV.parse(csv_data, headers: true).length
-    rescue ::CSV::MalformedCSVError
-      0
+      @row_count ||= begin
+        count = 0
+        ::CSV.new(csv_data, headers: true).each { count += 1 }
+        count
+      rescue ::CSV::MalformedCSVError
+        0
+      end
     end
 
     def headers

data/app/models/completion_kit/run.rb

@@ -116,16 +116,19 @@ module CompletionKit
       end
     end
 
+    def reviews_for_summary
+      @reviews_for_summary ||= Review.where(response_id: responses.select(:id)).to_a
+    end
+
     def avg_score
-      all_reviews = responses.flat_map(&:reviews)
-      scores = all_reviews.map(&:ai_score).compact.map(&:to_f)
+      scores = reviews_for_summary.map(&:ai_score).compact.map(&:to_f)
       return nil if scores.empty?
 
       (scores.sum / scores.length).round(2)
     end
 
     def metric_averages
-      responses.flat_map(&:reviews).group_by(&:metric_name).filter_map do |name, reviews|
+      reviews_for_summary.group_by(&:metric_name).filter_map do |name, reviews|
         scored = reviews.select { |r| r.ai_score.present? }
         if scored.any?
           scores = scored.map { |r| r.ai_score.to_f }
@@ -141,7 +144,7 @@ module CompletionKit
     end
 
     def check_pass_rate
-      resolved = responses.flat_map(&:reviews).reject { |r| r.passed.nil? }
+      resolved = reviews_for_summary.reject { |r| r.passed.nil? }
       return nil if resolved.empty?
 
       passed = resolved.count { |r| r.passed == true }

data/app/views/completion_kit/runs/show.html.erb

@@ -14,10 +14,6 @@
   </div>
 <% end %>
 
-<% if @run.dataset %>
-  <% dataset_lines = @run.dataset.csv_data.to_s.lines %>
-  <% dataset_preview_lines = dataset_lines.first(50) %>
-<% end %>
 
 <% if CompletionKit.config.judge_agreement_enabled %>
   <% stale_summary = @run.stale_review_summary %>
@@ -130,16 +126,22 @@
       <div class="ck-modal__body">
         <%
           require "csv"
-          parsed_rows = []
           parse_error = nil
+          headers = []
+          body_rows = []
+          remaining = 0
           begin
-            parsed_rows = ::CSV.parse(@run.dataset.csv_data.to_s)
+            reader = ::CSV.new(@run.dataset.csv_data.to_s)
+            headers = reader.shift || []
+            50.times do
+              row = reader.shift
+              break unless row
+              body_rows << row
+            end
+            remaining = [@run.dataset.row_count - body_rows.length, 0].max
           rescue ::CSV::MalformedCSVError => e
             parse_error = e.message
           end
-          headers = parsed_rows.first || []
-          body_rows = parsed_rows.drop(1).first(50)
-          remaining = parsed_rows.drop(1).count - body_rows.count
         %>
         <% if parse_error %>
           <p class="ck-field-hint" style="color: var(--ck-warning); padding: 1rem;">Could not parse CSV: <%= parse_error %></p>

data/lib/completion_kit/version.rb

@@ -1,3 +1,3 @@
 module CompletionKit
-  VERSION = "0.28.10"
+  VERSION = "0.28.11"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: completion-kit
 version: !ruby/object:Gem::Version
-  version: 0.28.10
+  version: 0.28.11
 platform: ruby
 authors:
 - Damien Bastin