@rpcbase/agents-sdk 0.21.0 → 0.22.0

package/dist/usage.js

@@ -0,0 +1,124 @@
+import { models as e } from "@rpcbase/db";
+//#region src/usage.ts
+var t = 1e6, n = 1e6, r = {
+	requestCount: 0,
+	inputTokens: 0,
+	outputTokens: 0,
+	totalTokens: 0,
+	billingAmountMicros: 0
+}, i = (e) => typeof e != "number" || !Number.isFinite(e) || e <= 0 ? 0 : Math.trunc(e), a = (e) => {
+	if (!e || typeof e != "object" || Array.isArray(e)) return;
+	let t = e, n = t.totalUsage ?? t.usage;
+	if (n !== void 0) return n;
+	if (!t.error || typeof t.error != "object" || Array.isArray(t.error)) return;
+	let r = t.error;
+	return r.totalUsage ?? r.usage;
+}, o = (e, t) => {
+	if (typeof e != "number" || !Number.isSafeInteger(e) || e < 0) throw Error(`${t} must be a non-negative safe integer.`);
+	return e;
+}, s = (e) => {
+	let t = typeof e == "string" ? e.trim().toUpperCase() : "";
+	if (!/^[A-Z]{3}$/.test(t)) throw Error("Agent usage billing currency must be a three-letter ISO code.");
+	return t;
+}, c = (e) => {
+	let t = typeof e == "string" ? e.trim() : "";
+	if (!t) throw Error("Agent usage billing price version is required.");
+	return t;
+}, l = (e, t) => {
+	let r = i(e.inputTokens), a = i(e.outputTokens), l = o(t.inputPriceMicrosPerMillion, "inputPriceMicrosPerMillion"), u = o(t.outputPriceMicrosPerMillion, "outputPriceMicrosPerMillion"), d = (r * l + a * u) / n, f = Math.round(d);
+	if (!Number.isSafeInteger(f)) throw Error("Agent usage billing amount exceeds the supported integer range.");
+	return {
+		billingCurrency: s(t.currency),
+		billingPriceVersion: c(t.version),
+		inputPriceMicrosPerMillion: l,
+		outputPriceMicrosPerMillion: u,
+		billingAmountMicros: f
+	};
+}, u = (e) => Number.isFinite(e) ? e / t : 0, d = (e) => {
+	let t = typeof e == "string" ? e.trim() : "";
+	if (!t) return "UTC";
+	try {
+		return new Intl.DateTimeFormat("en", { timeZone: t }).format(), t;
+	} catch {
+		return "UTC";
+	}
+}, f = (e) => ({
+	requestCount: { $sum: 1 },
+	inputTokens: { $sum: { $ifNull: ["$inputTokens", 0] } },
+	outputTokens: { $sum: { $ifNull: ["$outputTokens", 0] } },
+	totalTokens: { $sum: { $ifNull: ["$totalTokens", { $add: [{ $ifNull: ["$inputTokens", 0] }, { $ifNull: ["$outputTokens", 0] }] }] } },
+	billingAmountMicros: { $sum: { $cond: [
+		{ $eq: ["$billingCurrency", e] },
+		{ $ifNull: ["$billingAmountMicros", 0] },
+		0
+	] } }
+}), p = (e) => {
+	let t = i(e?.inputTokens), n = i(e?.outputTokens), r = e?.totalTokens === void 0 || e.totalTokens === null ? t + n : i(e.totalTokens);
+	return {
+		requestCount: i(e?.requestCount),
+		inputTokens: t,
+		outputTokens: n,
+		totalTokens: r,
+		billingAmountMicros: i(e?.billingAmountMicros)
+	};
+}, m = async (t) => {
+	let n = t.tenantId.trim(), i = t.workspaceId.trim();
+	if (!n || !i) throw Error("Agent usage summary requires tenantId and workspaceId.");
+	if (!(t.startAt instanceof Date) || !Number.isFinite(t.startAt.getTime())) throw Error("Agent usage summary startAt must be a valid date.");
+	if (!(t.endAt instanceof Date) || !Number.isFinite(t.endAt.getTime())) throw Error("Agent usage summary endAt must be a valid date.");
+	if (t.startAt >= t.endAt) throw Error("Agent usage summary endAt must be after startAt.");
+	let a = s(t.currency ?? "EUR"), o = f(a), [c] = await (await e.get("RBAgentsUsageEvent", t.ctx)).aggregate([{ $match: {
+		tenantId: n,
+		workspaceId: i,
+		createdAt: {
+			$gte: t.startAt,
+			$lt: t.endAt
+		}
+	} }, { $facet: {
+		summary: [{ $group: {
+			_id: null,
+			...o
+		} }],
+		dailyUsage: [{ $group: {
+			_id: { $dateToString: {
+				date: "$createdAt",
+				format: "%Y-%m-%d",
+				timezone: d(t.timeZone)
+			} },
+			...o
+		} }, { $sort: { _id: 1 } }],
+		byModel: [{ $group: {
+			_id: {
+				provider: "$provider",
+				model: "$model"
+			},
+			...o
+		} }, { $sort: {
+			billingAmountMicros: -1,
+			inputTokens: -1,
+			outputTokens: -1,
+			"_id.model": 1
+		} }]
+	} }]);
+	return {
+		currency: a,
+		summary: c?.summary?.[0] ? p(c.summary[0]) : { ...r },
+		dailyUsage: (c?.dailyUsage ?? []).flatMap((e) => {
+			let t = typeof e._id == "string" ? e._id : "";
+			return t ? [{
+				date: t,
+				...p(e)
+			}] : [];
+		}),
+		byModel: (c?.byModel ?? []).flatMap((e) => {
+			let t = typeof e._id?.provider == "string" ? e._id.provider : "", n = typeof e._id?.model == "string" ? e._id.model : "";
+			return n ? [{
+				provider: t,
+				model: n,
+				...p(e)
+			}] : [];
+		})
+	};
+};
+//#endregion
+export { u as agentUsageMicrosToMajor, l as calculateAgentUsageBilling, m as getAgentUsageSummary, a as getReportedAgentUsage };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rpcbase/agents-sdk",
-  "version": "0.21.0",
+  "version": "0.22.0",
   "publishConfig": {
     "access": "public",
     "registry": "https://registry.npmjs.org/"
@@ -65,6 +65,11 @@
       "types": "./dist/runtime.d.ts",
       "default": "./dist/runtime.js"
     },
+    "./usage": {
+      "import": "./dist/usage.js",
+      "types": "./dist/usage.d.ts",
+      "default": "./dist/usage.js"
+    },
     "./api-catalog-search": {
       "import": "./dist/api-catalog-search.js",
       "types": "./dist/api-catalog-search.d.ts",