@drawbridge/drawbridge-utils 0.0.26 → 0.0.28

package/dist/ai.cjs

@@ -0,0 +1,328 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// ai.js
+var ai_exports = {};
+__export(ai_exports, {
+  google: () => google,
+  priceForRequest: () => priceForRequest,
+  pricing: () => pricing
+});
+module.exports = __toCommonJS(ai_exports);
+var import_genai = require("@google/genai");
+
+// circuit.js
+var CLOSED = "CLOSED";
+var OPEN = "OPEN";
+var HALF_OPEN = "HALF_OPEN";
+var circuit = ({
+  name,
+  threshold = 5,
+  timeout = 3e4
+}) => {
+  let state = CLOSED;
+  let failures = 0;
+  let openedAt = null;
+  const trip = () => {
+    state = OPEN;
+    openedAt = Date.now();
+    console.error(`Circuit breaker OPEN: ${name}`);
+  };
+  const reset = () => {
+    state = CLOSED;
+    failures = 0;
+    openedAt = null;
+    console.log(`Circuit breaker CLOSED: ${name}`);
+  };
+  return async (fn) => {
+    if (state === OPEN) {
+      if (Date.now() - openedAt >= timeout) {
+        state = HALF_OPEN;
+      } else {
+        const error = new Error(`${name} is temporarily unavailable`);
+        error.status = 503;
+        throw error;
+      }
+    }
+    try {
+      const result = await fn();
+      if (state === HALF_OPEN) {
+        reset();
+      } else {
+        failures = 0;
+      }
+      return result;
+    } catch (error) {
+      failures++;
+      if (state === HALF_OPEN || failures >= threshold) {
+        trip();
+      }
+      throw error;
+    }
+  };
+};
+
+// transactions.js
+var recordTransaction = async ({
+  db,
+  authenticated,
+  userId,
+  category,
+  tags,
+  amount,
+  balance,
+  provider,
+  stripeInvoiceId,
+  stripeEventId
+}) => {
+  await db.create({
+    authenticated,
+    collection: "transactions",
+    data: {
+      userId,
+      billable: {
+        item: "ai"
+      },
+      category,
+      tags,
+      amount,
+      balance,
+      ...provider && { provider },
+      ...stripeInvoiceId && { stripeInvoiceId },
+      ...stripeEventId && { stripeEventId }
+    }
+  });
+};
+var InsufficientCreditsError = class extends Error {
+  constructor(message = "Insufficient AI credits") {
+    super(message);
+    this.name = "InsufficientCreditsError";
+    this.status = 402;
+  }
+};
+var debit = async ({
+  db,
+  authenticated,
+  userId,
+  amount,
+  category,
+  tags = [],
+  provider,
+  stripeInvoiceId,
+  stripeEventId
+}) => {
+  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: "users",
+    query: {
+      id: userId,
+      "balance.ai": {
+        $gte: amount
+      }
+    },
+    data: {
+      $inc: {
+        "balance.ai": -amount
+      }
+    }
+  });
+  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();
+  }
+  const before = after + amount;
+  await recordTransaction({
+    db,
+    authenticated,
+    userId,
+    category,
+    tags,
+    amount: -amount,
+    balance: {
+      before,
+      after
+    },
+    provider,
+    stripeInvoiceId,
+    stripeEventId
+  });
+  return {
+    amount: -amount,
+    balance: {
+      before,
+      after
+    }
+  };
+};
+
+// ai.js
+var google_client = new import_genai.GoogleGenAI({
+  apiKey: process.env.GOOGLE_API_KEY
+});
+var googleCircuit = circuit({
+  name: "google-ai",
+  threshold: 3,
+  timeout: 6e4
+});
+var models = {
+  image: "gemini-2.5-flash-image",
+  text: "gemini-2.5-flash"
+};
+var pricing = {
+  "gemini-2.5-flash": {
+    cached: 3,
+    input: 30,
+    output: 250
+  },
+  "gemini-2.5-flash-image": {
+    cached: 3,
+    input: 30,
+    output: 250
+  }
+};
+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);
+};
+var tokensFromMetadata = (metadata) => ({
+  cached: Number(metadata == null ? void 0 : metadata.cachedContentTokenCount) || 0,
+  input: Number(metadata == null ? void 0 : metadata.promptTokenCount) || 0,
+  output: Number(metadata == null ? void 0 : metadata.candidatesTokenCount) || 0
+});
+var billRequest = async ({
+  db,
+  authenticated,
+  userId,
+  model,
+  usage
+}) => {
+  const amount = priceForRequest(model, usage);
+  await debit({
+    db,
+    authenticated,
+    userId,
+    amount,
+    category: "usage",
+    tags: ["google", model],
+    provider: {
+      name: "google",
+      model,
+      tokens: usage,
+      snapshot: pricing[model]
+    }
+  });
+};
+var google = {
+  image: async ({
+    config = {},
+    prompt = [],
+    reference = null
+  }, {
+    db,
+    authenticated,
+    userId
+  }) => {
+    var _a, _b, _c, _d, _e, _f;
+    const model = models.image;
+    const parts = [
+      ...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)
+        });
+      }
+      return { content };
+    } catch (error) {
+      throw new Error((error == null ? void 0 : error.message) || "AI error, please try again");
+    }
+  },
+  text: async ({
+    config = {},
+    prompt = []
+  }, {
+    db,
+    authenticated,
+    userId
+  }) => {
+    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");
+    }
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  google,
+  priceForRequest,
+  pricing
+});

package/dist/ai.d.cts

@@ -0,0 +1,244 @@
+import { GoogleGenAI } from '@google/genai';
+import { circuit } from './circuit.cjs';
+import { debit } from './transactions.cjs';
+
+// AI request entry point. Wraps the Google GenAI client with a per-process
+// circuit breaker, owns the per-model pricing table, and atomically debits
+// the user's balance.ai based on the token usage returned by the provider.
+// Both drawbridge-api routes and drawbridge-sync workers use this so the
+// wrapper, pricing, and accounting all stay in one place.
+//
+//   import { google } from '@drawbridge/drawbridge-utils/ai';
+//
+//   const { content } = await google.text(
+//     { prompt : [ 'Write me a haiku' ] },
+//     { db, authenticated, userId }
+//   );
+//
+// If the user runs out of credits between the gating middleware and the
+// debit call (a real but rare race), `debit` throws InsufficientCreditsError
+// — the Google call has already happened, accepted trade-off vs. doing a
+// second pre-flight balance read.
+
+
+const google_client = new GoogleGenAI({
+	apiKey : process.env.GOOGLE_API_KEY
+});
+
+const googleCircuit = circuit({
+	name : 'google-ai',
+	threshold : 3,
+	timeout : 60000
+});
+
+const models = {
+	image : 'gemini-2.5-flash-image',
+	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
+// https://github.com/BerriAI/litellm/blob/main/litellm/model_prices_and_context_window_backup.json
+// before any change.
+const pricing = {
+	'gemini-2.5-flash' : {
+		cached : 3,
+		input : 30,
+		output : 250
+	},
+	'gemini-2.5-flash-image' : {
+		cached : 3,
+		input : 30,
+		output : 250
+	}
+};
+
+// 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.
+const priceForRequest = ( model, usage ) => {
+
+	const rates = pricing[ model ];
+
+	if( !rates ){
+
+		throw new Error( `Unknown model for pricing: ${ model }` );
+
+	}
+
+	const tokens = {
+		cached : Number( usage?.cached ) || 0,
+		input : Number( usage?.input ) || 0,
+		output : Number( usage?.output ) || 0
+	};
+
+	const sum = (
+		tokens.cached * rates.cached +
+		tokens.input * rates.input +
+		tokens.output * rates.output
+	);
+
+	return Math.ceil( sum / 1_000_000 );
+
+};
+
+// Map Google's usageMetadata shape into the pricing.tokens shape.
+const tokensFromMetadata = ( metadata ) => ({
+	cached : Number( metadata?.cachedContentTokenCount ) || 0,
+	input : Number( metadata?.promptTokenCount ) || 0,
+	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.
+const billRequest = async ({
+	db,
+	authenticated,
+	userId,
+	model,
+	usage
+}) => {
+
+	const amount = priceForRequest( model, usage );
+
+	await debit({
+		db,
+		authenticated,
+		userId,
+		amount,
+		category : 'usage',
+		tags : [ 'google', model ],
+		provider : {
+			name : 'google',
+			model,
+			tokens : usage,
+			snapshot : pricing[ model ]
+		}
+	});
+
+};
+
+const google = {
+
+	image : async (
+		{
+			config = {},
+			prompt = [],
+			reference = null
+		},
+		{
+			db,
+			authenticated,
+			userId
+		}
+	) => {
+
+		const model = models.image;
+
+		const parts = [
+			...( 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 = 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.' );
+
+			}
+
+			if( userId ){
+
+				await billRequest({
+					db,
+					authenticated,
+					userId,
+					model,
+					usage : tokensFromMetadata( response?.usageMetadata )
+				});
+
+			}
+
+			return { content };
+
+		} catch ( error ) {
+
+			throw new Error( error?.message || 'AI error, please try again' );
+
+		}
+
+	},
+
+	text : async (
+		{
+			config = {},
+			prompt = []
+		},
+		{
+			db,
+			authenticated,
+			userId
+		}
+	) => {
+
+		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;
+
+			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?.usageMetadata )
+				});
+
+			}
+
+			return { content };
+
+		} catch ( error ) {
+
+			throw new Error( error?.message || 'AI error, please try again' );
+
+		}
+
+	}
+
+};
+
+export { google, priceForRequest, pricing };