llm_cost_tracker 0.7.0 → 0.7.2

data/app/views/llm_cost_tracker/data_quality/index.html.erb

@@ -1,9 +1,11 @@
-<% total = @stats.total_calls %>
-<% streaming_count = @stats.streaming_count %>
-<% streaming_missing_usage = @stats.streaming_missing_usage_count %>
-<% calls_with_provider_response_id = @stats.provider_response_id_column_present ? total - @stats.missing_provider_response_id_count : nil %>
-<% billable_tokens = @stats.input_tokens + @stats.output_tokens + @stats.cache_read_input_tokens.to_i + @stats.cache_write_input_tokens.to_i %>
-<% hidden_output_share = coverage_percent(@stats.hidden_output_tokens.to_i, @stats.output_tokens) %>
+<% total = @stats.total_calls.to_i %>
+<% unknown_pricing_count = @stats.unknown_pricing_count.to_i %>
+<% untagged_calls_count = @stats.untagged_calls_count.to_i %>
+<% missing_latency_count = @stats.missing_latency_count&.to_i %>
+<% streaming_count = @stats.streaming_count&.to_i %>
+<% streaming_missing_usage = @stats.streaming_missing_usage_count&.to_i %>
+<% missing_provider_response_id_count = @stats.missing_provider_response_id_count&.to_i %>
+<% calls_with_provider_response_id = total - missing_provider_response_id_count %>
 
 <section class="lct-panel lct-toolbar">
   <div class="lct-toolbar-head">
@@ -38,15 +40,13 @@
                        id: "lct-quality-model" %>
       </div>
 
-      <% if LlmCostTracker::LlmApiCall.stream_column? %>
-        <div class="lct-field">
-          <label for="lct-quality-stream">Stream</label>
-          <%= select_tag :stream,
-                         options_for_select(LlmCostTracker::DashboardFilterHelper::STREAM_FILTER_OPTIONS, params[:stream]),
-                         include_blank: "All calls",
-                         id: "lct-quality-stream" %>
-        </div>
-      <% end %>
+      <div class="lct-field">
+        <label for="lct-quality-stream">Stream</label>
+        <%= select_tag :stream,
+                       options_for_select(LlmCostTracker::DashboardFilterHelper::STREAM_FILTER_OPTIONS, params[:stream]),
+                       include_blank: "All calls",
+                       id: "lct-quality-stream" %>
+      </div>
 
       <div class="lct-filter-actions">
         <button class="lct-button" type="submit">Apply</button>
@@ -79,53 +79,47 @@
       <div class="lct-stat-grid">
         <article class="lct-stat">
           <p class="lct-stat-label">Unknown pricing</p>
-          <p class="lct-stat-value"><%= number(@stats.unknown_pricing_count) %></p>
-          <p class="lct-stat-sub"><%= percent(coverage_percent(@stats.unknown_pricing_count, total)) %> of calls</p>
+          <p class="lct-stat-value"><%= number(unknown_pricing_count) %></p>
+          <p class="lct-stat-sub"><%= percent(coverage_percent(unknown_pricing_count, total)) %> of calls</p>
         </article>
 
         <article class="lct-stat">
           <p class="lct-stat-label">Calls without tags</p>
-          <p class="lct-stat-value"><%= number(@stats.untagged_calls_count) %></p>
-          <p class="lct-stat-sub"><%= percent(coverage_percent(@stats.untagged_calls_count, total)) %> of calls</p>
+          <p class="lct-stat-value"><%= number(untagged_calls_count) %></p>
+          <p class="lct-stat-sub"><%= percent(coverage_percent(untagged_calls_count, total)) %> of calls</p>
         </article>
 
-        <% if @stats.latency_column_present %>
-          <article class="lct-stat">
-            <p class="lct-stat-label">Missing latency</p>
-            <p class="lct-stat-value"><%= number(@stats.missing_latency_count) %></p>
-            <p class="lct-stat-sub"><%= percent(coverage_percent(@stats.missing_latency_count, total)) %> of calls</p>
-          </article>
-        <% end %>
-
-        <% if @stats.stream_column_present %>
-          <article class="lct-stat">
-            <p class="lct-stat-label">Streaming calls</p>
-            <p class="lct-stat-value"><%= number(streaming_count) %></p>
-            <p class="lct-stat-sub"><%= percent(coverage_percent(streaming_count, total)) %> of calls</p>
-          </article>
+        <article class="lct-stat">
+          <p class="lct-stat-label">Missing latency</p>
+          <p class="lct-stat-value"><%= number(missing_latency_count) %></p>
+          <p class="lct-stat-sub"><%= percent(coverage_percent(missing_latency_count, total)) %> of calls</p>
+        </article>
 
-          <% if streaming_missing_usage && streaming_count.positive? %>
-            <article class="lct-stat">
-              <p class="lct-stat-label">Streams without usage</p>
-              <p class="lct-stat-value"><%= number(streaming_missing_usage) %></p>
-              <p class="lct-stat-sub"><%= percent(coverage_percent(streaming_missing_usage, streaming_count)) %> of streams</p>
-            </article>
-          <% end %>
-        <% end %>
+        <article class="lct-stat">
+          <p class="lct-stat-label">Streaming calls</p>
+          <p class="lct-stat-value"><%= number(streaming_count) %></p>
+          <p class="lct-stat-sub"><%= percent(coverage_percent(streaming_count, total)) %> of calls</p>
+        </article>
 
-        <% if @stats.provider_response_id_column_present %>
+        <% if streaming_count.positive? %>
           <article class="lct-stat">
-            <p class="lct-stat-label">Calls with provider response ID</p>
-            <p class="lct-stat-value"><%= number(calls_with_provider_response_id) %></p>
-            <p class="lct-stat-sub"><%= percent(coverage_percent(calls_with_provider_response_id, total)) %> of calls</p>
+            <p class="lct-stat-label">Streams without usage</p>
+            <p class="lct-stat-value"><%= number(streaming_missing_usage) %></p>
+            <p class="lct-stat-sub"><%= percent(coverage_percent(streaming_missing_usage, streaming_count)) %> of streams</p>
           </article>
         <% end %>
 
-        <% if @stats.usage_breakdown_column_present && @stats.output_tokens.positive? %>
+        <article class="lct-stat">
+          <p class="lct-stat-label">Calls with provider response ID</p>
+          <p class="lct-stat-value"><%= number(calls_with_provider_response_id) %></p>
+          <p class="lct-stat-sub"><%= percent(coverage_percent(calls_with_provider_response_id, total)) %> of calls</p>
+        </article>
+
+        <% if @hidden_output_summary %>
           <article class="lct-stat">
             <p class="lct-stat-label">Hidden output share</p>
-            <p class="lct-stat-value"><%= percent(hidden_output_share) %></p>
-            <p class="lct-stat-sub"><%= number(@stats.hidden_output_tokens) %> of <%= number(@stats.output_tokens) %> output tokens</p>
+            <p class="lct-stat-value"><%= percent(@hidden_output_summary.fetch(:share_percent)) %></p>
+            <p class="lct-stat-sub"><%= number(@hidden_output_summary.fetch(:hidden_output_tokens)) %> of <%= number(@hidden_output_summary.fetch(:output_tokens)) %> output tokens</p>
           </article>
         <% end %>
       </div>
@@ -151,33 +145,31 @@
           </tr>
         </thead>
         <tbody>
-          <% cost_coverage = coverage_percent(total - @stats.unknown_pricing_count, total) %>
+          <% cost_coverage = coverage_percent(total - unknown_pricing_count, total) %>
           <tr>
             <td>Cost (pricing known)</td>
             <td class="lct-num"><%= percent(cost_coverage) %></td>
-            <td class="lct-num"><%= number(total - @stats.unknown_pricing_count) %></td>
+            <td class="lct-num"><%= number(total - unknown_pricing_count) %></td>
             <td><%= render "llm_cost_tracker/shared/bar", value: cost_coverage, max: 100.0 %></td>
           </tr>
 
-          <% tag_coverage = coverage_percent(total - @stats.untagged_calls_count, total) %>
+          <% tag_coverage = coverage_percent(total - untagged_calls_count, total) %>
           <tr>
             <td>Tags (at least one tag)</td>
             <td class="lct-num"><%= percent(tag_coverage) %></td>
-            <td class="lct-num"><%= number(total - @stats.untagged_calls_count) %></td>
+            <td class="lct-num"><%= number(total - untagged_calls_count) %></td>
             <td><%= render "llm_cost_tracker/shared/bar", value: tag_coverage, max: 100.0 %></td>
           </tr>
 
-          <% if @stats.latency_column_present %>
-            <% latency_coverage = coverage_percent(total - @stats.missing_latency_count, total) %>
-            <tr>
-              <td>Latency</td>
-              <td class="lct-num"><%= percent(latency_coverage) %></td>
-              <td class="lct-num"><%= number(total - @stats.missing_latency_count) %></td>
-              <td><%= render "llm_cost_tracker/shared/bar", value: latency_coverage, max: 100.0 %></td>
-            </tr>
-          <% end %>
+          <% latency_coverage = coverage_percent(total - missing_latency_count, total) %>
+          <tr>
+            <td>Latency</td>
+            <td class="lct-num"><%= percent(latency_coverage) %></td>
+            <td class="lct-num"><%= number(total - missing_latency_count) %></td>
+            <td><%= render "llm_cost_tracker/shared/bar", value: latency_coverage, max: 100.0 %></td>
+          </tr>
 
-          <% if @stats.stream_column_present && streaming_count.to_i.positive? && streaming_missing_usage %>
+          <% if streaming_count.to_i.positive? %>
             <% stream_coverage = coverage_percent(streaming_count - streaming_missing_usage, streaming_count) %>
             <tr>
               <td>Streaming usage captured</td>
@@ -187,15 +179,13 @@
             </tr>
           <% end %>
 
-          <% if @stats.provider_response_id_column_present %>
-            <% provider_response_id_coverage = coverage_percent(calls_with_provider_response_id, total) %>
-            <tr>
-              <td>Provider response ID</td>
-              <td class="lct-num"><%= percent(provider_response_id_coverage) %></td>
-              <td class="lct-num"><%= number(calls_with_provider_response_id) %></td>
-              <td><%= render "llm_cost_tracker/shared/bar", value: provider_response_id_coverage, max: 100.0 %></td>
-            </tr>
-          <% end %>
+          <% provider_response_id_coverage = coverage_percent(calls_with_provider_response_id, total) %>
+          <tr>
+            <td>Provider response ID</td>
+            <td class="lct-num"><%= percent(provider_response_id_coverage) %></td>
+            <td class="lct-num"><%= number(calls_with_provider_response_id) %></td>
+            <td><%= render "llm_cost_tracker/shared/bar", value: provider_response_id_coverage, max: 100.0 %></td>
+          </tr>
         </tbody>
       </table>
     </section>
@@ -227,21 +217,19 @@
             <td>Attribution by tenant, user, or feature becomes less useful.</td>
             <td>Pass <code class="lct-code">tags:</code> from middleware using request context.</td>
           </tr>
-          <% if @stats.latency_column_present %>
-            <tr>
-              <td>Missing latency</td>
-              <td>Slow requests become harder to isolate on the calls page.</td>
-              <td>Make sure latency capture is enabled on every tracked request.</td>
-            </tr>
-          <% end %>
-          <% if @stats.stream_column_present && streaming_missing_usage.to_i.positive? %>
+          <tr>
+            <td>Missing latency</td>
+            <td>Slow requests become harder to isolate on the calls page.</td>
+            <td>Make sure latency capture is enabled on every tracked request.</td>
+          </tr>
+          <% if streaming_missing_usage.to_i.positive? %>
             <tr>
               <td>Streams without usage</td>
               <td>Token totals undercount when streaming responses drop the final usage event.</td>
               <td>Send OpenAI requests with <code class="lct-code">stream_options: { include_usage: true }</code>, or wrap custom clients with <code class="lct-code">LlmCostTracker.track_stream</code>.</td>
             </tr>
           <% end %>
-          <% if @stats.provider_response_id_column_present && @stats.missing_provider_response_id_count.to_i.positive? %>
+          <% if missing_provider_response_id_count.to_i.positive? %>
             <tr>
               <td>Missing provider response IDs</td>
               <td>Proof of provider-issued responses is weaker when calls cannot be tied back to provider objects.</td>
@@ -253,62 +241,47 @@
     </section>
   </section>
 
-  <% if @stats.usage_breakdown_column_present %>
-    <section class="lct-panel">
-      <div class="lct-section-head">
-        <div>
-          <h2 class="lct-section-title">Usage breakdown</h2>
-        </div>
+  <section class="lct-panel">
+    <div class="lct-section-head">
+      <div>
+        <h2 class="lct-section-title">Token usage</h2>
       </div>
+    </div>
 
-      <div class="lct-table-wrap">
-        <table class="lct-table lct-table-compact">
-          <thead>
-            <tr>
-              <th>Bucket</th>
-              <th class="lct-num">Tokens</th>
-              <th class="lct-num">Share</th>
-              <th class="lct-num">Cost</th>
-            </tr>
-          </thead>
-          <tbody>
-            <tr>
-              <td>Regular input</td>
-              <td class="lct-num"><%= number(@stats.input_tokens) %></td>
-              <td class="lct-num"><%= percent(coverage_percent(@stats.input_tokens, billable_tokens)) %></td>
-              <td class="lct-num"><%= money(@stats.input_cost) %></td>
-            </tr>
-            <tr>
-              <td>Cache read input</td>
-              <td class="lct-num"><%= number(@stats.cache_read_input_tokens) %></td>
-              <td class="lct-num"><%= percent(coverage_percent(@stats.cache_read_input_tokens, billable_tokens)) %></td>
-              <td class="lct-num<%= ' lct-num-muted' if @stats.cache_read_input_cost.nil? %>"><%= optional_money(@stats.cache_read_input_cost) %></td>
-            </tr>
-            <tr>
-              <td>Cache write input</td>
-              <td class="lct-num"><%= number(@stats.cache_write_input_tokens) %></td>
-              <td class="lct-num"><%= percent(coverage_percent(@stats.cache_write_input_tokens, billable_tokens)) %></td>
-              <td class="lct-num<%= ' lct-num-muted' if @stats.cache_write_input_cost.nil? %>"><%= optional_money(@stats.cache_write_input_cost) %></td>
-            </tr>
-            <tr>
-              <td>Output</td>
-              <td class="lct-num"><%= number(@stats.output_tokens) %></td>
-              <td class="lct-num"><%= percent(coverage_percent(@stats.output_tokens, billable_tokens)) %></td>
-              <td class="lct-num"><%= money(@stats.output_cost) %></td>
-            </tr>
+    <div class="lct-table-wrap">
+      <table class="lct-table lct-table-compact">
+        <thead>
+          <tr>
+            <th>Bucket</th>
+            <th class="lct-num">Tokens</th>
+            <th class="lct-num">Share</th>
+            <th class="lct-num">Cost</th>
+          </tr>
+        </thead>
+        <tbody>
+          <% @usage_rows.each do |row| %>
+            <% token_key = row.fetch(:token_key) %>
+            <% cost_key = row.fetch(:cost_key) %>
+            <% cost_value = row.fetch(:cost_value) %>
             <tr>
-              <td>Hidden output</td>
-              <td class="lct-num"><%= number(@stats.hidden_output_tokens) %></td>
-              <td class="lct-num"><%= percent(hidden_output_share) %> of output</td>
-              <td class="lct-num lct-num-muted">n/a</td>
+              <td><%= token_usage_quality_label(token_key) %></td>
+              <td class="lct-num"><%= number(row.fetch(:token_value)) %></td>
+              <% if row.fetch(:share_basis) == :output %>
+                <td class="lct-num"><%= percent(row.fetch(:share_percent)) %> of output</td>
+              <% else %>
+                <td class="lct-num"><%= percent(row.fetch(:share_percent)) %></td>
+              <% end %>
+              <td class="lct-num<%= ' lct-num-muted' if cost_key.nil? || cost_value.nil? %>">
+                <%= cost_key ? optional_money(cost_value) : "n/a" %>
+              </td>
             </tr>
-          </tbody>
-        </table>
-      </div>
-    </section>
-  <% end %>
+          <% end %>
+        </tbody>
+      </table>
+    </div>
+  </section>
 
-  <% unless @stats.unknown_pricing_by_model.empty? %>
+  <% unless @unknown_pricing_by_model.empty? %>
     <section class="lct-panel">
       <div class="lct-section-head">
         <div>
@@ -328,11 +301,11 @@
             </tr>
           </thead>
           <tbody>
-            <% @stats.unknown_pricing_by_model.each do |model, count| %>
+            <% @unknown_pricing_by_model.each do |row| %>
               <tr>
-                <td><code class="lct-code"><%= model %></code></td>
-                <td class="lct-num"><%= number(count) %></td>
-                <td class="lct-num"><%= percent(total.positive? ? (count.to_f / total) * 100.0 : 0.0) %></td>
+                <td><code class="lct-code"><%= row.model %></code></td>
+                <td class="lct-num"><%= number(row.calls) %></td>
+                <td class="lct-num"><%= percent(total.positive? ? (row.calls.to_f / total) * 100.0 : 0.0) %></td>
               </tr>
             <% end %>
           </tbody>

data/app/views/llm_cost_tracker/models/index.html.erb

@@ -37,8 +37,8 @@
                        options_for_select(
                          [["Total spend", "cost"],
                           ["Call volume", "calls"],
-                          ["Avg cost / call", "avg_cost"]] +
-                         (@latency_available ? [["Avg latency", "latency"]] : []),
+                          ["Avg cost / call", "avg_cost"],
+                          ["Avg latency", "latency"]],
                          @sort.presence || "cost"
                        ),
                        id: "lct-models-sort" %>
@@ -71,13 +71,12 @@
             <th>Provider</th>
             <th>Model</th>
             <th class="lct-num">Calls</th>
-            <th class="lct-num">Input</th>
+            <th class="lct-num">Total tokens</th>
+            <th class="lct-num">Regular input</th>
             <th class="lct-num">Output</th>
             <th class="lct-num">Total cost</th>
             <th class="lct-num">Avg cost / call</th>
-            <% if @latency_available %>
-              <th class="lct-num">Avg latency</th>
-            <% end %>
+            <th class="lct-num">Avg latency</th>
             <th></th>
           </tr>
         </thead>
@@ -87,13 +86,13 @@
               <td><%= row.provider %></td>
               <td><code class="lct-code"><%= row.model %></code></td>
               <td class="lct-num"><%= number(row.calls) %></td>
+              <td class="lct-num"><%= format_tokens(row.total_tokens) %></td>
               <td class="lct-num"><%= format_tokens(row.input_tokens) %></td>
               <td class="lct-num"><%= format_tokens(row.output_tokens) %></td>
               <td class="lct-num"><%= money(row.total_cost) %></td>
               <td class="lct-num"><%= money(row.average_cost_per_call) %></td>
-              <% if @latency_available %>
-                <td class="lct-num<%= ' lct-num-muted' if row.average_latency_ms.nil? %>"><%= row.average_latency_ms ? "#{number(row.average_latency_ms.round)}ms" : "n/a" %></td>
-              <% end %>
+              <% average_latency_ms = row.average_latency_ms %>
+              <td class="lct-num<%= ' lct-num-muted' if average_latency_ms.nil? %>"><%= average_latency_ms ? "#{number(average_latency_ms.round)}ms" : "n/a" %></td>
               <td><%= link_to "Calls", calls_path(calls_query_for_model(provider: row.provider, model: row.model)), class: "lct-button lct-button-secondary lct-button-compact" %></td>
             </tr>
           <% end %>

data/app/views/llm_cost_tracker/shared/setup_required.html.erb

@@ -1,7 +1,18 @@
 <section class="lct-panel lct-empty">
   <h2 class="lct-state-title">Setup required</h2>
   <p class="lct-state-copy">
-    The <span class="lct-code">llm_api_calls</span> table is not available yet.
-    Run <span class="lct-code">rails generate llm_cost_tracker:install</span> and migrate your database.
+    <%= @setup_message || "The llm_api_calls table is not available yet." %>
+    <% if @setup_details.present? %>
+      Run <span class="lct-code">bin/rails llm_cost_tracker:doctor</span>, apply the listed migrations, and migrate your database.
+    <% else %>
+      Run <span class="lct-code">rails generate llm_cost_tracker:install</span> and migrate your database.
+    <% end %>
   </p>
+  <% if @setup_details.present? %>
+    <ul class="lct-state-copy">
+      <% @setup_details.each do |detail| %>
+        <li><code class="lct-code"><%= detail %></code></li>
+      <% end %>
+    </ul>
+  <% end %>
 </section>

data/app/views/llm_cost_tracker/tags/show.html.erb

@@ -1,14 +1,14 @@
-<% share_base = @tagged_calls.positive? ? @tagged_calls.to_f : 1.0 %>
+<% share_base = @breakdown.tagged_calls.positive? ? @breakdown.tagged_calls.to_f : 1.0 %>
 
 <section class="lct-panel lct-toolbar">
   <div class="lct-toolbar-head">
     <div>
       <p class="lct-muted"><%= link_to "← All tag keys", tags_path(current_query) %></p>
-      <h2 class="lct-section-title">Tag: <code class="lct-code"><%= @tag_key %></code></h2>
+      <h2 class="lct-section-title">Tag: <code class="lct-code"><%= params[:key] %></code></h2>
     </div>
   </div>
 
-  <form class="lct-filters" action="<%= tag_path(@tag_key) %>" method="get">
+  <form class="lct-filters" action="<%= tag_path(params[:key]) %>" method="get">
     <div class="lct-filter-row lct-filter-row-basic">
       <div class="lct-field">
         <label for="lct-tag-show-from">From</label>
@@ -38,49 +38,49 @@
 
       <div class="lct-filter-actions">
         <button class="lct-button" type="submit">Apply</button>
-        <%= link_to("Reset", tag_path(@tag_key), class: "lct-button lct-button-secondary") if any_filter_applied? %>
+        <%= link_to("Reset", tag_path(params[:key]), class: "lct-button lct-button-secondary") if any_filter_applied? %>
       </div>
     </div>
   </form>
 
-  <%= render "llm_cost_tracker/shared/active_filters", chips: active_tag_filters, clear_path: tag_path(@tag_key) %>
+  <%= render "llm_cost_tracker/shared/active_filters", chips: active_tag_filters, clear_path: tag_path(params[:key]) %>
 
   <p class="lct-summary-row">
-    <span><strong><%= number(@tagged_calls) %></strong> tagged calls</span>
-    <span><strong><%= percent(coverage_percent(@tagged_calls, @total_calls)) %></strong> coverage</span>
-    <span><strong><%= number(@distinct_values) %></strong> distinct values</span>
+    <span><strong><%= number(@breakdown.tagged_calls) %></strong> tagged calls</span>
+    <span><strong><%= percent(coverage_percent(@breakdown.tagged_calls, @breakdown.total_calls)) %></strong> coverage</span>
+    <span><strong><%= number(@breakdown.distinct_values) %></strong> distinct values</span>
   </p>
 
-  <% if @tag_values_limited %>
-    <p class="lct-toolbar-note">Showing top <%= number(@tag_value_limit) %> values by spend.</p>
+  <% if @breakdown.distinct_values > @breakdown.rows.size %>
+    <p class="lct-toolbar-note">Showing top <%= number(@breakdown.limit) %> values by spend.</p>
   <% end %>
 </section>
 
-<% if @rows.empty? %>
+<% if @breakdown.rows.empty? %>
   <section class="lct-panel lct-empty">
-    <h2 class="lct-state-title">No calls tagged with <%= @tag_key %></h2>
+    <h2 class="lct-state-title">No calls tagged with <%= params[:key] %></h2>
     <p class="lct-state-copy">Values for this key will appear here once matching calls carry the tag in the current slice.</p>
     <div class="lct-state-actions">
-      <%= link_to "Clear filters", tag_path(@tag_key), class: "lct-button lct-button-secondary" %>
+      <%= link_to "Clear filters", tag_path(params[:key]), class: "lct-button lct-button-secondary" %>
     </div>
   </section>
 <% else %>
   <section class="lct-stat-grid lct-stat-grid-spaced">
     <article class="lct-stat">
       <p class="lct-stat-label">Tagged calls</p>
-      <p class="lct-stat-value"><%= number(@tagged_calls) %></p>
-      <p class="lct-stat-copy">Rows that include <code class="lct-code"><%= @tag_key %></code></p>
+      <p class="lct-stat-value"><%= number(@breakdown.tagged_calls) %></p>
+      <p class="lct-stat-copy">Rows that include <code class="lct-code"><%= params[:key] %></code></p>
     </article>
 
     <article class="lct-stat">
       <p class="lct-stat-label">Coverage</p>
-      <p class="lct-stat-value"><%= percent(coverage_percent(@tagged_calls, @total_calls)) %></p>
-      <p class="lct-stat-copy"><%= number(@total_calls) %> total calls in this slice</p>
+      <p class="lct-stat-value"><%= percent(coverage_percent(@breakdown.tagged_calls, @breakdown.total_calls)) %></p>
+      <p class="lct-stat-copy"><%= number(@breakdown.total_calls) %> total calls in this slice</p>
     </article>
 
     <article class="lct-stat">
       <p class="lct-stat-label">Distinct values</p>
-      <p class="lct-stat-value"><%= number(@distinct_values) %></p>
+      <p class="lct-stat-value"><%= number(@breakdown.distinct_values) %></p>
       <p class="lct-stat-copy">Unique values currently visible</p>
     </article>
   </section>
@@ -99,7 +99,7 @@
           </tr>
         </thead>
         <tbody>
-          <% @rows.each do |row| %>
+          <% @breakdown.rows.each do |row| %>
             <tr>
               <td><code class="lct-code"><%= row.value %></code></td>
               <td class="lct-num"><%= number(row.calls) %></td>
@@ -110,7 +110,7 @@
                 <% if row.value == "(untagged)" %>
                   <span class="lct-muted">n/a</span>
                 <% else %>
-                  <%= link_to "Calls", calls_path(calls_query_for_tag(key: @tag_key, value: row.value)), class: "lct-button lct-button-secondary lct-button-compact" %>
+                  <%= link_to "Calls", calls_path(calls_query_for_tag(key: params[:key], value: row.value)), class: "lct-button lct-button-secondary lct-button-compact" %>
                 <% end %>
               </td>
             </tr>

data/lib/llm_cost_tracker/budget.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative "logging"
+require_relative "ledger"
 
 module LlmCostTracker
   class Budget
@@ -12,7 +13,7 @@ module LlmCostTracker
         budgets = enforce_period_budgets(config)
         return if budgets.empty?
 
-        totals = active_record_totals(budgets.keys, time: Time.now.utc)
+        totals = LlmCostTracker::Ledger::Period::Totals.call(budgets.keys, time: Time.now.utc)
 
         budgets.each do |period, budget|
           total = totals.fetch(period)
@@ -23,7 +24,7 @@ module LlmCostTracker
 
       def check!(event)
         config = LlmCostTracker.configuration
-        return unless event.cost
+        return unless event.total_cost
 
         check_per_call_budget(event, config)
         budgets = check_period_budgets(config)
@@ -42,7 +43,7 @@ module LlmCostTracker
         budget = config.per_call_budget
         return unless budget
 
-        call_cost = event.cost.total_cost
+        call_cost = event.total_cost
         return unless call_cost >= budget
 
         handle_exceeded(budget_type: :per_call, total: call_cost, budget: budget, last_event: event)
@@ -65,16 +66,7 @@ module LlmCostTracker
       def totals_for_check(event, budgets)
         return {} if budgets.empty?
 
-        active_record_totals(budgets.keys, time: event.tracked_at)
-      end
-
-      def active_record_totals(periods, time:)
-        require_relative "llm_api_call" unless defined?(LlmCostTracker::LlmApiCall)
-        require_relative "storage/active_record_store" unless defined?(LlmCostTracker::Storage::ActiveRecordStore)
-
-        LlmCostTracker::Storage::ActiveRecordStore.period_totals(periods, time: time)
-      rescue LoadError => e
-        raise Error, "ActiveRecord storage requires the active_record gem: #{e.message}"
+        LlmCostTracker::Ledger::Period::Totals.call(budgets.keys, time: event.tracked_at)
       end
 
       def handle_exceeded(budget_type:, total:, budget:, last_event: nil)
@@ -89,7 +81,7 @@ module LlmCostTracker
         if notify_exceeded?(config, budget_type: budget_type, total: total, budget: budget, last_event: last_event)
           config.on_budget_exceeded&.call(payload)
         end
-        raise BudgetExceededError.new(**payload) if raise_on_exceeded?(config)
+        raise BudgetExceededError.new(**payload) if %i[raise block_requests].include?(config.budget_exceeded_behavior)
       end
 
       def budget_payload(budget_type:, total:, budget:, last_event:)
@@ -108,14 +100,10 @@ module LlmCostTracker
       def notify_exceeded?(config, budget_type:, total:, budget:, last_event:)
         return false unless config.on_budget_exceeded
         return true unless config.budget_exceeded_behavior == :notify
-        return true unless last_event&.cost
+        return true unless last_event&.total_cost
         return true if budget_type == :per_call
 
-        total - last_event.cost.total_cost < budget
-      end
-
-      def raise_on_exceeded?(config)
-        %i[raise block_requests].include?(config.budget_exceeded_behavior)
+        total - last_event.total_cost < budget
       end
     end
   end

data/lib/llm_cost_tracker/capture/stream.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module LlmCostTracker
+  module Capture
+    module Stream
+      LIMIT_BYTES = 1_048_576
+    end
+  end
+end