@drawbridge/drawbridge-utils 0.0.29 → 0.0.32

package/dist/ai.cjs

@@ -19,6 +19,9 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // ai.js
 var ai_exports = {};
 __export(ai_exports, {
+  MARKUP: () => MARKUP,
+  cost: () => cost,
+  costForRequest: () => costForRequest,
   google: () => google,
   priceForRequest: () => priceForRequest,
   pricing: () => pricing
@@ -78,105 +81,73 @@ var circuit = ({
 };
 
 // transactions.js
-var recordTransaction = async ({
+var insertTransaction = async ({
   db,
-  authenticated,
-  userId,
+  user,
+  type,
   category,
-  tags,
+  source,
   amount,
-  balance,
-  provider,
+  _id,
   stripeInvoiceId,
   stripeEventId,
-  session
+  session,
+  ...rest
 }) => {
+  var _a;
+  const beforeRaw = (_a = user == null ? void 0 : user.balance) == null ? void 0 : _a[type];
+  const balance = typeof beforeRaw === "number" ? { before: beforeRaw, after: beforeRaw + amount } : void 0;
   await db.create({
-    authenticated,
+    authenticated: user,
     collection: "transaction",
     data: {
-      userId,
-      billable: {
-        item: "ai"
-      },
+      ..._id && { _id },
+      userId: user == null ? void 0 : user.id,
+      type,
       category,
-      tags,
+      source,
       amount,
-      balance,
-      ...provider && { provider },
+      ...balance && { balance },
+      ...rest,
       ...stripeInvoiceId && { stripeInvoiceId },
       ...stripeEventId && { stripeEventId }
     },
     ...session && { options: { session } }
   });
 };
-var InsufficientCreditsError = class extends Error {
-  constructor(message = "Insufficient AI credits") {
-    super(message);
-    this.name = "InsufficientCreditsError";
-    this.status = 402;
-  }
-};
 var debit = async ({
   db,
-  authenticated,
-  userId,
+  user,
   amount,
+  type,
   category,
-  tags = [],
-  provider,
+  source,
   stripeInvoiceId,
   stripeEventId,
-  session
+  session,
+  ...rest
 }) => {
-  var _a, _b, _c;
   if (!Number.isInteger(amount) || amount <= 0) {
     throw new Error(`debit() requires a positive integer amount, got ${amount}`);
   }
-  const result = await db.update({
-    authenticated,
-    collection: "user",
-    query: {
-      id: userId,
-      "balance.ai": {
-        $gte: amount
-      }
-    },
-    data: {
-      $inc: {
-        "balance.ai": -amount
-      }
-    },
-    ...session && { options: { session } }
-  });
-  const after = ((_b = (_a = result == null ? void 0 : result.value) == null ? void 0 : _a.balance) == null ? void 0 : _b.ai) ?? ((_c = result == null ? void 0 : result.balance) == null ? void 0 : _c.ai);
-  if (typeof after !== "number") {
-    throw new InsufficientCreditsError();
+  if (!type) {
+    throw new Error('debit() requires a type (e.g., "ai")');
+  }
+  if (!source) {
+    throw new Error('debit() requires a source ("system" | "admin" | "user")');
   }
-  const before = after + amount;
-  await recordTransaction({
+  await insertTransaction({
     db,
-    authenticated,
-    userId,
+    user,
+    type,
     category,
-    tags,
+    source,
     amount: -amount,
-    balance: {
-      before,
-      after
-    },
-    provider,
     stripeInvoiceId,
     stripeEventId,
-    session
+    session,
+    ...rest
   });
-  return {
-    amount: -amount,
-    balance: {
-      before,
-      after
-    }
-  };
 };
 
 // ai.js
@@ -192,7 +163,7 @@ var models = {
   image: "gemini-2.5-flash-image",
   text: "gemini-2.5-flash"
 };
-var pricing = {
+var cost = {
   "gemini-2.5-flash": {
     cached: 3,
     input: 30,
@@ -204,18 +175,33 @@ var pricing = {
     output: 250
   }
 };
+var MARKUP = 1.3;
+var pricing = Object.fromEntries(
+  Object.entries(cost).map(([model, rates]) => [
+    model,
+    {
+      cached: Math.round(rates.cached * MARKUP),
+      input: Math.round(rates.input * MARKUP),
+      output: Math.round(rates.output * MARKUP)
+    }
+  ])
+);
+var sumCents = (rates, tokens) => Math.ceil(
+  (Number((tokens == null ? void 0 : tokens.cached) || 0) * rates.cached + Number((tokens == null ? void 0 : tokens.input) || 0) * rates.input + Number((tokens == null ? void 0 : tokens.output) || 0) * rates.output) / 1e6
+);
 var priceForRequest = (model, usage) => {
   const rates = pricing[model];
   if (!rates) {
     throw new Error(`Unknown model for pricing: ${model}`);
   }
-  const tokens = {
-    cached: Number(usage == null ? void 0 : usage.cached) || 0,
-    input: Number(usage == null ? void 0 : usage.input) || 0,
-    output: Number(usage == null ? void 0 : usage.output) || 0
-  };
-  const sum = tokens.cached * rates.cached + tokens.input * rates.input + tokens.output * rates.output;
-  return Math.ceil(sum / 1e6);
+  return sumCents(rates, usage);
+};
+var costForRequest = (model, usage) => {
+  const rates = cost[model];
+  if (!rates) {
+    throw new Error(`Unknown model for cost: ${model}`);
+  }
+  return sumCents(rates, usage);
 };
 var tokensFromMetadata = (metadata) => ({
   cached: Number(metadata == null ? void 0 : metadata.cachedContentTokenCount) || 0,
@@ -224,24 +210,32 @@ var tokensFromMetadata = (metadata) => ({
 });
 var billRequest = async ({
   db,
-  authenticated,
-  userId,
+  user,
   model,
   usage
 }) => {
-  const amount = priceForRequest(model, usage);
+  const gross = priceForRequest(model, usage);
+  const cogs = costForRequest(model, usage);
   await debit({
     db,
-    authenticated,
-    userId,
-    amount,
+    user,
+    amount: gross,
+    type: "ai",
     category: "usage",
-    tags: ["google", model],
-    provider: {
+    source: "user",
+    ai: {
       name: "google",
       model,
       tokens: usage,
-      snapshot: pricing[model]
+      rates: {
+        wholesale: cost[model],
+        retail: pricing[model]
+      },
+      totals: {
+        cost: cogs,
+        net: gross - cogs,
+        gross
+      }
     }
   });
 };
@@ -252,8 +246,7 @@ var google = {
     reference = null
   }, {
     db,
-    authenticated,
-    userId
+    user
   }) => {
     var _a, _b, _c, _d, _e, _f;
     const model = models.image;
@@ -261,72 +254,64 @@ var google = {
       ...reference ? [{ inlineData: { data: reference.data, mimeType: reference.mimeType } }] : [],
       { text: prompt.join("\n") }
     ];
-    try {
-      const response = await googleCircuit(() => google_client.models.generateContent({
-        model,
-        contents: { parts },
-        config: {
-          ...config,
-          responseModalities: ["Image"]
-        }
-      }));
-      const content = (_f = (_e = (_d = (_c = (_b = (_a = response == null ? void 0 : response.candidates) == null ? void 0 : _a[0]) == null ? void 0 : _b.content) == null ? void 0 : _c.parts) == null ? void 0 : _d[0]) == null ? void 0 : _e.inlineData) == null ? void 0 : _f.data;
-      if (!content) {
-        throw new Error("Could not generate from prompt. Please try again with a different prompt.");
-      }
-      if (userId) {
-        await billRequest({
-          db,
-          authenticated,
-          userId,
-          model,
-          usage: tokensFromMetadata(response == null ? void 0 : response.usageMetadata)
-        });
+    const response = await googleCircuit(() => google_client.models.generateContent({
+      model,
+      contents: { parts },
+      config: {
+        ...config,
+        responseModalities: ["Image"]
       }
-      return { content };
-    } catch (error) {
-      throw new Error((error == null ? void 0 : error.message) || "AI error, please try again");
+    }));
+    const content = (_f = (_e = (_d = (_c = (_b = (_a = response == null ? void 0 : response.candidates) == null ? void 0 : _a[0]) == null ? void 0 : _b.content) == null ? void 0 : _c.parts) == null ? void 0 : _d[0]) == null ? void 0 : _e.inlineData) == null ? void 0 : _f.data;
+    if (!content) {
+      throw new Error("Could not generate from prompt. Please try again with a different prompt.");
     }
+    if (user == null ? void 0 : user.id) {
+      await billRequest({
+        db,
+        user,
+        model,
+        usage: tokensFromMetadata(response == null ? void 0 : response.usageMetadata)
+      });
+    }
+    return { content };
   },
   text: async ({
     config = {},
     prompt = []
   }, {
     db,
-    authenticated,
-    userId
+    user
   }) => {
     const model = models.text;
-    try {
-      const response = await googleCircuit(() => google_client.models.generateContent({
-        config: {
-          ...config,
-          responseModalities: ["Text"]
-        },
-        contents: prompt.join("\n"),
-        model
-      }));
-      const content = response == null ? void 0 : response.text;
-      if (!content) {
-        throw new Error("Could not generate from prompt. Please try again with a different prompt.");
-      }
-      if (userId) {
-        await billRequest({
-          db,
-          authenticated,
-          userId,
-          model,
-          usage: tokensFromMetadata(response == null ? void 0 : response.usageMetadata)
-        });
-      }
-      return { content };
-    } catch (error) {
-      throw new Error((error == null ? void 0 : error.message) || "AI error, please try again");
+    const response = await googleCircuit(() => google_client.models.generateContent({
+      config: {
+        ...config,
+        responseModalities: ["Text"]
+      },
+      contents: prompt.join("\n"),
+      model
+    }));
+    const content = response == null ? void 0 : response.text;
+    if (!content) {
+      throw new Error("Could not generate from prompt. Please try again with a different prompt.");
+    }
+    if (user == null ? void 0 : user.id) {
+      await billRequest({
+        db,
+        user,
+        model,
+        usage: tokensFromMetadata(response == null ? void 0 : response.usageMetadata)
+      });
     }
+    return { content };
   }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  MARKUP,
+  cost,
+  costForRequest,
   google,
   priceForRequest,
   pricing

package/dist/ai.d.cts

@@ -36,14 +36,12 @@ const models = {
 	text : 'gemini-2.5-flash'
 };
 
-// Per-model pricing in cents per 1,000,000 tokens. Source of truth for what
-// users get charged; reviewed in PRs.
-//
-// Values mirror LiteLLM's published pricing at the time of writing. Model
-// providers update pricing periodically — verify against
+// Raw per-model provider rates in cents per 1,000,000 tokens. This is what
+// Drawbridge pays the provider — the wholesale cost. Sync from LiteLLM:
 // https://github.com/BerriAI/litellm/blob/main/litellm/model_prices_and_context_window_backup.json
-// before any change.
-const pricing = {
+// Never apply markup here; the MARKUP constant below derives the retail
+// `pricing` table.
+const cost = {
 	'gemini-2.5-flash' : {
 		cached : 3,
 		input : 30,
@@ -56,9 +54,37 @@ const pricing = {
 	}
 };
 
-// Cost in cents for a single request, rounded up so we never charge fractional
-// cents and never under-bill. Throws on an unknown model so a typo doesn't
-// silently fall back to a wrong price.
+// Markup multiplier applied to every model's cost to produce the retail
+// price. 1.3 = 30% margin. Change one number, all pricing updates. Mirrors
+// the per-customer markup Stripe rate cards used to provide; keeping it
+// global (rather than per-model or per-plan) for simplicity — easy to
+// refactor later if tier-based pricing becomes a real need.
+const MARKUP = 1.3;
+
+// Derived retail pricing in cents per 1,000,000 tokens — what users pay.
+// Rounded so the per-1M rate is a whole-cent value; per-request rounding
+// happens in priceForRequest via Math.ceil so we never under-bill.
+const pricing = Object.fromEntries(
+	Object.entries( cost ).map( ( [ model, rates ] ) => [
+		model,
+		{
+			cached : Math.round( rates.cached * MARKUP ),
+			input : Math.round( rates.input * MARKUP ),
+			output : Math.round( rates.output * MARKUP )
+		}
+	] )
+);
+
+const sumCents = ( rates, tokens ) => Math.ceil(
+	(
+		Number( tokens?.cached || 0 ) * rates.cached +
+		Number( tokens?.input || 0 ) * rates.input +
+		Number( tokens?.output || 0 ) * rates.output
+	) / 1_000_000
+);
+
+// What we charge the user, rounded up so we never under-bill. Throws on an
+// unknown model so a typo doesn't silently fall back to a wrong price.
 const priceForRequest = ( model, usage ) => {
 
 	const rates = pricing[ model ];
@@ -69,19 +95,23 @@ const priceForRequest = ( model, usage ) => {
 
 	}
 
-	const tokens = {
-		cached : Number( usage?.cached ) || 0,
-		input : Number( usage?.input ) || 0,
-		output : Number( usage?.output ) || 0
-	};
+	return sumCents( rates, usage );
+
+};
 
-	const sum = (
-		tokens.cached * rates.cached +
-		tokens.input * rates.input +
-		tokens.output * rates.output
-	);
+// What we pay the provider. Recorded on each transaction row so margin is
+// computable per request without backfilling from git history.
+const costForRequest = ( model, usage ) => {
 
-	return Math.ceil( sum / 1_000_000 );
+	const rates = cost[ model ];
+
+	if( !rates ){
+
+		throw new Error( `Unknown model for cost: ${ model }` );
+
+	}
+
+	return sumCents( rates, usage );
 
 };
 
@@ -92,30 +122,41 @@ const tokensFromMetadata = ( metadata ) => ({
 	output : Number( metadata?.candidatesTokenCount ) || 0
 });
 
-// Compute cost from response usage and atomically debit the user. Shared by
-// both image/text handlers so the accounting path is identical for both.
+// Compute gross + cost from response usage and atomically debit the user.
+// Records the full income-statement line on the transaction row's `ai`
+// sub-doc: wholesale + retail per-1M rate snapshots, plus rolled-up cost,
+// net, and gross totals. Margin per request is `ai.net`; margin per query
+// is a single $sum: '$ai.net'.
 const billRequest = async ({
 	db,
-	authenticated,
-	userId,
+	user,
 	model,
 	usage
 }) => {
 
-	const amount = priceForRequest( model, usage );
+	const gross = priceForRequest( model, usage );
+	const cogs = costForRequest( model, usage );
 
 	await debit({
 		db,
-		authenticated,
-		userId,
-		amount,
+		user,
+		amount : gross,
+		type : 'ai',
 		category : 'usage',
-		tags : [ 'google', model ],
-		provider : {
+		source : 'user',
+		ai : {
 			name : 'google',
 			model,
 			tokens : usage,
-			snapshot : pricing[ model ]
+			rates : {
+				wholesale : cost[ model ],
+				retail : pricing[ model ]
+			},
+			totals : {
+				cost : cogs,
+				net : gross - cogs,
+				gross
+			}
 		}
 	});
 
@@ -131,8 +172,7 @@ const google = {
 		},
 		{
 			db,
-			authenticated,
-			userId
+			user
 		}
 	) => {
 
@@ -143,45 +183,36 @@ const google = {
 			{ text : prompt.join( '\n' ) }
 		];
 
-		try {
-
-			const response = await googleCircuit( () => google_client.models.generateContent({
-				model,
-				contents : { parts },
-				config : {
-					...config,
-					responseModalities : [ 'Image' ]
-				}
-			}) );
-
-			const content = response?.candidates?.[ 0 ]?.content?.parts?.[ 0 ]?.inlineData?.data;
-
-			if( !content ){
-
-				throw new Error( 'Could not generate from prompt. Please try again with a different prompt.' );
-
+		const response = await googleCircuit( () => google_client.models.generateContent({
+			model,
+			contents : { parts },
+			config : {
+				...config,
+				responseModalities : [ 'Image' ]
 			}
+		}) );
 
-			if( userId ){
+		const content = response?.candidates?.[ 0 ]?.content?.parts?.[ 0 ]?.inlineData?.data;
 
-				await billRequest({
-					db,
-					authenticated,
-					userId,
-					model,
-					usage : tokensFromMetadata( response?.usageMetadata )
-				});
+		if( !content ){
 
-			}
+			throw new Error( 'Could not generate from prompt. Please try again with a different prompt.' );
 
-			return { content };
+		}
 
-		} catch ( error ) {
+		if( user?.id ){
 
-			throw new Error( error?.message || 'AI error, please try again' );
+			await billRequest({
+				db,
+				user,
+				model,
+				usage : tokensFromMetadata( response?.usageMetadata )
+			});
 
 		}
 
+		return { content };
+
 	},
 
 	text : async (
@@ -191,54 +222,44 @@ const google = {
 		},
 		{
 			db,
-			authenticated,
-			userId
+			user
 		}
 	) => {
 
 		const model = models.text;
 
-		try {
-
-			const response = await googleCircuit( () => google_client.models.generateContent({
-				config : {
-					...config,
-					responseModalities : [ 'Text' ]
-				},
-				contents : prompt.join( '\n' ),
-				model
-			}) );
-
-			const content = response?.text;
+		const response = await googleCircuit( () => google_client.models.generateContent({
+			config : {
+				...config,
+				responseModalities : [ 'Text' ]
+			},
+			contents : prompt.join( '\n' ),
+			model
+		}) );
 
-			if( !content ){
+		const content = response?.text;
 
-				throw new Error( 'Could not generate from prompt. Please try again with a different prompt.' );
+		if( !content ){
 
-			}
-
-			if( userId ){
-
-				await billRequest({
-					db,
-					authenticated,
-					userId,
-					model,
-					usage : tokensFromMetadata( response?.usageMetadata )
-				});
-
-			}
+			throw new Error( 'Could not generate from prompt. Please try again with a different prompt.' );
 
-			return { content };
+		}
 
-		} catch ( error ) {
+		if( user?.id ){
 
-			throw new Error( error?.message || 'AI error, please try again' );
+			await billRequest({
+				db,
+				user,
+				model,
+				usage : tokensFromMetadata( response?.usageMetadata )
+			});
 
 		}
 
+		return { content };
+
 	}
 
 };
 
-export { google, priceForRequest, pricing };
+export { MARKUP, cost, costForRequest, google, priceForRequest, pricing };