nolo-cli 0.1.8 → 0.1.10

package/ai/token/calculatePrice.ts

@@ -0,0 +1,544 @@
+// 路径: ai/token/calculatePrice.ts
+
+import { nolotusId } from "core/init";
+import { Model, ModelPrice } from "ai/llm/types";
+import { getModelConfig } from "ai/llm/providers";
+import {
+  isOpenAIFlexPricedModel,
+  type OpenAIServiceTier,
+  resolveOpenAIServiceTier,
+} from "integrations/openai/flexTier";
+
+// ==================== 接口定义 ====================
+
+/**
+ * 这里的 Usage 是「已经 normalize 过」的结构
+ * ——来自 normalizeUsage，字段名统一成这几个
+ */
+export interface Usage {
+  input_tokens: number;
+  output_tokens: number;
+  cache_creation_input_tokens: number;
+  cache_read_input_tokens: number;
+  /**
+   * 对于 OpenRouter 等支持 usage.cost 的 provider：
+   * usage.cost 是 provider 返回的账户扣费金额。
+   * OpenRouter 官方 credits 以 USD 为基准，本仓库仍需再换算成平台 credits。
+   */
+  cost?: number;
+  billing_service_tier?: string;
+  image_generation_count?: number;
+}
+
+export interface ExternalPrice {
+  input: number;
+  output: number;
+  creatorId?: string;
+}
+
+interface CalculatePriceParams {
+  modelName: string;
+  usage: Usage;
+  externalPrice?: ExternalPrice;
+  /**
+   * provider 名称（deepseek / openrouter / anthropic 等）。
+   * apiSource="custom" 的 agent 不填此字段，传 undefined 即可。
+   * 注意：不要依赖参数默认值，调用方应显式传值或不传。
+   */
+  provider?: string;
+  billingServiceTier?: string;
+  sharingLevel?: "default" | "split" | "full";
+}
+
+export interface PriceResult {
+  cost: number;
+  pay: Record<string, number>;
+}
+
+export interface CostBreakdown {
+  regular: number;
+  charge: number;
+  details?: {
+    inputCost: number;
+    outputCost: number;
+    cachingWriteCost: number;
+    cachingReadCost: number;
+  };
+}
+
+const OPENAI_BUILTIN_IMAGE_FALLBACK_MODEL = "gpt-image-2";
+
+const resolveOpenAIBuiltInImageSurcharge = (
+  model: Model,
+  usage: Usage,
+  provider?: string
+): number => {
+  if (provider !== "openai" || model.hasImageOutput) return 0;
+  const imageGenerationCount =
+    typeof usage.image_generation_count === "number" &&
+    Number.isFinite(usage.image_generation_count)
+      ? usage.image_generation_count
+      : 0;
+  if (imageGenerationCount <= 0) return 0;
+
+  try {
+    const imageModel = getModelConfig("openai", OPENAI_BUILTIN_IMAGE_FALLBACK_MODEL);
+    if (typeof imageModel.pricePerImage !== "number") return 0;
+    return imageModel.pricePerImage * imageGenerationCount;
+  } catch {
+    return 0;
+  }
+};
+
+// ==================== 核心逻辑：阶梯定价解析 ====================
+
+const resolveModelPrice = (model: Model, usage: Usage): ModelPrice => {
+  let activePrice = { ...model.price };
+
+  if (model.pricingStrategy?.type === "tiered_context") {
+    const contextSize = usage.input_tokens || 0;
+    const tiers = model.pricingStrategy.tiers || [];
+
+    const sortedTiers = [...tiers].sort((a, b) => a.minContext - b.minContext);
+
+    for (const tier of sortedTiers) {
+      if (contextSize >= tier.minContext) {
+        activePrice = { ...tier.price };
+      }
+    }
+  }
+
+  return activePrice;
+};
+
+const scaleModelPrice = (price: ModelPrice, multiplier: number): ModelPrice => ({
+  input: price.input * multiplier,
+  output: price.output * multiplier,
+  ...(typeof price.cachingWrite === "number"
+    ? { cachingWrite: price.cachingWrite * multiplier }
+    : {}),
+  ...(typeof price.cachingRead === "number"
+    ? { cachingRead: price.cachingRead * multiplier }
+    : {}),
+  ...(typeof price.inputCacheHit === "number"
+    ? { inputCacheHit: price.inputCacheHit * multiplier }
+    : {}),
+});
+
+const scaleModelInputOutputPrice = (
+  price: ModelPrice,
+  multiplier: number
+): ModelPrice => ({
+  ...price,
+  input: price.input * multiplier,
+  output: price.output * multiplier,
+});
+
+const resolveGoogleServiceTierPrice = (
+  model: Model,
+  price: ModelPrice,
+  billingServiceTier?: string
+): ModelPrice => {
+  const normalizedTier =
+    typeof billingServiceTier === "string"
+      ? billingServiceTier.trim().toLowerCase()
+      : "";
+  const imageOutputPrice =
+    typeof model.imageTokenPricePerMillion === "number"
+      ? model.imageTokenPricePerMillion
+      : undefined;
+  const priceWithImageOutput =
+    typeof imageOutputPrice === "number"
+      ? { ...price, output: imageOutputPrice }
+      : price;
+
+  if (normalizedTier === "flex" || normalizedTier === "batch") {
+    return scaleModelInputOutputPrice(priceWithImageOutput, 0.5);
+  }
+
+  if (normalizedTier === "priority") {
+    return scaleModelInputOutputPrice(priceWithImageOutput, 1.8);
+  }
+
+  return priceWithImageOutput;
+};
+
+const resolveEffectiveModelPrice = ({
+  model,
+  usage,
+  provider,
+  billingServiceTier,
+}: {
+  model: Model;
+  usage: Usage;
+  provider: string;
+  billingServiceTier?: string;
+}): ModelPrice => {
+  const resolvedPrice = resolveModelPrice(model, usage);
+  if (provider === "google") {
+    return resolveGoogleServiceTierPrice(model, resolvedPrice, billingServiceTier);
+  }
+  const effectiveOpenAIServiceTier = resolveOpenAIServiceTier({
+    providerName: provider,
+    model: model.name,
+    requestedServiceTier: billingServiceTier as OpenAIServiceTier | undefined,
+  });
+
+  if (provider === "openai" && effectiveOpenAIServiceTier === "flex") {
+    if (isOpenAIFlexPricedModel(model.name)) {
+      return resolvedPrice;
+    }
+    return scaleModelPrice(resolvedPrice, 0.5);
+  }
+
+  return resolvedPrice;
+};
+
+// ==================== 辅助逻辑：价格保护 ====================
+
+const getEffectivePrices = (
+  resolvedPrice: ModelPrice,
+  externalPrice?: ExternalPrice
+): {
+  input: number;
+  output: number;
+  cachingWrite: number;
+  cachingRead: number;
+} => {
+  const effectiveInputPrice = Math.max(
+    externalPrice?.input || 0,
+    resolvedPrice.input
+  );
+  const effectiveOutputPrice = Math.max(
+    externalPrice?.output || 0,
+    resolvedPrice.output
+  );
+
+  return {
+    input: effectiveInputPrice,
+    output: effectiveOutputPrice,
+    cachingWrite: resolvedPrice.cachingWrite || 0,
+    cachingRead: resolvedPrice.cachingRead || 0,
+  };
+};
+
+// ==================== 具体计算函数 ====================
+
+const calculateAnthropicCost = (
+  resolvedPrice: ModelPrice,
+  usage: Usage,
+  externalPrice?: ExternalPrice
+): CostBreakdown => {
+  const {
+    input_tokens,
+    output_tokens,
+    cache_creation_input_tokens,
+    cache_read_input_tokens,
+  } = usage;
+
+  const {
+    input: effectiveInputPrice,
+    output: effectiveOutputPrice,
+    cachingWrite: effectiveCachingWritePrice,
+    cachingRead: effectiveCachingReadPrice,
+  } = getEffectivePrices(resolvedPrice, externalPrice);
+
+  const regularInputTokens = input_tokens - cache_read_input_tokens;
+
+  const regularTotal =
+    (regularInputTokens * resolvedPrice.input +
+      output_tokens * resolvedPrice.output +
+      cache_creation_input_tokens * (resolvedPrice.cachingWrite || 0) +
+      cache_read_input_tokens * (resolvedPrice.cachingRead || 0)) /
+    1_000_000;
+
+  const chargeTotal =
+    (regularInputTokens * effectiveInputPrice +
+      output_tokens * effectiveOutputPrice +
+      cache_creation_input_tokens * effectiveCachingWritePrice +
+      cache_read_input_tokens * effectiveCachingReadPrice) /
+    1_000_000;
+
+  return {
+    regular: regularTotal,
+    charge: chargeTotal,
+    details: {
+      inputCost: (regularInputTokens * resolvedPrice.input) / 1_000_000,
+      outputCost: (output_tokens * resolvedPrice.output) / 1_000_000,
+      cachingWriteCost:
+        (cache_creation_input_tokens * (resolvedPrice.cachingWrite || 0)) /
+        1_000_000,
+      cachingReadCost:
+        (cache_read_input_tokens * (resolvedPrice.cachingRead || 0)) /
+        1_000_000,
+    },
+  };
+};
+
+const calculateCacheBasedCost = (
+  resolvedPrice: ModelPrice,
+  usage: Usage,
+  externalPrice?: ExternalPrice
+): CostBreakdown => {
+  const { input_tokens, output_tokens, cache_read_input_tokens } = usage;
+  const { input: effectiveInputPrice, output: effectiveOutputPrice } =
+    getEffectivePrices(resolvedPrice, externalPrice);
+
+  const cacheMissTokens = input_tokens - cache_read_input_tokens;
+  const cacheHitPrice = resolvedPrice.inputCacheHit || 0;
+
+  const regularTotal =
+    (cacheMissTokens * resolvedPrice.input +
+      cache_read_input_tokens * cacheHitPrice +
+      output_tokens * resolvedPrice.output) /
+    1_000_000;
+
+  const chargeTotal =
+    (cacheMissTokens * effectiveInputPrice +
+      cache_read_input_tokens * cacheHitPrice +
+      output_tokens * effectiveOutputPrice) /
+    1_000_000;
+
+  return {
+    regular: regularTotal,
+    charge: chargeTotal,
+    details: {
+      inputCost: (cacheMissTokens * resolvedPrice.input) / 1_000_000,
+      outputCost: (output_tokens * resolvedPrice.output) / 1_000_000,
+      cachingReadCost: (cache_read_input_tokens * cacheHitPrice) / 1_000_000,
+      cachingWriteCost: 0,
+    },
+  };
+};
+
+const calculateSimpleCost = (
+  resolvedPrice: ModelPrice,
+  usage: Usage,
+  externalPrice?: ExternalPrice
+): CostBreakdown => {
+  const { input_tokens, output_tokens } = usage;
+  const { input: effectiveInputPrice, output: effectiveOutputPrice } =
+    getEffectivePrices(resolvedPrice, externalPrice);
+
+  const regularTotal =
+    (input_tokens * resolvedPrice.input +
+      output_tokens * resolvedPrice.output) /
+    1_000_000;
+
+  const chargeTotal =
+    (input_tokens * effectiveInputPrice +
+      output_tokens * effectiveOutputPrice) /
+    1_000_000;
+
+  return {
+    regular: regularTotal,
+    charge: chargeTotal,
+    details: {
+      inputCost: (input_tokens * resolvedPrice.input) / 1_000_000,
+      outputCost: (output_tokens * resolvedPrice.output) / 1_000_000,
+      cachingWriteCost: 0,
+      cachingReadCost: 0,
+    },
+  };
+};
+
+const calculateOpenRouterFallbackCost = (
+  resolvedPrice: ModelPrice,
+  usage: Usage,
+  externalPrice?: ExternalPrice
+): CostBreakdown => {
+  if (
+    typeof resolvedPrice.cachingWrite === "number" ||
+    typeof resolvedPrice.cachingRead === "number"
+  ) {
+    return calculateAnthropicCost(resolvedPrice, usage, externalPrice);
+  }
+
+  if (typeof resolvedPrice.inputCacheHit === "number") {
+    return calculateCacheBasedCost(resolvedPrice, usage, externalPrice);
+  }
+
+  return calculateSimpleCost(resolvedPrice, usage, externalPrice);
+};
+
+// OpenRouter usage.cost 是按其账户 credit（USD 基准）返回，
+// 本平台统一按 1 USD = 8 credits 记账，因此这里仍需做一次换算。
+const OPENROUTER_COST_MULTIPLIER = 8;
+
+const calculateOpenRouterCost = (usage: Usage): CostBreakdown => {
+  if (!usage || typeof usage.cost !== "number" || usage.cost <= 0) {
+    return {
+      regular: 0,
+      charge: 0,
+      details: {
+        inputCost: 0,
+        outputCost: 0,
+        cachingWriteCost: 0,
+        cachingReadCost: 0,
+      },
+    };
+  }
+
+  const regular = usage.cost * OPENROUTER_COST_MULTIPLIER;
+
+  return {
+    regular,
+    charge: regular,
+    details: {
+      inputCost: regular,
+      outputCost: 0,
+      cachingWriteCost: 0,
+      cachingReadCost: 0,
+    },
+  };
+};
+
+// ==================== 路由与分发 ====================
+
+const calculateBasicCost = (
+  model: Model,
+  usage: Usage,
+  provider: string,
+  externalPrice?: ExternalPrice,
+  billingServiceTier?: OpenAIServiceTier
+): CostBreakdown => {
+  if (!usage || typeof usage.input_tokens !== "number") {
+    throw new Error("Invalid usage data");
+  }
+
+  const resolvedPrice = resolveEffectiveModelPrice({
+    model,
+    usage,
+    provider,
+    billingServiceTier,
+  });
+
+  switch (provider) {
+    case "deepseek":
+    case "openai":
+    case "deepinfra":
+      return calculateCacheBasedCost(resolvedPrice, usage, externalPrice);
+
+    case "anthropic":
+      return calculateAnthropicCost(resolvedPrice, usage, externalPrice);
+
+    case "google":
+      if (model.name.includes("gemini-3")) {
+        return calculateAnthropicCost(resolvedPrice, usage, externalPrice);
+      }
+      return calculateSimpleCost(resolvedPrice, usage, externalPrice);
+
+    case "openrouter": {
+      const orCost = calculateOpenRouterCost(usage);
+      if (orCost.regular > 0) return orCost;
+      return calculateOpenRouterFallbackCost(
+        resolvedPrice,
+        usage,
+        externalPrice
+      );
+    }
+
+    case "mistral":
+    case "xai":
+    case "fireworks":
+    default:
+      return calculateSimpleCost(resolvedPrice, usage, externalPrice);
+  }
+};
+
+const calculatePayDistribution = (
+  costs: CostBreakdown,
+  externalPrice?: ExternalPrice,
+  sharingLevel: "default" | "split" | "full"
+): Record<string, number> => {
+  const pay: Record<string, number> = {};
+
+  pay[nolotusId] = costs.regular;
+
+  const sharingRatios = {
+    default: 0,
+    split: 0.5,
+    full: 1,
+  };
+
+  if (externalPrice?.creatorId) {
+    const profit = Math.max(0, costs.charge - costs.regular);
+
+    if (profit > 0) {
+      switch (sharingLevel) {
+        case "split":
+          pay[externalPrice.creatorId] = profit * sharingRatios.split;
+          break;
+        case "full":
+          pay[externalPrice.creatorId] = profit;
+          break;
+        default:
+          break;
+      }
+    }
+  }
+
+  return Object.fromEntries(
+    Object.entries(pay).map(([key, value]) => [
+      key,
+      Number(value.toFixed(6)),
+    ])
+  );
+};
+
+export const calculatePrice = ({
+  modelName,
+  usage,
+  externalPrice,
+  // 不设默认值，undefined 表示 custom/未知 provider，走 zeroCostModel 路径
+  provider,
+  billingServiceTier,
+  sharingLevel = "default",
+}: CalculatePriceParams): PriceResult => {
+  // apiSource="custom" 的 agent 不一定有 provider，或 provider 不在 MODEL_MAP 里。
+  // 此时平台不承担 API 成本（由 agent owner 自费），但 agent 可能对外设有 inputPrice/outputPrice，
+  // 仍需走 externalPrice 计费路径向用户收费并分成给创作者。
+  // 用零成本的虚拟 model 代替，让 calculateBasicCost 正常走 externalPrice 分支。
+  let model: ReturnType<typeof getModelConfig>;
+  try {
+    model = getModelConfig(provider, modelName);
+  } catch {
+    const zeroCostModel: Model = {
+      name: modelName,
+      hasVision: false,
+      price: { input: 0, output: 0 },
+    };
+    const costs = calculateBasicCost(
+      zeroCostModel,
+      usage,
+      "custom",
+      externalPrice,
+      billingServiceTier
+    );
+    const pay = calculatePayDistribution(costs, externalPrice, sharingLevel);
+    return { cost: Number(costs.charge.toFixed(6)), pay };
+  }
+
+  const costs = calculateBasicCost(
+    model,
+    usage,
+    provider,
+    externalPrice,
+    billingServiceTier
+  );
+  const openaiBuiltInImageSurcharge = resolveOpenAIBuiltInImageSurcharge(
+    model,
+    usage,
+    provider
+  );
+  if (openaiBuiltInImageSurcharge > 0) {
+    costs.regular += openaiBuiltInImageSurcharge;
+    costs.charge += openaiBuiltInImageSurcharge;
+  }
+  const pay = calculatePayDistribution(costs, externalPrice, sharingLevel);
+
+  return {
+    cost: Number(costs.charge.toFixed(6)),
+    pay,
+  };
+};

package/ai/token/db.ts

@@ -0,0 +1,98 @@
+import { createTokenStatsKey } from "database/keys";
+import { TokenUsageData } from "./types";
+import { ulid } from "ulid";
+import { createInitialDayStats } from "./stats";
+import { format } from "date-fns";
+import { pino } from "pino";
+import { DataType } from "create/types";
+import { read, write } from "database/dbSlice";
+
+interface TokenCount {
+  input: number;
+  output: number;
+}
+
+interface ModelStats {
+  count: number;
+  tokens: TokenCount;
+  cost: number;
+}
+
+interface DayStats {
+  userId: string;
+  period: "day";
+  timeKey: string;
+  total: { count: number; tokens: TokenCount; cost: number };
+  models: Record<string, ModelStats>;
+  providers: Record<string, ModelStats>;
+}
+
+const logger = pino({ name: "token-db" });
+
+const updateStats = (
+  data: TokenUsageData,
+  stats: ModelStats = { count: 0, tokens: { input: 0, output: 0 }, cost: 0 }
+) => ({
+  count: stats.count + 1,
+  tokens: {
+    input: stats.tokens.input + data.input_tokens,
+    output: stats.tokens.output + data.output_tokens,
+  },
+  cost: stats.cost + data.cost,
+});
+
+export const saveTokenUsage = async (data: TokenUsageData, thunkApi) => {
+  try {
+    const dateKey = format(Date.now(), "yyyy-MM-dd");
+    const key = createTokenStatsKey(data.userId, dateKey);
+
+    // Get current stats or create initial stats
+    const currentStats = await thunkApi.dispatch(read({
+      dbKey: key
+    })).unwrap();
+    const stats = currentStats?.total
+      ? currentStats
+      : createInitialDayStats(data.userId, dateKey);
+
+    // Update stats with new data
+    const cleanModels = { ...stats.models };
+    ["unknown", "undefined"].forEach((key) => delete cleanModels[key]);
+
+    const modelName = data.model || "unknown";
+    const providerName = data.provider || "unknown";
+
+    const updatedStats = {
+      ...stats,
+      total: updateStats(data, stats.total),
+      models: {
+        ...cleanModels,
+        [modelName]: updateStats(data, cleanModels[modelName]),
+      },
+      providers: {
+        ...stats.providers,
+        [providerName]: updateStats(data, stats.providers[providerName]),
+      },
+    };
+
+    logger.info({ updatedStats }, "Updated day stats");
+
+    // Save updated stats
+    await thunkApi.dispatch(
+      write({
+        data: { ...updatedStats, id: key, type: DataType.TOKEN },
+        customKey: key,
+      })
+    );
+
+    logger.info({ currentStats, updatedStats }, "Saved day stats");
+
+    return {
+      success: true,
+      id: ulid(Date.now()),
+      record: updatedStats,
+    };
+  } catch (error) {
+    logger.error({ error }, "Failed to save token usage");
+    throw error;
+  }
+};