llm_cost_tracker 0.2.0.alpha1 → 0.2.0

data/app/views/llm_cost_tracker/errors/invalid_filter.html.erb

@@ -1,4 +1,4 @@
-<section class="lct-panel">
-  <h2>Invalid filter</h2>
-  <p class="lct-muted"><%= @error_message %></p>
+<section class="lct-panel lct-empty">
+  <h2 class="lct-state-title">Invalid filter</h2>
+  <p class="lct-state-copy"><%= @error_message %></p>
 </section>

data/app/views/llm_cost_tracker/errors/not_found.html.erb

@@ -1,5 +1,5 @@
-<section class="lct-panel">
-  <h2>Call not found</h2>
-  <p class="lct-muted">The requested LLM API call could not be found.</p>
+<section class="lct-panel lct-empty">
+  <h2 class="lct-state-title">Call not found</h2>
+  <p class="lct-state-copy">The requested LLM API call could not be found.</p>
   <p><%= link_to "Back to calls", calls_path, class: "lct-button lct-button-secondary" %></p>
 </section>

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

@@ -1,91 +1,99 @@
-<section class="lct-panel">
-  <h2 class="lct-section-title">Models</h2>
+<section class="lct-panel lct-toolbar">
+  <div class="lct-toolbar-head">
+    <h2 class="lct-section-title">Models</h2>
+  </div>
+
   <form class="lct-filters" action="<%= models_path %>" method="get">
-    <div class="lct-field">
-      <label for="lct-from">From</label>
-      <input id="lct-from" type="date" name="from" value="<%= params[:from] %>">
-    </div>
+    <div class="lct-filter-row lct-filter-row-with-sort">
+      <div class="lct-field">
+        <label for="lct-models-from">From</label>
+        <input id="lct-models-from" data-lct-filter-input type="date" name="from" value="<%= params[:from] %>">
+      </div>
 
-    <div class="lct-field">
-      <label for="lct-to">To</label>
-      <input id="lct-to" type="date" name="to" value="<%= params[:to] %>">
-    </div>
+      <div class="lct-field">
+        <label for="lct-models-to">To</label>
+        <input id="lct-models-to" type="date" name="to" value="<%= params[:to] %>">
+      </div>
 
-    <div class="lct-field">
-      <label for="lct-provider">Provider</label>
-      <input id="lct-provider" type="text" name="provider" value="<%= params[:provider] %>">
-    </div>
+      <div class="lct-field">
+        <label for="lct-models-provider">Provider</label>
+        <%= select_tag :provider,
+                       options_for_select(provider_filter_options, params[:provider]),
+                       include_blank: "All providers",
+                       id: "lct-models-provider" %>
+      </div>
 
-    <div class="lct-field">
-      <label for="lct-model">Model</label>
-      <input id="lct-model" type="text" name="model" value="<%= params[:model] %>">
-    </div>
-
-    <div class="lct-field">
-      <label for="lct-tag-key">Tag key</label>
-      <input id="lct-tag-key" type="text" name="tag_key" value="<%= params[:tag_key] %>">
-    </div>
+      <div class="lct-field">
+        <label for="lct-models-model">Model</label>
+        <%= select_tag :model,
+                       options_for_select(model_filter_options, params[:model]),
+                       include_blank: "All models",
+                       id: "lct-models-model" %>
+      </div>
 
-    <div class="lct-field">
-      <label for="lct-tag-value">Tag value</label>
-      <input id="lct-tag-value" type="text" name="tag_value" value="<%= params[:tag_value] %>">
-    </div>
-
-    <div class="lct-field">
-      <label for="lct-sort">Sort by</label>
-      <select id="lct-sort" name="sort">
-        <option value="cost"     <%= "selected" if @sort.blank? || @sort == "cost" %>>Total spend</option>
-        <option value="calls"    <%= "selected" if @sort == "calls" %>>Call volume</option>
-        <option value="avg_cost" <%= "selected" if @sort == "avg_cost" %>>Avg cost / call</option>
-        <% if @latency_available %>
-          <option value="latency" <%= "selected" if @sort == "latency" %>>Avg latency</option>
-        <% end %>
-      </select>
-    </div>
+      <div class="lct-field">
+        <label for="lct-models-sort">Sort</label>
+        <select id="lct-models-sort" name="sort">
+          <option value="cost" <%= "selected" if @sort.blank? || @sort == "cost" %>>Total spend</option>
+          <option value="calls" <%= "selected" if @sort == "calls" %>>Call volume</option>
+          <option value="avg_cost" <%= "selected" if @sort == "avg_cost" %>>Avg cost / call</option>
+          <% if @latency_available %>
+            <option value="latency" <%= "selected" if @sort == "latency" %>>Avg latency</option>
+          <% end %>
+        </select>
+      </div>
 
-    <div class="lct-button-row">
-      <button class="lct-button" type="submit">Apply</button>
-      <%= link_to "Reset", models_path, class: "lct-button lct-button-secondary" %>
+      <div class="lct-filter-actions">
+        <button class="lct-button" type="submit">Apply</button>
+        <%= link_to("Reset", models_path, 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: models_path %>
 </section>
 
 <% if @rows.empty? %>
   <section class="lct-panel lct-empty">
-    <h2 class="lct-section-title">No models</h2>
-    <p class="lct-muted">Tracked models will appear here when they match the current filters.</p>
+    <h2 class="lct-state-title">No models in this slice</h2>
+    <p class="lct-state-copy">Tracked models will appear here once calls match the current provider, model, and date filters.</p>
+    <div class="lct-state-actions">
+      <%= link_to "Clear filters", models_path, class: "lct-button lct-button-secondary" %>
+    </div>
   </section>
 <% else %>
   <section class="lct-panel">
     <div class="lct-table-wrap">
-      <table class="lct-table">
+      <table class="lct-table lct-table-compact">
         <thead>
           <tr>
             <th>Provider</th>
             <th>Model</th>
-            <th>Calls</th>
-            <th>Input</th>
-            <th>Output</th>
-            <th>Total cost</th>
-            <th>Avg cost / call</th>
+            <th class="lct-num">Calls</th>
+            <th class="lct-num">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>Avg latency</th>
+              <th class="lct-num">Avg latency</th>
             <% end %>
+            <th></th>
           </tr>
         </thead>
         <tbody>
           <% @rows.each do |row| %>
             <tr>
               <td><%= row.provider %></td>
-              <td><%= row.model %></td>
-              <td><%= number(row.calls) %></td>
-              <td><%= format_tokens(row.input_tokens) %></td>
-              <td><%= format_tokens(row.output_tokens) %></td>
-              <td><%= money(row.total_cost) %></td>
-              <td><%= money(row.average_cost_per_call) %></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.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><%= row.average_latency_ms ? "#{number(row.average_latency_ms.round)}ms" : "n/a" %></td>
+                <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 %>
+              <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 %>
         </tbody>

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

@@ -0,0 +1,16 @@
+<% if chips.any? %>
+  <div class="lct-chip-row" aria-label="Active filters">
+    <% chips.each do |chip| %>
+      <span class="lct-chip">
+        <span class="lct-chip-label"><%= chip[:label] %></span>
+        <span><%= chip[:value] %></span>
+        <% if chip[:path] %>
+          <%= link_to "×", chip[:path], class: "lct-chip-remove", aria: { label: "Remove #{chip[:label]} #{chip[:value]}" } %>
+        <% end %>
+      </span>
+    <% end %>
+    <% if local_assigns[:clear_path] %>
+      <%= link_to "Clear all", clear_path, class: "lct-clear-link" %>
+    <% end %>
+  </div>
+<% end %>

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

@@ -0,0 +1,23 @@
+<% visible_segments = stack_segments(segments) %>
+
+<% if visible_segments.empty? %>
+  <p class="lct-call-breakdown-empty"><%= empty_message %></p>
+<% else %>
+  <div class="lct-stack-track" aria-hidden="true">
+    <% visible_segments.each do |segment| %>
+      <span class="lct-stack-fill <%= segment[:css_class] %>" style="width: <%= segment[:percent].round(2) %>%"></span>
+    <% end %>
+  </div>
+
+  <div class="lct-stack-legend">
+    <% visible_segments.each do |segment| %>
+      <div class="lct-stack-legend-item">
+        <span class="lct-stack-key">
+          <span class="lct-stack-swatch <%= segment[:css_class] %>"></span>
+          <span><%= segment[:label] %></span>
+        </span>
+        <span class="lct-stack-meta"><%= segment[:formatted_value] %> · <%= percent(segment[:percent]) %></span>
+      </div>
+    <% end %>
+  </div>
+<% end %>

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

@@ -0,0 +1,18 @@
+<%# locals: series: Array[{ label:, cost: }], comparison_series: nil %>
+<% if series.blank? %>
+  <div class="lct-chart-empty">No spend in this range.</div>
+<% else %>
+  <%= spend_chart_svg(series, comparison_points: local_assigns[:comparison_series]) %>
+  <div class="lct-chart-legend">
+    <span><%= series.first[:label] %></span>
+    <% if local_assigns[:comparison_series].present? %>
+      <span class="lct-chart-legend-compare">
+        <span class="lct-chart-key"><span class="lct-chart-key-line"></span> Current</span>
+        <span class="lct-chart-key"><span class="lct-chart-key-line lct-chart-key-line-secondary"></span> Previous</span>
+      </span>
+    <% else %>
+      <span>Peak <%= money(series.map { |p| p[:cost] }.max) %></span>
+    <% end %>
+    <span><%= series.last[:label] %></span>
+  </div>
+<% end %>

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

@@ -0,0 +1,15 @@
+<%# locals: tags: Hash, limit: Integer (optional) %>
+<% entries = tag_chip_entries(tags, limit: local_assigns.fetch(:limit, 3)) %>
+<% if entries.empty? %>
+  <span class="lct-tag-empty">(untagged)</span>
+<% else %>
+  <span class="lct-tag-chips" title="<%= safe_json(tags) %>">
+    <% entries.each do |entry| %>
+      <% if entry[:more] %>
+        <span class="lct-tag-chip lct-tag-chip-more">+<%= entry[:more] %></span>
+      <% else %>
+        <span class="lct-tag-chip"><%= entry[:key] %>=<%= entry[:value] %></span>
+      <% end %>
+    <% end %>
+  </span>
+<% end %>

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

@@ -1,5 +1,6 @@
-<section class="lct-panel">
-  <p class="lct-muted">
+<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.
   </p>

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

@@ -1,19 +1,63 @@
+<section class="lct-panel lct-toolbar">
+  <div class="lct-toolbar-head">
+    <h2 class="lct-section-title">Tag keys</h2>
+  </div>
+
+  <form class="lct-filters" action="<%= tags_path %>" method="get">
+    <div class="lct-filter-row lct-filter-row-basic">
+      <div class="lct-field">
+        <label for="lct-tags-from">From</label>
+        <input id="lct-tags-from" data-lct-filter-input type="date" name="from" value="<%= params[:from] %>">
+      </div>
+
+      <div class="lct-field">
+        <label for="lct-tags-to">To</label>
+        <input id="lct-tags-to" type="date" name="to" value="<%= params[:to] %>">
+      </div>
+
+      <div class="lct-field">
+        <label for="lct-tags-provider">Provider</label>
+        <%= select_tag :provider,
+                       options_for_select(provider_filter_options, params[:provider]),
+                       include_blank: "All providers",
+                       id: "lct-tags-provider" %>
+      </div>
+
+      <div class="lct-field">
+        <label for="lct-tags-model">Model</label>
+        <%= select_tag :model,
+                       options_for_select(model_filter_options, params[:model]),
+                       include_blank: "All models",
+                       id: "lct-tags-model" %>
+      </div>
+
+      <div class="lct-filter-actions">
+        <button class="lct-button" type="submit">Apply</button>
+        <%= link_to("Reset", tags_path, 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: tags_path %>
+</section>
+
 <% if @rows.empty? %>
   <section class="lct-panel lct-empty">
-    <h2 class="lct-section-title">No tag keys found</h2>
-    <p class="lct-muted">No calls with tags have been recorded yet. Tag keys will appear here once data is available.</p>
+    <h2 class="lct-state-title">No tag keys found</h2>
+    <p class="lct-state-copy">Tag keys will appear here once tagged calls exist in the current slice.</p>
+    <div class="lct-state-actions">
+      <%= link_to "Clear filters", tags_path, class: "lct-button lct-button-secondary" %>
+    </div>
   </section>
 <% else %>
   <section class="lct-panel">
-    <h2 class="lct-section-title">Tag Keys</h2>
-    <p class="lct-muted">All tag keys present in the dataset. Click a key to explore spend and call distribution by value.</p>
     <div class="lct-table-wrap">
-      <table class="lct-table">
+      <table class="lct-table lct-table-compact">
         <thead>
           <tr>
-            <th>Tag Key</th>
-            <th>Calls with this key</th>
-            <th>Distinct values</th>
+            <th>Tag key</th>
+            <th class="lct-num">Calls with this key</th>
+            <th class="lct-num">Distinct values</th>
             <th></th>
           </tr>
         </thead>
@@ -21,9 +65,9 @@
           <% @rows.each do |row| %>
             <tr>
               <td><code class="lct-code"><%= row.key %></code></td>
-              <td><%= number(row.calls_count) %></td>
-              <td><%= number(row.distinct_values) %></td>
-              <td><%= link_to "Breakdown →", tag_path(row.key), class: "lct-button lct-button-secondary" %></td>
+              <td class="lct-num"><%= number(row.calls_count) %></td>
+              <td class="lct-num"><%= number(row.distinct_values) %></td>
+              <td><%= link_to "Breakdown", tag_path(row.key, current_query), class: "lct-button lct-button-secondary lct-button-compact" %></td>
             </tr>
           <% end %>
         </tbody>
@@ -31,4 +75,3 @@
     </div>
   </section>
 <% end %>
-

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

@@ -1,65 +1,114 @@
-<section class="lct-panel">
-  <p class="lct-muted"><%= link_to "← All tag keys", tags_path %></p>
-  <h2 class="lct-section-title">Tag key: <code class="lct-code"><%= @tag_key %></code></h2>
-  <% if @total_calls.positive? %>
-    <p class="lct-muted">
-      <%= number(@tagged_calls) %> of <%= number(@total_calls) %> calls have this key
-      (<%= percent(coverage_percent(@tagged_calls, @total_calls)) %> coverage) &middot;
-      <%= number(@distinct_values) %> distinct <%= @distinct_values == 1 ? "value" : "values" %>
-    </p>
-  <% end %>
-  <form class="lct-filters" action="<%= tag_path(@tag_key) %>" method="get">
-    <div class="lct-field">
-      <label for="lct-from">From</label>
-      <input id="lct-from" type="date" name="from" value="<%= params[:from] %>">
-    </div>
+<% share_base = @tagged_calls.positive? ? @tagged_calls.to_f : 1.0 %>
 
-    <div class="lct-field">
-      <label for="lct-to">To</label>
-      <input id="lct-to" type="date" name="to" value="<%= params[:to] %>">
+<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>
     </div>
+  </div>
 
-    <div class="lct-field">
-      <label for="lct-provider">Provider</label>
-      <input id="lct-provider" type="text" name="provider" value="<%= params[:provider] %>">
-    </div>
+  <form class="lct-filters" action="<%= tag_path(@tag_key) %>" method="get">
+    <div class="lct-filter-row lct-filter-row-basic">
+      <div class="lct-field">
+        <label for="lct-tag-show-from">From</label>
+        <input id="lct-tag-show-from" data-lct-filter-input type="date" name="from" value="<%= params[:from] %>">
+      </div>
 
-    <div class="lct-field">
-      <label for="lct-model">Model</label>
-      <input id="lct-model" type="text" name="model" value="<%= params[:model] %>">
-    </div>
+      <div class="lct-field">
+        <label for="lct-tag-show-to">To</label>
+        <input id="lct-tag-show-to" type="date" name="to" value="<%= params[:to] %>">
+      </div>
+
+      <div class="lct-field">
+        <label for="lct-tag-show-provider">Provider</label>
+        <%= select_tag :provider,
+                       options_for_select(provider_filter_options, params[:provider]),
+                       include_blank: "All providers",
+                       id: "lct-tag-show-provider" %>
+      </div>
+
+      <div class="lct-field">
+        <label for="lct-tag-show-model">Model</label>
+        <%= select_tag :model,
+                       options_for_select(model_filter_options, params[:model]),
+                       include_blank: "All models",
+                       id: "lct-tag-show-model" %>
+      </div>
 
-    <div class="lct-button-row">
-      <button class="lct-button" type="submit">Apply</button>
-      <%= link_to "Reset", tag_path(@tag_key), class: "lct-button lct-button-secondary" %>
+      <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? %>
+      </div>
     </div>
   </form>
+
+  <%= render "llm_cost_tracker/shared/active_filters", chips: active_tag_filters, clear_path: tag_path(@tag_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>
+  </p>
 </section>
 
 <% if @rows.empty? %>
   <section class="lct-panel lct-empty">
-    <h2 class="lct-section-title">No calls tagged with <%= @tag_key %></h2>
-    <p class="lct-muted">Calls carrying this tag will appear here when they match the current filters.</p>
+    <h2 class="lct-state-title">No calls tagged with <%= @tag_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" %>
+    </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>
+    </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>
+    </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-copy">Unique values currently visible</p>
+    </article>
+  </section>
+
   <section class="lct-panel">
     <div class="lct-table-wrap">
-      <table class="lct-table">
+      <table class="lct-table lct-table-compact">
         <thead>
           <tr>
             <th>Value</th>
-            <th>Calls</th>
-            <th>Total cost</th>
-            <th>Avg cost / call</th>
+            <th class="lct-num">Calls</th>
+            <th class="lct-num">Share</th>
+            <th class="lct-num">Total cost</th>
+            <th class="lct-num">Avg cost / call</th>
+            <th></th>
           </tr>
         </thead>
         <tbody>
           <% @rows.each do |row| %>
             <tr>
-              <td><%= row.value %></td>
-              <td><%= number(row.calls) %></td>
-              <td><%= money(row.total_cost) %></td>
-              <td><%= money(row.average_cost_per_call) %></td>
+              <td><code class="lct-code"><%= row.value %></code></td>
+              <td class="lct-num"><%= number(row.calls) %></td>
+              <td class="lct-num"><%= percent((row.calls / share_base) * 100.0) %></td>
+              <td class="lct-num"><%= money(row.total_cost) %></td>
+              <td class="lct-num"><%= money(row.average_cost_per_call) %></td>
+              <td>
+                <% 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" %>
+                <% end %>
+              </td>
             </tr>
           <% end %>
         </tbody>

data/config/routes.rb

@@ -7,4 +7,7 @@ LlmCostTracker::Engine.routes.draw do
   get "tags",      to: "tags#index",  as: :tags
   get "tags/:key", to: "tags#show",   as: :tag, format: false
   get "data_quality", to: "data_quality#index", as: :data_quality
+
+  get "assets/#{LlmCostTracker::Assets.stylesheet_filename}",
+      to: "assets#stylesheet", as: :stylesheet
 end

data/lib/llm_cost_tracker/assets.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require "digest"
+
+module LlmCostTracker
+  module Assets
+    ROOT = File.expand_path("../../app/assets/llm_cost_tracker", __dir__)
+    STYLESHEET = "application.css"
+
+    class << self
+      def root
+        ROOT
+      end
+
+      def stylesheet_fingerprint
+        @stylesheet_fingerprint ||= Digest::SHA256.file(File.join(ROOT, STYLESHEET)).hexdigest[0, 12]
+      end
+
+      def stylesheet_filename
+        "application-#{stylesheet_fingerprint}.css"
+      end
+    end
+  end
+end

data/lib/llm_cost_tracker/engine.rb

@@ -3,6 +3,8 @@
 require "rails"
 require_relative "../llm_cost_tracker"
 require_relative "engine_compatibility"
+require_relative "assets"
+require "rack/files"
 
 LlmCostTracker::EngineCompatibility.check_rails_version!(Rails.version)
 

data/lib/llm_cost_tracker/llm_api_call.rb

@@ -100,7 +100,7 @@ module LlmCostTracker
 
       case connection.adapter_name
       when /postgres/i
-        json_column = tags_json_column? ? column : "(#{column})::jsonb"
+        json_column = tags_jsonb_column? ? column : "(#{column})::jsonb"
         "#{json_column}->>#{connection.quote(key)}"
       when /mysql/i
         "JSON_UNQUOTE(JSON_EXTRACT(#{column}, #{connection.quote(json_path(key))}))"

data/lib/llm_cost_tracker/price_registry.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "json"
+require "monitor"
 require "yaml"
 
 require_relative "logging"
@@ -11,14 +12,17 @@ module LlmCostTracker
     EMPTY_PRICES = {}.freeze
     PRICE_KEYS = %w[input cached_input output cache_read_input cache_creation_input].freeze
     METADATA_KEYS = %w[_source _updated _notes].freeze
+    MUTEX = Monitor.new
 
     class << self
       def builtin_prices
-        @builtin_prices ||= normalize_price_table(raw_registry.fetch("models", {})).freeze
+        @builtin_prices ||= MUTEX.synchronize do
+          @builtin_prices || normalize_price_table(raw_registry.fetch("models", {})).freeze
+        end
       end
 
       def metadata
-        @metadata ||= raw_registry.fetch("metadata", {}).freeze
+        @metadata ||= MUTEX.synchronize { @metadata || raw_registry.fetch("metadata", {}).freeze }
       end
 
       def normalize_price_table(table)
@@ -35,9 +39,14 @@ module LlmCostTracker
         cached = @file_prices_cache
         return cached[:value] if cached && cached[:key] == cache_key
 
-        value = normalize_file_prices(price_file_models(load_price_file(path)), path: path).freeze
-        @file_prices_cache = { key: cache_key, value: value }.freeze
-        value
+        MUTEX.synchronize do
+          cached = @file_prices_cache
+          return cached[:value] if cached && cached[:key] == cache_key
+
+          value = normalize_file_prices(price_file_models(load_price_file(path)), path: path).freeze
+          @file_prices_cache = { key: cache_key, value: value }.freeze
+          value
+        end
       rescue Errno::ENOENT, JSON::ParserError, Psych::Exception, ArgumentError, TypeError, NoMethodError => e
         raise Error, "Unable to load prices_file #{path.inspect}: #{e.message}"
       end
@@ -45,7 +54,9 @@ module LlmCostTracker
       private
 
       def raw_registry
-        @raw_registry ||= JSON.parse(File.read(DEFAULT_PRICES_PATH)).freeze
+        @raw_registry ||= MUTEX.synchronize do
+          @raw_registry || JSON.parse(File.read(DEFAULT_PRICES_PATH)).freeze
+        end
       end
 
       def normalize_price_entry(price)