dsh-plugin-billing 0.0.0-stage → 0.1.1

package/src/pricing-body.js

@@ -0,0 +1,161 @@
+/**
+ * 计费单价与计价 —— 会被编译进浏览器 bundle（见 `scripts/build-client.mjs`）。
+ *
+ * 本文件是**纯声明**（没有 import/export），因为 bundle 是外壳直接执行的 classic
+ * script，只能通过 `require` 拿统一基座里的共享模块。构建脚本把它与
+ * `client-body.js` 一起拼进 factory 闭包。
+ *
+ * ## 单价来源
+ *
+ * DeepSeek 官方「模型 & 价格」页（元 / 百万 tokens，**高峰时段**）：
+ * https://api-docs.deepseek.com/zh-cn/quick_start/pricing
+ *
+ * 官方口径：空闲时段价格是高峰时段的一半；高峰期（北京时间）为周一至周五
+ * 09:00–12:00 与 14:00–18:00，其余为空闲时段。因此这里只登记高峰价，
+ * 空闲价由 `rateFactor()` 折算，避免两套数字各自漂移。
+ *
+ * ## 计费口径
+ *
+ * 扣费 = token 消耗量 × 单价。四个桶对应提供方的三类计费项：
+ *
+ * | 桶 | 来源 | 计费方式 |
+ * |---|---|---|
+ * | 缓存未命中输入 | `usage.inputTokens` | 缓存未命中单价 |
+ * | 缓存命中输入 | `usage.cacheReadTokens` | 缓存命中单价 |
+ * | 缓存写入 | `usage.cacheWriteTokens` | 并入缓存未命中（DeepSeek 不单列此项） |
+ * | 输出 | `usage.outputTokens` | 输出单价 |
+ *
+ * `usage.reasoningTokens` 是输出的一部分（官方与 harness 的 token-meter 同口径），
+ * **不重复计入**。
+ */
+
+/** 计费币种。与余额接口返回的 `currency` 对齐。 */
+const CURRENCY = 'CNY';
+
+/**
+ * 高峰时段单价（元 / 百万 tokens），构建期由 `scripts/build-client.mjs` 从仓库根的
+ * `pricing.json` 内联进来。key 必须与 harness 路由里的模型 id 完全一致
+ * （来自 `request/context` 的 `model`）。
+ *
+ * 这只是**兜底**：Host 侧每次请求都读同一份 `pricing.json`（并支持
+ * `$DSH_HOME/billing-prices.json` 覆盖），通过路由的 `prices` 字段下发。因此改价只需编辑
+ * 那个 JSON —— 不用重新构建 bundle，也不用重启服务。
+ */
+const DEFAULT_PRICES = {/*__PRICES__*/};
+
+/** 定价页来源标注的兜底值（正常由路由下发 `priceSource`）。 */
+const PRICE_SOURCE = 'DeepSeek 官方定价页';
+
+/**
+ * 选用哪张价格表：**路由下发的优先**，缺失时退回构建期内联的兜底表。
+ * @param data - 路由响应体（可能为 null）。
+ * @returns 高峰价表。
+ */
+function priceTable(data) {
+  if (data !== null && typeof data === 'object' && data.prices !== null && typeof data.prices === 'object') {
+    const keys = Object.keys(data.prices);
+    if (keys.length > 0) return data.prices;
+  }
+  return DEFAULT_PRICES;
+}
+
+/** 高峰时段（北京时间当天分钟数区间，左闭右开）。 */
+const PEAK_WINDOWS = [
+  [9 * 60, 12 * 60],
+  [14 * 60, 18 * 60],
+];
+
+/** 由 UTC 毫秒换到北京时间的固定偏移；不依赖运行机器的本地时区。 */
+const BEIJING_OFFSET_MS = 8 * 60 * 60 * 1000;
+
+/**
+ * 判断某一时刻是否落在高峰时段。
+ * 用 UTC getter 读取“加上固定偏移后的墙上时间”，因此主机时区无关。
+ * @param atMs - epoch 毫秒。
+ * @returns 高峰时段为 true；周末恒为 false。
+ */
+function isPeak(atMs) {
+  const shifted = new Date(atMs + BEIJING_OFFSET_MS);
+  const weekday = shifted.getUTCDay();
+  if (weekday === 0 || weekday === 6) return false;
+  const minutes = shifted.getUTCHours() * 60 + shifted.getUTCMinutes();
+  return PEAK_WINDOWS.some(function (window) {
+    return minutes >= window[0] && minutes < window[1];
+  });
+}
+
+/**
+ * 高峰倍率：高峰 1，空闲 0.5（官方“空闲为高峰一半”）。
+ * @param atMs - epoch 毫秒。
+ * @returns 倍率。
+ */
+function rateFactor(atMs) {
+  return isPeak(atMs) ? 1 : 0.5;
+}
+
+/**
+ * 取某模型在某时刻的三档单价（元 / 百万 tokens）。
+ * @param model - 模型 id。
+ * @param atMs - epoch 毫秒。
+ * @param table - 可选的价格表；缺省用构建期内联的兜底表。
+ * @returns `{ cacheHit, cacheMiss, output, peak, factor }`；模型未登记时为 `undefined`。
+ */
+function ratesFor(model, atMs, table) {
+  const peak = (table ?? DEFAULT_PRICES)[model];
+  if (peak === undefined) return undefined;
+  const factor = rateFactor(atMs);
+  return {
+    cacheHit: peak.cacheHit * factor,
+    cacheMiss: peak.cacheMiss * factor,
+    output: peak.output * factor,
+    peak: factor === 1,
+    factor: factor,
+  };
+}
+
+/**
+ * 从一个 `TokenUsage` 取出四个计费桶。缺字段按 0 计。
+ * @param usage - 提供方或 harness 记录的用量对象。
+ * @returns `{ cacheMiss, cacheRead, cacheWrite, output }`。
+ */
+function bucketsOf(usage) {
+  const read = function (key) {
+    const value = usage === null || typeof usage !== 'object' ? undefined : usage[key];
+    return typeof value === 'number' && Number.isFinite(value) && value > 0 ? value : 0;
+  };
+  return {
+    cacheMiss: read('inputTokens'),
+    cacheRead: read('cacheReadTokens'),
+    cacheWrite: read('cacheWriteTokens'),
+    output: read('outputTokens'),
+  };
+}
+
+/** 四个桶逐项相加。 */
+function addBuckets(left, right) {
+  return {
+    cacheMiss: left.cacheMiss + right.cacheMiss,
+    cacheRead: left.cacheRead + right.cacheRead,
+    cacheWrite: left.cacheWrite + right.cacheWrite,
+    output: left.output + right.output,
+  };
+}
+
+/** 桶的总 token 数。 */
+function bucketsTotal(buckets) {
+  return buckets.cacheMiss + buckets.cacheRead + buckets.cacheWrite + buckets.output;
+}
+
+/**
+ * 按单价把桶折成金额。
+ * 缓存写入并入“缓存未命中”档，与 DeepSeek 的计费口径一致。
+ * @param buckets - 四个计费桶。
+ * @param rates - `ratesFor()` 的返回值。
+ * @returns 金额（`CURRENCY` 单位）。
+ */
+function costOf(buckets, rates) {
+  const billedMiss = buckets.cacheMiss + buckets.cacheWrite;
+  return (
+    (billedMiss * rates.cacheMiss + buckets.cacheRead * rates.cacheHit + buckets.output * rates.output) / 1e6
+  );
+}