feat(cipx): add query_source to cipx_spends for the AI-spend model-switch lever

ai-cost-backend's "switch default model" savings estimate needs to scope to
the main conversation loop only (subagent/auxiliary calls aren't affected by
the default-model setting), but query_source was dropped when cipx_spends
moved from spans.metadata JSON to typed columns. Re-adds it as its own typed
column, extracted the same way as the other cipx.call fields.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>

Author: petrotiurin

apps/opik-backend/src/main/java/com/comet/opik/domain/CipxSpendDAO.java

@@ -53,7 +53,8 @@ public record SpanRow(
             long uInput,
             long uCacheRead,
             long uCacheCreation,
-            long uOutput) {
+            long uOutput,
+            @NonNull String querySource) {
 
         public static SpanRow from(UUID spanId, UUID traceId, UUID projectId, JsonNode metadata, Instant startTime) {
             JsonNode call = metadata.path("cipx").path("call");
@@ -68,6 +69,7 @@ public static SpanRow from(UUID spanId, UUID traceId, UUID projectId, JsonNode m
                     .uCacheRead(usage.path("cache_read_input_tokens").asLong(0))
                     .uCacheCreation(usage.path("cache_creation_input_tokens").asLong(0))
                     .uOutput(usage.path("output_tokens").asLong(0))
+                    .querySource(call.path("query_source").asText(""))
                     .build();
         }
     }
@@ -77,7 +79,7 @@ public static SpanRow from(UUID spanId, UUID traceId, UUID projectId, JsonNode m
     private static final String INSERT = """
             INSERT INTO cipx_spends
                 (workspace_id, project_id, trace_id, span_id, start_time, model,
-                 u_input, u_cache_read, u_cache_creation, u_output)
+                 u_input, u_cache_read, u_cache_creation, u_output, query_source)
             SETTINGS log_comment = '<log_comment>'
             FORMAT Values
                 <items:{item |
@@ -91,7 +93,8 @@ public static SpanRow from(UUID spanId, UUID traceId, UUID projectId, JsonNode m
                         :u_input<item.index>,
                         :u_cache_read<item.index>,
                         :u_cache_creation<item.index>,
-                        :u_output<item.index>
+                        :u_output<item.index>,
+                        :query_source<item.index>
                     )
                     <if(item.hasNext)>,<endif>
                 }>
@@ -120,7 +123,7 @@ private Publisher<? extends Result> insert(List<SpanRow> rows, String workspaceI
         // Positional binds: the driver resolves named binds with a linear indexOf over the statement's
         // parameter list (quadratic per statement), while bind(int) is a direct array write. Indices
         // follow the placeholders' first-appearance order in the rendered SQL: workspace_id once at 0
-        // (repeats dedup), then 9 parameters per row tuple in template order.
+        // (repeats dedup), then 10 parameters per row tuple in template order.
         statement.bind(0, workspaceId);
         int index = 1;
         for (SpanRow row : rows) {
@@ -132,7 +135,8 @@ private Publisher<? extends Result> insert(List<SpanRow> rows, String workspaceI
                     .bind(index++, row.uInput())
                     .bind(index++, row.uCacheRead())
                     .bind(index++, row.uCacheCreation())
-                    .bind(index++, row.uOutput());
+                    .bind(index++, row.uOutput())
+                    .bind(index++, row.querySource());
         }
 
         return statement.execute();

apps/opik-backend/src/main/resources/liquibase/db-app-analytics/migrations/000101_add_query_source_to_cipx_spend.sql

@@ -0,0 +1,12 @@
+--liquibase formatted sql
+--changeset petrotiurin:000101_add_query_source_to_cipx_spend
+--comment: Add query_source (cipx.call.query_source) to cipx_spends so AI-spend levers can scope to the main conversation loop
+
+-- cipx emits metadata.cipx.call.query_source ('main' for the primary conversation loop, something
+-- else for subagent/auxiliary calls). The "switch default model" savings lever needs to scope to
+-- 'main' only -- subagent calls aren't affected by changing the default model setting -- so this
+-- needs its own typed column rather than being folded into an existing one.
+ALTER TABLE ${ANALYTICS_DB_DATABASE_NAME}.cipx_spends ON CLUSTER '{cluster}'
+    ADD COLUMN IF NOT EXISTS query_source LowCardinality(String) DEFAULT '';
+
+--rollback ALTER TABLE ${ANALYTICS_DB_DATABASE_NAME}.cipx_spends ON CLUSTER '{cluster}' DROP COLUMN IF EXISTS query_source;