@drawbridge/drawbridge-utils 0.0.35 → 0.0.36

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";
@@ -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

@@ -7,6 +7,7 @@ import {
 
 // 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/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.36"
 }