codecane 1.0.206 → 1.0.235

package/dist/common/bigquery/client.js

@@ -0,0 +1,246 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setupBigQuery = setupBigQuery;
+exports.insertTrace = insertTrace;
+exports.insertRelabel = insertRelabel;
+exports.getRecentTraces = getRecentTraces;
+exports.getRecentRelabels = getRecentRelabels;
+exports.getTracesWithoutRelabels = getTracesWithoutRelabels;
+exports.getTracesWithRelabels = getTracesWithRelabels;
+exports.getTracesAndRelabelsForUser = getTracesAndRelabelsForUser;
+const bigquery_1 = require("@google-cloud/bigquery");
+const logger_1 = require("../util/logger");
+const schema_1 = require("./schema");
+const DATASET = process.env.NEXT_PUBLIC_CB_ENVIRONMENT === 'production'
+    ? 'codebuff_data'
+    : 'codebuff_data_dev';
+const TRACES_TABLE = 'traces';
+const RELABELS_TABLE = 'relabels';
+// Create a single BigQuery client instance to be used by all functions
+const client = new bigquery_1.BigQuery();
+async function setupBigQuery(dataset = DATASET) {
+    try {
+        // Ensure dataset exists
+        const [ds] = await client.dataset(dataset).get({ autoCreate: true });
+        // Ensure tables exist
+        await ds.table(TRACES_TABLE).get({
+            autoCreate: true,
+            schema: schema_1.TRACES_SCHEMA,
+            timePartitioning: {
+                type: 'MONTH',
+                field: 'created_at',
+            },
+            clustering: {
+                fields: ['user_id', 'agent_step_id'],
+            },
+        });
+        await ds.table(RELABELS_TABLE).get({
+            autoCreate: true,
+            schema: schema_1.RELABELS_SCHEMA,
+            timePartitioning: {
+                type: 'MONTH',
+                field: 'created_at',
+            },
+            clustering: {
+                fields: ['user_id', 'agent_step_id'],
+            },
+        });
+    }
+    catch (error) {
+        console.log('Failed to initialize BigQuery', JSON.stringify(error));
+        logger_1.logger.error({ error }, 'Failed to initialize BigQuery');
+    }
+}
+async function insertTrace(trace, dataset = DATASET) {
+    try {
+        // Create a copy of the trace and stringify payload if needed
+        const traceToInsert = {
+            ...trace,
+            payload: trace.payload && typeof trace.payload !== 'string'
+                ? JSON.stringify(trace.payload)
+                : trace.payload,
+        };
+        await client.dataset(dataset).table(TRACES_TABLE).insert(traceToInsert);
+        logger_1.logger.debug({ traceId: trace.id, type: trace.type }, 'Inserted trace into BigQuery');
+        return true;
+    }
+    catch (error) {
+        logger_1.logger.error({ error, traceId: trace.id }, 'Failed to insert trace into BigQuery');
+        return false;
+    }
+}
+async function insertRelabel(relabel, dataset = DATASET) {
+    try {
+        // Stringify payload if needed
+        const relabelToInsert = {
+            ...relabel,
+            payload: relabel.payload && typeof relabel.payload !== 'string'
+                ? JSON.stringify(relabel.payload)
+                : relabel.payload,
+        };
+        await client.dataset(dataset).table(RELABELS_TABLE).insert(relabelToInsert);
+        logger_1.logger.debug({ relabelId: relabel.id }, 'Inserted relabel into BigQuery');
+        return true;
+    }
+    catch (error) {
+        logger_1.logger.error({ error, relabelId: relabel.id }, 'Failed to insert relabel into BigQuery');
+        return false;
+    }
+}
+async function getRecentTraces(limit = 10, dataset = DATASET) {
+    const query = `
+    SELECT * FROM ${dataset}.${TRACES_TABLE}
+    ORDER BY created_at DESC
+    LIMIT ${limit}
+  `;
+    const [rows] = await client.query(query);
+    // Parse the payload as JSON if it's a string
+    return rows.map((row) => ({
+        ...row,
+        payload: typeof row.payload === 'string' ? JSON.parse(row.payload) : row.payload,
+    }));
+}
+async function getRecentRelabels(limit = 10, dataset = DATASET) {
+    const query = `
+    SELECT * FROM ${dataset}.${RELABELS_TABLE}
+    ORDER BY created_at DESC
+    LIMIT ${limit}
+  `;
+    const [rows] = await client.query(query);
+    // Parse the payload as JSON if it's a string
+    return rows.map((row) => ({
+        ...row,
+        payload: typeof row.payload === 'string' ? JSON.parse(row.payload) : row.payload,
+    }));
+}
+async function getTracesWithoutRelabels(model, limit = 100, userId = undefined, dataset = DATASET) {
+    // TODO: Optimize query, maybe only get traces in last 30 days etc
+    const query = `
+    SELECT t.*
+    FROM \`${dataset}.${TRACES_TABLE}\` t
+    LEFT JOIN (
+      SELECT r.agent_step_id, r.user_id, JSON_EXTRACT_SCALAR(r.payload, '$.user_input_id') as user_input_id
+      FROM \`${dataset}.${RELABELS_TABLE}\` r
+      WHERE r.model = '${model}'
+      ${userId ? `AND r.user_id = '${userId}'` : ''}
+    ) r
+    ON t.agent_step_id = r.agent_step_id
+       AND t.user_id = r.user_id
+       AND JSON_EXTRACT_SCALAR(t.payload, '$.user_input_id') = r.user_input_id
+    WHERE t.type = 'get-relevant-files'
+      AND r.agent_step_id IS NULL
+      ${userId ? `AND t.user_id = '${userId}'` : ''}
+    ORDER BY t.created_at DESC
+    LIMIT ${limit}
+  `;
+    const [rows] = await client.query(query);
+    // Parse the payload as JSON if it's a string
+    return rows.map((row) => ({
+        ...row,
+        payload: typeof row.payload === 'string' ? JSON.parse(row.payload) : row.payload,
+    }));
+}
+async function getTracesWithRelabels(model, limit = 100, dataset = DATASET) {
+    // Get traces that DO have matching relabels for the specified model
+    const query = `
+  SELECT 
+    ANY_VALUE(t) as trace,
+    ARRAY_AGG(r ORDER BY r.created_at DESC LIMIT 1)[OFFSET(0)] as relabel
+  FROM \`${dataset}.${TRACES_TABLE}\` t
+  INNER JOIN (
+    SELECT *
+    FROM \`${dataset}.${RELABELS_TABLE}\` r
+    WHERE r.model = '${model}'
+  ) r
+  ON t.agent_step_id = r.agent_step_id
+     AND t.user_id = r.user_id
+     AND JSON_EXTRACT_SCALAR(t.payload, '$.user_input_id') = JSON_EXTRACT_SCALAR(r.payload, '$.user_input_id')
+  WHERE t.type = 'get-relevant-files'
+    AND JSON_EXTRACT_SCALAR(t.payload, '$.output') IS NOT NULL
+    AND JSON_EXTRACT_SCALAR(r.payload, '$.output') IS NOT NULL
+  GROUP BY t.agent_step_id
+  ORDER BY MAX(t.created_at) DESC
+  LIMIT ${limit}
+  `;
+    const [rows] = await client.query(query);
+    // Filter out any results where either trace or relabel data is missing
+    const res = rows
+        .filter((row) => row.trace && row.relabel)
+        .map((row) => ({
+        trace: row.trace,
+        relabel: row.relabel,
+    }));
+    // Parse the payload as JSON if it's a string
+    return res.map((row) => ({
+        ...row,
+        trace: {
+            ...row.trace,
+            payload: typeof row.trace.payload === 'string'
+                ? JSON.parse(row.trace.payload)
+                : row.trace.payload,
+        },
+        relabel: {
+            ...row.relabel,
+            payload: typeof row.relabel.payload === 'string'
+                ? JSON.parse(row.relabel.payload)
+                : row.relabel.payload,
+        },
+    }));
+}
+async function getTracesAndRelabelsForUser(userId, limit = 50, dataset = DATASET) {
+    // Get recent traces for the user and any associated relabels
+    const query = `
+  WITH traces AS (
+    SELECT
+      id,
+      agent_step_id,
+      user_id,
+      created_at,
+      type,
+      payload
+    FROM \`${dataset}.${TRACES_TABLE}\`
+    WHERE user_id = '${userId}' AND type = 'get-relevant-files'
+    ORDER BY created_at DESC
+    LIMIT ${limit}
+  )
+  SELECT
+    t.id,
+    ANY_VALUE(t.agent_step_id) as agent_step_id,
+    ANY_VALUE(t.user_id) as user_id,
+    ANY_VALUE(t.created_at) as created_at,
+    ANY_VALUE(t.type) as type,
+    ANY_VALUE(t.payload) as payload,
+    ARRAY_AGG(r IGNORE NULLS) as relabels
+  FROM traces t
+  LEFT JOIN \`${dataset}.${RELABELS_TABLE}\` r
+  ON t.agent_step_id = r.agent_step_id
+     AND t.user_id = r.user_id
+     AND JSON_EXTRACT_SCALAR(t.payload, '$.user_input_id') = JSON_EXTRACT_SCALAR(r.payload, '$.user_input_id')
+  GROUP BY t.id
+  ORDER BY ANY_VALUE(t.created_at) DESC
+  `;
+    const [rows] = await client.query(query);
+    // Process and parse the results
+    return rows.map((row) => {
+        // Create trace object from individual fields
+        const trace = {
+            id: row.id,
+            agent_step_id: row.agent_step_id,
+            user_id: row.user_id,
+            created_at: row.created_at,
+            type: row.type,
+            payload: typeof row.payload === 'string' ? JSON.parse(row.payload) : row.payload,
+        };
+        // Parse relabel payloads (if any exist)
+        const relabels = row.relabels && row.relabels.length > 0
+            ? row.relabels.map((relabel) => ({
+                ...relabel,
+                payload: typeof relabel.payload === 'string'
+                    ? JSON.parse(relabel.payload)
+                    : relabel.payload,
+            }))
+            : [];
+        return { trace, relabels };
+    });
+}
+//# sourceMappingURL=client.js.map

package/dist/common/bigquery/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/bigquery/client.ts"],"names":[],"mappings":";;AAsBA,sCAgCC;AAED,kCAyBC;AAED,sCAyBC;AAED,0CAgBC;AAED,8CAgBC;AAED,4DAiCC;AAED,sDAuDC;AAED,kEAkEC;AAhTD,qDAAiD;AAEjD,2CAAuC;AACvC,qCAMiB;AAEjB,MAAM,OAAO,GACX,OAAO,CAAC,GAAG,CAAC,0BAA0B,KAAK,YAAY;IACrD,CAAC,CAAC,eAAe;IACjB,CAAC,CAAC,mBAAmB,CAAA;AAEzB,MAAM,YAAY,GAAG,QAAQ,CAAA;AAC7B,MAAM,cAAc,GAAG,UAAU,CAAA;AAEjC,uEAAuE;AACvE,MAAM,MAAM,GAAG,IAAI,mBAAQ,EAAE,CAAA;AAEtB,KAAK,UAAU,aAAa,CAAC,UAAkB,OAAO;IAC3D,IAAI,CAAC;QACH,wBAAwB;QACxB,MAAM,CAAC,EAAE,CAAC,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAA;QAEpE,sBAAsB;QACtB,MAAM,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC;YAC/B,UAAU,EAAE,IAAI;YAChB,MAAM,EAAE,sBAAa;YACrB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,YAAY;aACpB;YACD,UAAU,EAAE;gBACV,MAAM,EAAE,CAAC,SAAS,EAAE,eAAe,CAAC;aACrC;SACF,CAAC,CAAA;QACF,MAAM,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC;YACjC,UAAU,EAAE,IAAI;YAChB,MAAM,EAAE,wBAAe;YACvB,gBAAgB,EAAE;gBAChB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,YAAY;aACpB;YACD,UAAU,EAAE;gBACV,MAAM,EAAE,CAAC,SAAS,EAAE,eAAe,CAAC;aACrC;SACF,CAAC,CAAA;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAA;QACnE,eAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,+BAA+B,CAAC,CAAA;IAC1D,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,WAAW,CAAC,KAAY,EAAE,UAAkB,OAAO;IACvE,IAAI,CAAC;QACH,6DAA6D;QAC7D,MAAM,aAAa,GAAG;YACpB,GAAG,KAAK;YACR,OAAO,EACL,KAAK,CAAC,OAAO,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;gBAChD,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC;gBAC/B,CAAC,CAAC,KAAK,CAAC,OAAO;SACpB,CAAA;QAED,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAA;QAEvE,eAAM,CAAC,KAAK,CACV,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EACvC,8BAA8B,CAC/B,CAAA;QACD,OAAO,IAAI,CAAA;IACb,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CACV,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,EAC5B,sCAAsC,CACvC,CAAA;QACD,OAAO,KAAK,CAAA;IACd,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,aAAa,CACjC,OAAgB,EAChB,UAAkB,OAAO;IAEzB,IAAI,CAAC;QACH,8BAA8B;QAC9B,MAAM,eAAe,GAAG;YACtB,GAAG,OAAO;YACV,OAAO,EACL,OAAO,CAAC,OAAO,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;gBACpD,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;gBACjC,CAAC,CAAC,OAAO,CAAC,OAAO;SACtB,CAAA;QAED,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;QAE3E,eAAM,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EAAE,gCAAgC,CAAC,CAAA;QACzE,OAAO,IAAI,CAAA;IACb,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CACV,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EAChC,wCAAwC,CACzC,CAAA;QACD,OAAO,KAAK,CAAA;IACd,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,eAAe,CACnC,QAAgB,EAAE,EAClB,UAAkB,OAAO;IAEzB,MAAM,KAAK,GAAG;oBACI,OAAO,IAAI,YAAY;;YAE/B,KAAK;GACd,CAAA;IACD,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IACxC,6CAA6C;IAC7C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxB,GAAG,GAAG;QACN,OAAO,EACL,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO;KAC1E,CAAC,CAAY,CAAA;AAChB,CAAC;AAEM,KAAK,UAAU,iBAAiB,CACrC,QAAgB,EAAE,EAClB,UAAkB,OAAO;IAEzB,MAAM,KAAK,GAAG;oBACI,OAAO,IAAI,cAAc;;YAEjC,KAAK;GACd,CAAA;IACD,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IACxC,6CAA6C;IAC7C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxB,GAAG,GAAG;QACN,OAAO,EACL,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO;KAC1E,CAAC,CAAc,CAAA;AAClB,CAAC;AAEM,KAAK,UAAU,wBAAwB,CAC5C,KAAa,EACb,QAAgB,GAAG,EACnB,SAA6B,SAAS,EACtC,UAAkB,OAAO;IAEzB,kEAAkE;IAClE,MAAM,KAAK,GAAG;;aAEH,OAAO,IAAI,YAAY;;;eAGrB,OAAO,IAAI,cAAc;yBACf,KAAK;QACtB,MAAM,CAAC,CAAC,CAAC,oBAAoB,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE;;;;;;;QAO3C,MAAM,CAAC,CAAC,CAAC,oBAAoB,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE;;YAEvC,KAAK;GACd,CAAA;IAED,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IACxC,6CAA6C;IAC7C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxB,GAAG,GAAG;QACN,OAAO,EACL,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO;KAC1E,CAAC,CAA4B,CAAA;AAChC,CAAC;AAEM,KAAK,UAAU,qBAAqB,CACzC,KAAa,EACb,QAAgB,GAAG,EACnB,UAAkB,OAAO;IAEzB,oEAAoE;IACpE,MAAM,KAAK,GAAG;;;;WAIL,OAAO,IAAI,YAAY;;;aAGrB,OAAO,IAAI,cAAc;uBACf,KAAK;;;;;;;;;;UAUlB,KAAK;GACZ,CAAA;IAED,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IAExC,uEAAuE;IACvE,MAAM,GAAG,GAAG,IAAI;SACb,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,OAAO,CAAC;SACzC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACb,KAAK,EAAE,GAAG,CAAC,KAA8B;QACzC,OAAO,EAAE,GAAG,CAAC,OAAkB;KAChC,CAAC,CAAC,CAAA;IAEL,6CAA6C;IAC7C,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACvB,GAAG,GAAG;QACN,KAAK,EAAE;YACL,GAAG,GAAG,CAAC,KAAK;YACZ,OAAO,EACL,OAAO,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,QAAQ;gBACnC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC;gBAC/B,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO;SACxB;QACD,OAAO,EAAE;YACP,GAAG,GAAG,CAAC,OAAO;YACd,OAAO,EACL,OAAO,GAAG,CAAC,OAAO,CAAC,OAAO,KAAK,QAAQ;gBACrC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC;gBACjC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO;SAC1B;KACF,CAAC,CAAyD,CAAA;AAC7D,CAAC;AAEM,KAAK,UAAU,2BAA2B,CAC/C,MAAc,EACd,QAAgB,EAAE,EAClB,UAAkB,OAAO;IAEzB,6DAA6D;IAC7D,MAAM,KAAK,GAAG;;;;;;;;;aASH,OAAO,IAAI,YAAY;uBACb,MAAM;;YAEjB,KAAK;;;;;;;;;;;gBAWD,OAAO,IAAI,cAAc;;;;;;GAMtC,CAAA;IAED,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IAExC,gCAAgC;IAChC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACtB,6CAA6C;QAC7C,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,OAAO,EACL,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO;SACjD,CAAA;QAE1B,wCAAwC;QACxC,MAAM,QAAQ,GACZ,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;YACrC,CAAC,CAAE,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAY,EAAE,EAAE,CAAC,CAAC;gBACnC,GAAG,OAAO;gBACV,OAAO,EACL,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;oBACjC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;oBAC7B,CAAC,CAAC,OAAO,CAAC,OAAO;aACtB,CAAC,CAAe;YACnB,CAAC,CAAC,EAAE,CAAA;QAER,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAA;IAC5B,CAAC,CAAC,CAAA;AACJ,CAAC"}

package/dist/common/bigquery/schema.d.ts

@@ -0,0 +1,55 @@
+import { TableSchema } from '@google-cloud/bigquery';
+interface BaseEvent {
+    id: string;
+    agent_step_id: string;
+    user_id: string;
+}
+interface BasePayload {
+    user_input_id: string;
+    client_session_id: string;
+    fingerprint_id: string;
+}
+export type TraceType = 'get-relevant-files' | 'file-trees' | 'agent-response';
+export interface BaseTrace extends BaseEvent {
+    created_at: Date;
+    type: TraceType;
+    payload: unknown;
+}
+export interface GetRelevantFilesPayload extends BasePayload {
+    messages: unknown;
+    system: unknown;
+    output: string;
+    request_type: string;
+    cost_mode: string;
+    model?: string;
+}
+export interface GetRelevantFilesTrace extends BaseTrace {
+    type: 'get-relevant-files';
+    payload: GetRelevantFilesPayload;
+}
+interface FileTreePayload extends BasePayload {
+    filetrees: Record<number, string>;
+}
+export interface FileTreeTrace extends BaseTrace {
+    type: 'file-trees';
+    payload: FileTreePayload;
+}
+interface AgentResponsePayload extends BasePayload {
+    output: string;
+}
+export interface AgentResponseTrace extends BaseTrace {
+    type: 'agent-response';
+    payload: AgentResponsePayload;
+}
+export type Trace = GetRelevantFilesTrace | FileTreeTrace | AgentResponseTrace;
+export declare const TRACES_SCHEMA: TableSchema;
+interface RelabelPayload extends BasePayload {
+    output: string;
+}
+export interface Relabel extends BaseEvent {
+    created_at: Date;
+    model: string;
+    payload: RelabelPayload;
+}
+export declare const RELABELS_SCHEMA: TableSchema;
+export {};

package/dist/common/bigquery/schema.js

@@ -0,0 +1,24 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RELABELS_SCHEMA = exports.TRACES_SCHEMA = void 0;
+exports.TRACES_SCHEMA = {
+    fields: [
+        { name: 'id', type: 'STRING', mode: 'REQUIRED' }, // UUID
+        { name: 'agent_step_id', type: 'STRING', mode: 'REQUIRED' }, // Used to link traces together within a single agent step
+        { name: 'user_id', type: 'STRING', mode: 'REQUIRED' }, // user ID
+        { name: 'created_at', type: 'TIMESTAMP', mode: 'REQUIRED' },
+        { name: 'type', type: 'STRING', mode: 'REQUIRED' },
+        { name: 'payload', type: 'JSON', mode: 'REQUIRED' },
+    ],
+};
+exports.RELABELS_SCHEMA = {
+    fields: [
+        { name: 'id', type: 'STRING', mode: 'REQUIRED' }, // UUID
+        { name: 'agent_step_id', type: 'STRING', mode: 'REQUIRED' }, // Used to link traces together within a single agent step
+        { name: 'user_id', type: 'STRING', mode: 'REQUIRED' }, // user ID
+        { name: 'created_at', type: 'TIMESTAMP', mode: 'REQUIRED' },
+        { name: 'model', type: 'STRING', mode: 'REQUIRED' },
+        { name: 'payload', type: 'JSON', mode: 'REQUIRED' },
+    ],
+};
+//# sourceMappingURL=schema.js.map

package/dist/common/bigquery/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/bigquery/schema.ts"],"names":[],"mappings":";;;AA4Da,QAAA,aAAa,GAAgB;IACxC,MAAM,EAAE;QACN,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,OAAO;QACzD,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,0DAA0D;QACvH,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,UAAU;QACjE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE;QAC3D,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE;QAClD,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE;KACpD;CACF,CAAA;AAYY,QAAA,eAAe,GAAgB;IAC1C,MAAM,EAAE;QACN,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,OAAO;QACzD,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,0DAA0D;QACvH,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,UAAU;QACjE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE;QAC3D,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE;QACnD,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE;KACpD;CACF,CAAA"}

package/dist/common/billing/auto-topup.d.ts

@@ -0,0 +1,8 @@
+import type Stripe from 'stripe';
+interface AutoTopupValidationResult {
+    blockedReason: string | null;
+    validPaymentMethod: Stripe.PaymentMethod | null;
+}
+export declare function validateAutoTopupStatus(userId: string): Promise<AutoTopupValidationResult>;
+export declare function checkAndTriggerAutoTopup(userId: string): Promise<void>;
+export {};

package/dist/common/billing/auto-topup.js

@@ -0,0 +1,192 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.validateAutoTopupStatus = validateAutoTopupStatus;
+exports.checkAndTriggerAutoTopup = checkAndTriggerAutoTopup;
+const db_1 = __importDefault(require("../db"));
+const schema = __importStar(require("../db/schema"));
+const drizzle_orm_1 = require("drizzle-orm");
+const stripe_1 = require("../util/stripe");
+const logger_1 = require("../util/logger");
+const grant_credits_1 = require("./grant-credits");
+const balance_calculator_1 = require("./balance-calculator");
+const conversion_1 = require("./conversion");
+const string_1 = require("../util/string");
+const env_mjs_1 = require("src/env.mjs");
+const MINIMUM_PURCHASE_CREDITS = 500;
+class AutoTopupValidationError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'AutoTopupValidationError';
+    }
+}
+class AutoTopupPaymentError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'AutoTopupPaymentError';
+    }
+}
+async function validateAutoTopupStatus(userId) {
+    const logContext = { userId };
+    try {
+        const user = await db_1.default.query.user.findFirst({
+            where: (0, drizzle_orm_1.eq)(schema.user.id, userId),
+            columns: {
+                stripe_customer_id: true,
+            },
+        });
+        if (!user?.stripe_customer_id) {
+            throw new AutoTopupValidationError(`You don't have a valid account with us. Please log in at ${env_mjs_1.env.NEXT_PUBLIC_APP_URL}/login`);
+        }
+        const paymentMethods = await stripe_1.stripeServer.paymentMethods.list({
+            customer: user.stripe_customer_id,
+            type: 'card',
+        });
+        const validPaymentMethod = paymentMethods.data.find((pm) => pm.card?.exp_year &&
+            pm.card.exp_month &&
+            new Date(pm.card.exp_year, pm.card.exp_month - 1) > new Date());
+        if (!validPaymentMethod) {
+            throw new AutoTopupValidationError('You need a valid payment method to enable auto top-up. Try buying some credits!');
+        }
+        return {
+            blockedReason: null,
+            validPaymentMethod,
+        };
+    }
+    catch (error) {
+        const blockedReason = error instanceof AutoTopupValidationError
+            ? error.message
+            : 'Unable to verify payment method status.';
+        await disableAutoTopup(userId, blockedReason);
+        return {
+            blockedReason,
+            validPaymentMethod: null,
+        };
+    }
+}
+async function disableAutoTopup(userId, reason) {
+    await db_1.default
+        .update(schema.user)
+        .set({ auto_topup_enabled: false })
+        .where((0, drizzle_orm_1.eq)(schema.user.id, userId));
+    logger_1.logger.info({ userId, reason }, 'Disabled auto top-up');
+}
+async function processAutoTopupPayment(userId, amountToTopUp, stripeCustomerId, paymentMethod) {
+    const logContext = { userId, amountToTopUp };
+    const operationId = `auto-${userId}-${(0, string_1.generateCompactId)()}`;
+    const centsPerCredit = await (0, conversion_1.getUserCostPerCredit)(userId);
+    const amountInCents = (0, conversion_1.convertCreditsToUsdCents)(amountToTopUp, centsPerCredit);
+    if (amountInCents <= 0) {
+        throw new AutoTopupPaymentError('Invalid payment amount calculated');
+    }
+    const paymentIntent = await stripe_1.stripeServer.paymentIntents.create({
+        amount: amountInCents,
+        currency: 'usd',
+        customer: stripeCustomerId,
+        payment_method: paymentMethod.id,
+        off_session: true,
+        confirm: true,
+        description: `Auto top-up: ${amountToTopUp.toLocaleString()} credits`,
+        metadata: {
+            userId,
+            credits: amountToTopUp.toString(),
+            operationId,
+            grantType: 'purchase',
+            type: 'auto-topup',
+        },
+    });
+    if (paymentIntent.status !== 'succeeded') {
+        throw new AutoTopupPaymentError('Payment failed or requires action');
+    }
+    await (0, grant_credits_1.processAndGrantCredit)(userId, amountToTopUp, 'purchase', `Auto top-up of ${amountToTopUp.toLocaleString()} credits`, null, operationId);
+    logger_1.logger.info({
+        ...logContext,
+        operationId,
+        paymentIntentId: paymentIntent.id,
+    }, 'Auto top-up payment succeeded and credits granted');
+}
+async function checkAndTriggerAutoTopup(userId) {
+    const logContext = { userId };
+    try {
+        const user = await db_1.default.query.user.findFirst({
+            where: (0, drizzle_orm_1.eq)(schema.user.id, userId),
+            columns: {
+                id: true,
+                stripe_customer_id: true,
+                auto_topup_enabled: true,
+                auto_topup_threshold: true,
+                auto_topup_amount: true,
+                next_quota_reset: true,
+            },
+        });
+        if (!user ||
+            !user.auto_topup_enabled ||
+            user.auto_topup_threshold === null ||
+            user.auto_topup_amount === null ||
+            !user.stripe_customer_id) {
+            return;
+        }
+        const { blockedReason, validPaymentMethod } = await validateAutoTopupStatus(userId);
+        if (blockedReason || !validPaymentMethod) {
+            throw new Error(blockedReason || 'Auto top-up is not available.');
+        }
+        const { balance } = await (0, balance_calculator_1.calculateUsageAndBalance)(userId, user.next_quota_reset ?? new Date(0));
+        if (balance.totalRemaining >= user.auto_topup_threshold && balance.totalDebt === 0) {
+            return;
+        }
+        const amountToTopUp = balance.totalDebt > 0
+            ? Math.max(user.auto_topup_amount, balance.totalDebt)
+            : user.auto_topup_amount;
+        if (amountToTopUp < MINIMUM_PURCHASE_CREDITS) {
+            logger_1.logger.warn(logContext, `Auto-top-up triggered but amount ${amountToTopUp} is less than minimum ${MINIMUM_PURCHASE_CREDITS}. Skipping top-up. Check user settings.`);
+            return;
+        }
+        logger_1.logger.info({
+            ...logContext,
+            currentBalance: balance.totalRemaining,
+            currentDebt: balance.totalDebt,
+            threshold: user.auto_topup_threshold,
+            amountToTopUp,
+        }, `Auto-top-up triggered for user ${userId}. Attempting to purchase ${amountToTopUp} credits.`);
+        try {
+            await processAutoTopupPayment(userId, amountToTopUp, user.stripe_customer_id, validPaymentMethod);
+        }
+        catch (error) {
+            const message = error instanceof AutoTopupPaymentError
+                ? error.message
+                : 'Payment failed. Please check your payment method and re-enable auto top-up.';
+            await disableAutoTopup(userId, message);
+            throw new Error(message);
+        }
+    }
+    catch (error) {
+        logger_1.logger.error({ ...logContext, error }, `Error during auto-top-up check for user ${userId}`);
+        throw error;
+    }
+}
+//# sourceMappingURL=auto-topup.js.map

package/dist/common/billing/auto-topup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-topup.js","sourceRoot":"","sources":["../../src/billing/auto-topup.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiCA,0DAsDC;AAmED,4DAqFC;AA/OD,+CAAsB;AACtB,qDAAsC;AACtC,6CAAgC;AAChC,2CAA6C;AAC7C,2CAAuC;AACvC,mDAAuD;AACvD,6DAA+D;AAC/D,6CAA6E;AAC7E,2CAAkD;AAElD,yCAAiC;AAEjC,MAAM,wBAAwB,GAAG,GAAG,CAAA;AAOpC,MAAM,wBAAyB,SAAQ,KAAK;IAC1C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,0BAA0B,CAAA;IACxC,CAAC;CACF;AAED,MAAM,qBAAsB,SAAQ,KAAK;IACvC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,uBAAuB,CAAA;IACrC,CAAC;CACF;AAEM,KAAK,UAAU,uBAAuB,CAC3C,MAAc;IAEd,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,CAAA;IAE7B,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,YAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;YACzC,KAAK,EAAE,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC;YACjC,OAAO,EAAE;gBACP,kBAAkB,EAAE,IAAI;aACzB;SACF,CAAC,CAAA;QAEF,IAAI,CAAC,IAAI,EAAE,kBAAkB,EAAE,CAAC;YAC9B,MAAM,IAAI,wBAAwB,CAChC,4DAA4D,aAAG,CAAC,mBAAmB,QAAQ,CAC5F,CAAA;QACH,CAAC;QAED,MAAM,cAAc,GAAG,MAAM,qBAAY,CAAC,cAAc,CAAC,IAAI,CAAC;YAC5D,QAAQ,EAAE,IAAI,CAAC,kBAAkB;YACjC,IAAI,EAAE,MAAM;SACb,CAAC,CAAA;QAEF,MAAM,kBAAkB,GAAG,cAAc,CAAC,IAAI,CAAC,IAAI,CACjD,CAAC,EAAE,EAAE,EAAE,CACL,EAAE,CAAC,IAAI,EAAE,QAAQ;YACjB,EAAE,CAAC,IAAI,CAAC,SAAS;YACjB,IAAI,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,GAAG,IAAI,IAAI,EAAE,CACjE,CAAA;QAED,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxB,MAAM,IAAI,wBAAwB,CAChC,iFAAiF,CAClF,CAAA;QACH,CAAC;QAED,OAAO;YACL,aAAa,EAAE,IAAI;YACnB,kBAAkB;SACnB,CAAA;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,aAAa,GACjB,KAAK,YAAY,wBAAwB;YACvC,CAAC,CAAC,KAAK,CAAC,OAAO;YACf,CAAC,CAAC,yCAAyC,CAAA;QAE/C,MAAM,gBAAgB,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;QAE7C,OAAO;YACL,aAAa;YACb,kBAAkB,EAAE,IAAI;SACzB,CAAA;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,MAAc,EAAE,MAAc;IAC5D,MAAM,YAAE;SACL,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;SACnB,GAAG,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC;SAClC,KAAK,CAAC,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAA;IAEpC,eAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAA;AACzD,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,MAAc,EACd,aAAqB,EACrB,gBAAwB,EACxB,aAAmC;IAEnC,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,aAAa,EAAE,CAAA;IAC5C,MAAM,WAAW,GAAG,QAAQ,MAAM,IAAI,IAAA,0BAAiB,GAAE,EAAE,CAAA;IAE3D,MAAM,cAAc,GAAG,MAAM,IAAA,iCAAoB,EAAC,MAAM,CAAC,CAAA;IACzD,MAAM,aAAa,GAAG,IAAA,qCAAwB,EAAC,aAAa,EAAE,cAAc,CAAC,CAAA;IAE7E,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,qBAAqB,CAAC,mCAAmC,CAAC,CAAA;IACtE,CAAC;IAED,MAAM,aAAa,GAAG,MAAM,qBAAY,CAAC,cAAc,CAAC,MAAM,CAAC;QAC7D,MAAM,EAAE,aAAa;QACrB,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,gBAAgB;QAC1B,cAAc,EAAE,aAAa,CAAC,EAAE;QAChC,WAAW,EAAE,IAAI;QACjB,OAAO,EAAE,IAAI;QACb,WAAW,EAAE,gBAAgB,aAAa,CAAC,cAAc,EAAE,UAAU;QACrE,QAAQ,EAAE;YACR,MAAM;YACN,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE;YACjC,WAAW;YACX,SAAS,EAAE,UAAU;YACrB,IAAI,EAAE,YAAY;SACnB;KACF,CAAC,CAAA;IAEF,IAAI,aAAa,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;QACzC,MAAM,IAAI,qBAAqB,CAAC,mCAAmC,CAAC,CAAA;IACtE,CAAC;IAED,MAAM,IAAA,qCAAqB,EACzB,MAAM,EACN,aAAa,EACb,UAAU,EACV,kBAAkB,aAAa,CAAC,cAAc,EAAE,UAAU,EAC1D,IAAI,EACJ,WAAW,CACZ,CAAA;IAED,eAAM,CAAC,IAAI,CACT;QACE,GAAG,UAAU;QACb,WAAW;QACX,eAAe,EAAE,aAAa,CAAC,EAAE;KAClC,EACD,mDAAmD,CACpD,CAAA;AACH,CAAC;AAEM,KAAK,UAAU,wBAAwB,CAAC,MAAc;IAC3D,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,CAAA;IAE7B,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,YAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;YACzC,KAAK,EAAE,IAAA,gBAAE,EAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC;YACjC,OAAO,EAAE;gBACP,EAAE,EAAE,IAAI;gBACR,kBAAkB,EAAE,IAAI;gBACxB,kBAAkB,EAAE,IAAI;gBACxB,oBAAoB,EAAE,IAAI;gBAC1B,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,IAAI;aACvB;SACF,CAAC,CAAA;QAEF,IACE,CAAC,IAAI;YACL,CAAC,IAAI,CAAC,kBAAkB;YACxB,IAAI,CAAC,oBAAoB,KAAK,IAAI;YAClC,IAAI,CAAC,iBAAiB,KAAK,IAAI;YAC/B,CAAC,IAAI,CAAC,kBAAkB,EACxB,CAAC;YACD,OAAM;QACR,CAAC;QAED,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,GACzC,MAAM,uBAAuB,CAAC,MAAM,CAAC,CAAA;QAEvC,IAAI,aAAa,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,+BAA+B,CAAC,CAAA;QACnE,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,IAAA,6CAAwB,EAAC,MAAM,EAAE,IAAI,CAAC,gBAAgB,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;QAEhG,IAAI,OAAO,CAAC,cAAc,IAAI,IAAI,CAAC,oBAAoB,IAAI,OAAO,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;YACnF,OAAM;QACR,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,SAAS,GAAG,CAAC;YACzC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,CAAC,SAAS,CAAC;YACrD,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAA;QAE1B,IAAI,aAAa,GAAG,wBAAwB,EAAE,CAAC;YAC7C,eAAM,CAAC,IAAI,CACT,UAAU,EACV,oCAAoC,aAAa,yBAAyB,wBAAwB,yCAAyC,CAC5I,CAAA;YACD,OAAM;QACR,CAAC;QAED,eAAM,CAAC,IAAI,CACT;YACE,GAAG,UAAU;YACb,cAAc,EAAE,OAAO,CAAC,cAAc;YACtC,WAAW,EAAE,OAAO,CAAC,SAAS;YAC9B,SAAS,EAAE,IAAI,CAAC,oBAAoB;YACpC,aAAa;SACd,EACD,kCAAkC,MAAM,4BAA4B,aAAa,WAAW,CAC7F,CAAA;QAED,IAAI,CAAC;YACH,MAAM,uBAAuB,CAC3B,MAAM,EACN,aAAa,EACb,IAAI,CAAC,kBAAkB,EACvB,kBAAkB,CACnB,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GACX,KAAK,YAAY,qBAAqB;gBACpC,CAAC,CAAC,KAAK,CAAC,OAAO;gBACf,CAAC,CAAC,6EAA6E,CAAA;YAEnF,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;YACvC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAA;QAC1B,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CACV,EAAE,GAAG,UAAU,EAAE,KAAK,EAAE,EACxB,2CAA2C,MAAM,EAAE,CACpD,CAAA;QACD,MAAM,KAAK,CAAA;IACb,CAAC;AACH,CAAC"}

package/dist/common/billing/balance-calculator.d.ts

@@ -0,0 +1,57 @@
+import db from '../db';
+import { GrantType } from '../db/schema';
+export interface CreditBalance {
+    totalRemaining: number;
+    totalDebt: number;
+    netBalance: number;
+    breakdown: Partial<Record<GrantType, number>>;
+    principals: Partial<Record<GrantType, number>>;
+}
+export interface CreditUsageAndBalance {
+    usageThisCycle: number;
+    balance: CreditBalance;
+}
+export interface CreditConsumptionResult {
+    consumed: number;
+    fromPurchased: number;
+}
+type DbConn = Pick<typeof db, 'select' | 'update'>;
+/**
+ * Gets active grants for a user, ordered by expiration (soonest first), then priority, and creation date.
+ * Added optional `conn` param so callers inside a transaction can supply their TX object.
+ */
+export declare function getOrderedActiveGrants(userId: string, now: Date, conn?: DbConn): Promise<{
+    created_at: Date;
+    type: import("../types/grant").GrantType;
+    expires_at: Date | null;
+    operation_id: string;
+    user_id: string;
+    principal: number;
+    balance: number;
+    description: string | null;
+    priority: number;
+}[]>;
+/**
+ * Calculates both the current balance and usage in this cycle in a single query.
+ * This is more efficient than calculating them separately.
+ */
+export declare function calculateUsageAndBalance(userId: string, quotaResetDate: Date, now?: Date): Promise<CreditUsageAndBalance>;
+/**
+ * Updates the remaining amounts in credit grants after consumption.
+ * Follows priority order strictly - higher priority grants (lower number) are consumed first.
+ * Returns details about credit consumption including how many came from purchased credits.
+ *
+ * Uses SERIALIZABLE isolation to prevent concurrent modifications that could lead to
+ * incorrect credit usage (e.g., "double spending" credits).
+ *
+ * @param userId The ID of the user
+ * @param creditsToConsume Number of credits being consumed
+ * @returns Promise resolving to number of credits consumed
+ */
+export declare function consumeCredits(userId: string, creditsToConsume: number): Promise<CreditConsumptionResult>;
+/**
+ * Calculate the total credits used during the current billing cycle for a user
+ * by summing the difference between initial and remaining amounts for all relevant grants.
+ */
+export declare function calculateUsageThisCycle(userId: string, quotaResetDate: Date): Promise<number>;
+export {};