@drawbridge/drawbridge-utils 0.0.29 → 0.0.32

package/dist/ai.d.ts

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

package/dist/ai.js

@@ -1,6 +1,6 @@
 import {
   debit
-} from "./chunk-H735KDYS.js";
+} from "./chunk-RW2FJVIV.js";
 import {
   circuit
 } from "./chunk-6HH36OK6.js";
@@ -19,7 +19,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,
@@ -31,18 +31,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,
@@ -51,24 +66,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
+      }
     }
   });
 };
@@ -79,8 +102,7 @@ var google = {
     reference = null
   }, {
     db,
-    authenticated,
-    userId
+    user
   }) => {
     var _a, _b, _c, _d, _e, _f;
     const model = models.image;
@@ -88,71 +110,63 @@ 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 };
   }
 };
 export {
+  MARKUP,
+  cost,
+  costForRequest,
   google,
   priceForRequest,
   pricing

package/dist/chunk-RW2FJVIV.js

@@ -0,0 +1,110 @@
+// transactions.js
+var insertTransaction = async ({
+  db,
+  user,
+  type,
+  category,
+  source,
+  amount,
+  _id,
+  stripeInvoiceId,
+  stripeEventId,
+  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: user,
+    collection: "transaction",
+    data: {
+      ..._id && { _id },
+      userId: user == null ? void 0 : user.id,
+      type,
+      category,
+      source,
+      amount,
+      ...balance && { balance },
+      ...rest,
+      ...stripeInvoiceId && { stripeInvoiceId },
+      ...stripeEventId && { stripeEventId }
+    },
+    ...session && { options: { session } }
+  });
+};
+var credit = async ({
+  db,
+  user,
+  amount,
+  type,
+  category,
+  source,
+  _id,
+  stripeInvoiceId,
+  stripeEventId,
+  session,
+  ...rest
+}) => {
+  if (!Number.isInteger(amount) || amount <= 0) {
+    throw new Error(`credit() requires a positive integer amount, got ${amount}`);
+  }
+  if (!type) {
+    throw new Error('credit() requires a type (e.g., "ai")');
+  }
+  if (!source) {
+    throw new Error('credit() requires a source ("system" | "admin" | "user")');
+  }
+  await insertTransaction({
+    db,
+    user,
+    type,
+    category,
+    source,
+    amount,
+    _id,
+    stripeInvoiceId,
+    stripeEventId,
+    session,
+    ...rest
+  });
+};
+var debit = async ({
+  db,
+  user,
+  amount,
+  type,
+  category,
+  source,
+  stripeInvoiceId,
+  stripeEventId,
+  session,
+  ...rest
+}) => {
+  if (!Number.isInteger(amount) || amount <= 0) {
+    throw new Error(`debit() requires a positive integer amount, got ${amount}`);
+  }
+  if (!type) {
+    throw new Error('debit() requires a type (e.g., "ai")');
+  }
+  if (!source) {
+    throw new Error('debit() requires a source ("system" | "admin" | "user")');
+  }
+  await insertTransaction({
+    db,
+    user,
+    type,
+    category,
+    source,
+    amount: -amount,
+    stripeInvoiceId,
+    stripeEventId,
+    session,
+    ...rest
+  });
+};
+
+export {
+  credit,
+  debit
+};