@drawbridge/drawbridge-utils 0.0.35 → 0.0.37

package/dist/ai.cjs

@@ -30,6 +30,7 @@ __export(ai_exports, {
 });
 module.exports = __toCommonJS(ai_exports);
 var import_genai = require("@google/genai");
+var import_drawbridge_telemetry = require("@drawbridge/drawbridge-telemetry");
 
 // circuit.js
 var CLOSED = "CLOSED";
@@ -104,7 +105,7 @@ var insertTransaction = async ({
     collection: "transaction",
     data: {
       ..._id && { _id },
-      userId: user == null ? void 0 : user.id,
+      user: user == null ? void 0 : user.id,
       type,
       category,
       source,
@@ -239,6 +240,7 @@ var billRequest = async ({
 }) => {
   const gross = priceForRequest(model, usage, tools);
   const cogs = costForRequest(model, usage, tools);
+  const id = (0, import_drawbridge_telemetry.currentTraceId)();
   await debit({
     db,
     user,
@@ -246,6 +248,7 @@ var billRequest = async ({
     type: "ai",
     category: "usage",
     source: "user",
+    ...id && { trace: { id } },
     ai: {
       name: "google",
       model,

package/dist/ai.d.cts

@@ -1,4 +1,5 @@
 import { GoogleGenAI } from '@google/genai';
+import { currentTraceId } from '@drawbridge/drawbridge-telemetry';
 import { circuit } from './circuit.cjs';
 import { debit } from './transactions.cjs';
 
@@ -182,6 +183,16 @@ const billRequest = async ({
 	const gross = priceForRequest( model, usage, tools );
 	const cogs = costForRequest( model, usage, tools );
 
+	// Stamp the current trace ID so multiple AI calls inside one HTTP request
+	// (or one BullMQ job) share a grouping key — the UI can collapse rows
+	// per trace.id to show "this preview cost $0.12 across 5 calls" without
+	// the user needing to mentally fold them together. currentTraceId()
+	// returns undefined when called outside any traced scope, in which case
+	// the field is omitted from the row (additionalProperties:false would
+	// reject `trace: undefined`). Nested as `trace.id` to match the `ai` /
+	// `balance` sub-object pattern already on this document.
+	const id = currentTraceId();
+
 	await debit({
 		db,
 		user,
@@ -189,6 +200,7 @@ const billRequest = async ({
 		type : 'ai',
 		category : 'usage',
 		source : 'user',
+		...( id && { trace : { id } }),
 		ai : {
 			name : 'google',
 			model,

package/dist/ai.d.ts

@@ -1,4 +1,5 @@
 import { GoogleGenAI } from '@google/genai';
+import { currentTraceId } from '@drawbridge/drawbridge-telemetry';
 import { circuit } from './circuit.js';
 import { debit } from './transactions.js';
 
@@ -182,6 +183,16 @@ const billRequest = async ({
 	const gross = priceForRequest( model, usage, tools );
 	const cogs = costForRequest( model, usage, tools );
 
+	// Stamp the current trace ID so multiple AI calls inside one HTTP request
+	// (or one BullMQ job) share a grouping key — the UI can collapse rows
+	// per trace.id to show "this preview cost $0.12 across 5 calls" without
+	// the user needing to mentally fold them together. currentTraceId()
+	// returns undefined when called outside any traced scope, in which case
+	// the field is omitted from the row (additionalProperties:false would
+	// reject `trace: undefined`). Nested as `trace.id` to match the `ai` /
+	// `balance` sub-object pattern already on this document.
+	const id = currentTraceId();
+
 	await debit({
 		db,
 		user,
@@ -189,6 +200,7 @@ const billRequest = async ({
 		type : 'ai',
 		category : 'usage',
 		source : 'user',
+		...( id && { trace : { id } }),
 		ai : {
 			name : 'google',
 			model,

package/dist/ai.js

@@ -1,12 +1,13 @@
 import {
   debit
-} from "./chunk-RW2FJVIV.js";
+} from "./chunk-DLRNUBZ4.js";
 import {
   circuit
 } from "./chunk-6HH36OK6.js";
 
 // ai.js
 import { GoogleGenAI } from "@google/genai";
+import { currentTraceId } from "@drawbridge/drawbridge-telemetry";
 var google_client = new GoogleGenAI({
   apiKey: process.env.GOOGLE_API_KEY
 });
@@ -93,6 +94,7 @@ var billRequest = async ({
 }) => {
   const gross = priceForRequest(model, usage, tools);
   const cogs = costForRequest(model, usage, tools);
+  const id = currentTraceId();
   await debit({
     db,
     user,
@@ -100,6 +102,7 @@ var billRequest = async ({
     type: "ai",
     category: "usage",
     source: "user",
+    ...id && { trace: { id } },
     ai: {
       name: "google",
       model,

package/dist/{chunk-RW2FJVIV.js → chunk-DLRNUBZ4.js}

@@ -20,7 +20,7 @@ var insertTransaction = async ({
     collection: "transaction",
     data: {
       ..._id && { _id },
-      userId: user == null ? void 0 : user.id,
+      user: user == null ? void 0 : user.id,
       type,
       category,
       source,

package/dist/transactions.cjs

@@ -44,7 +44,7 @@ var insertTransaction = async ({
     collection: "transaction",
     data: {
       ..._id && { _id },
-      userId: user == null ? void 0 : user.id,
+      user: user == null ? void 0 : user.id,
       type,
       category,
       source,

package/dist/transactions.d.cts

@@ -50,7 +50,7 @@ const insertTransaction = async ({
 		collection : 'transaction',
 		data : {
 			...( _id && { _id }),
-			userId : user?.id,
+			user : user?.id,
 			type,
 			category,
 			source,

package/dist/transactions.d.ts

@@ -50,7 +50,7 @@ const insertTransaction = async ({
 		collection : 'transaction',
 		data : {
 			...( _id && { _id }),
-			userId : user?.id,
+			user : user?.id,
 			type,
 			category,
 			source,

package/dist/transactions.js

@@ -1,7 +1,7 @@
 import {
   credit,
   debit
-} from "./chunk-RW2FJVIV.js";
+} from "./chunk-DLRNUBZ4.js";
 export {
   credit,
   debit

package/package.json

@@ -2,6 +2,7 @@
   "type": "module",
   "dependencies": {
     "@drawbridge/drawbridge-agents": "0.0.10",
+    "@drawbridge/drawbridge-telemetry": "0.0.13",
     "@google/genai": "1.30.0",
     "axios": "1.16.0",
     "currency-codes": "2.2.0",
@@ -103,5 +104,5 @@
     "build": "tsup && npm publish"
   },
   "types": "dist/index.d.ts",
-  "version": "0.0.35"
+  "version": "0.0.37"
 }