課金変更が発効した日に、headless 実行のコストを工程ごとに計測する仕組みを入れた

// 単価は「100万トークンあたりの円」で持つ（MTok単価）。
// 値はあなたの契約・最新の価格表・為替で必ず置き換えてください。
// モデルごとに別テーブルを持てるよう、キーをモデル名にしています。
type Rate = { input: number; output: number; cacheWrite: number; cacheRead: number };
 
const RATE_TABLE: Record<string, Rate> = {
  // 例: 値はプレースホルダー。実際の単価に差し替えること
  "default": { input: 450, output: 2250, cacheWrite: 560, cacheRead: 45 },
};
 
function rateFor(model: string): Rate {
  // 完全一致がなければ default にフォールバック
  return RATE_TABLE[model] ?? RATE_TABLE["default"];
}
 
// 正規化済み usage を円に変換する
export function usageToYen(model: string, u: ReturnType<typeof normalizeUsage>) {
  const r = rateFor(model);
  const perMTok = (tokens: number, rate: number) => (tokens / 1_000_000) * rate;
  return (
    perMTok(u.input, r.input) +
    perMTok(u.output, r.output) +
    perMTok(u.cacheWrite, r.cacheWrite) +
    perMTok(u.cacheRead, r.cacheRead)
  );
}

import Anthropic from "@anthropic-ai/sdk";
 
const client = new Anthropic();
 
// 台帳への1レコード
type CostRecord = {
  ts: string;     // ISO 8601
  stage: string;  // 工程タグ（"generate" / "quality-gate" など）
  model: string;
  yen: number;
  tokens: ReturnType<typeof normalizeUsage>;
};
 
// 記録先は差し替え可能にしておく（後述）
export interface CostSink {
  write(rec: CostRecord): Promise<void>;
}
 
// messages.create を包むだけのラッパー
export async function meteredCreate(
  sink: CostSink,
  stage: string,
  params: Anthropic.MessageCreateParamsNonStreaming,
) {
  const res = await client.messages.create(params);
  const usage = normalizeUsage(res.usage);
  const yen = usageToYen(params.model, usage);
  // 記録の失敗で本処理を止めないため、await しても sink 側で握り潰す設計にする
  await sink.write({
    ts: new Date().toISOString(),
    stage,
    model: params.model,
    yen,
    tokens: usage,
  });
  return res;
}

// before:
// const res = await client.messages.create({ model, max_tokens, messages });
 
// after: stage を足すだけ
const res = await meteredCreate(sink, "generate", { model, max_tokens, messages });

import { appendFile } from "node:fs/promises";
 
// JSON Lines（1行1レコード）で日付別ファイルに追記する Sink
export class JsonlFileSink implements CostSink {
  constructor(private dir: string) {}
  async write(rec: CostRecord): Promise<void> {
    const day = rec.ts.slice(0, 10); // YYYY-MM-DD
    try {
      await appendFile(`${this.dir}/cost-${day}.jsonl`, JSON.stringify(rec) + "\n");
    } catch (e) {
      // 計測失敗で本処理を止めない
      console.warn("[cost-meter] write failed:", (e as Error).message);
    }
  }
}

import { readFile } from "node:fs/promises";
 
type StageAgg = { yen: number; calls: number; cacheReadTokens: number; totalInputTokens: number };
 
export async function dailyReport(dir: string, day: string) {
  const lines = (await readFile(`${dir}/cost-${day}.jsonl`, "utf8"))
    .split("\n").filter(Boolean);
 
  const byStage = new Map<string, StageAgg>();
  for (const line of lines) {
    const r: CostRecord = JSON.parse(line);
    const a = byStage.get(r.stage) ?? { yen: 0, calls: 0, cacheReadTokens: 0, totalInputTokens: 0 };
    a.yen += r.yen;
    a.calls += 1;
    a.cacheReadTokens += r.tokens.cacheRead;
    a.totalInputTokens += r.tokens.input + r.tokens.cacheRead + r.tokens.cacheWrite;
    byStage.set(r.stage, a);
  }
 
  // 円の多い順に並べる
  const rows = [...byStage.entries()]
    .map(([stage, a]) => ({
      stage,
      yen: Math.round(a.yen * 100) / 100,
      calls: a.calls,
      // 入力に占めるキャッシュ読込の割合（高いほどキャッシュが効いている）
      cacheReadRatio: a.totalInputTokens
        ? Math.round((a.cacheReadTokens / a.totalInputTokens) * 100)
        : 0,
    }))
    .sort((x, y) => y.yen - x.yen);
 
  return { day, total: rows.reduce((s, r) => s + r.yen, 0), rows };
}