@qjp88995/agent-runtime 0.0.0-stage → 0.1.0

package/src/turn/usage.ts

@@ -0,0 +1,56 @@
+/**
+ * 把 AI SDK 的 usage 拍平成 `Record<string, number>`。
+ *
+ * AI SDK v7 的 `LanguageModelUsage` 是**两层**的，而**缓存命中数只存在于嵌套
+ * 层**：`inputTokenDetails.{noCacheTokens, cacheReadTokens, cacheWriteTokens}`
+ * 与 `outputTokenDetails.{textTokens, reasoningTokens}`。
+ *
+ * 本仓从 `turn-envelope` 到 api-client 的 wire 再到 `chat_message.token_usage`
+ * 一路都声明成 `Record<string, number>`，原先那句
+ * `typeof v === 'number' ? v : 0` 会把两个 details 对象直接压成数字 `0`——
+ * 落库里的 `"inputTokenDetails": 0` 就是这么来的，缓存命中数从来没被记下来过，
+ * 而且它长得像一条「本次缓存命中 0」的真实观测，比缺字段更能骗人。
+ *
+ * **选择拍平而不是把类型改成嵌套**：嵌套要改内核、api-client、前端三层加一张
+ * 表，而这五个键名本来就互不重名、拿出来也仍然自解释。
+ *
+ * **`raw` 显式跳过**：那是各家 provider 的原始形状（deepseek 叫
+ * `prompt_cache_hit_tokens`、openai 叫 `cached_tokens`），混进来会让同一个概念
+ * 在不同 provider 下出现不同键名——正是 SDK 归一化想消掉的东西。
+ *
+ * @returns 没有任何数值时返回 `undefined`（调用方据此判断「这一步没有用量」，
+ *          空对象是真值、会被当成有用量）。
+ */
+export function flattenUsage(
+  usage: unknown
+): Record<string, number> | undefined {
+  if (!usage || typeof usage !== 'object') return undefined;
+
+  const out: Record<string, number> = {};
+  const take = (key: string, value: unknown) => {
+    if (typeof value === 'number' && Number.isFinite(value)) out[key] = value;
+  };
+
+  for (const [k, v] of Object.entries(usage as Record<string, unknown>)) {
+    if (k === 'raw') continue;
+    if (v && typeof v === 'object') {
+      for (const [nk, nv] of Object.entries(v as Record<string, unknown>)) {
+        take(nk, nv);
+      }
+      continue;
+    }
+    take(k, v);
+  }
+
+  return Object.keys(out).length > 0 ? out : undefined;
+}
+
+/** 逐键求和到累加器（跨 step 汇总一轮的用量）。 */
+export function addUsage(
+  acc: Record<string, number>,
+  step: Record<string, number> | undefined
+): void {
+  for (const [k, v] of Object.entries(step ?? {})) {
+    acc[k] = (acc[k] ?? 0) + v;
+  }
+}

package/README.md

@@ -1,3 +0,0 @@
-# Temporary Holding Version
-
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.