nolo-cli 0.1.8 → 0.1.9

package/ai/llm/getPricing.ts

@@ -0,0 +1,199 @@
+import type { Agent } from "app/types";
+import { getModelConfig } from "ai/llm/providers";
+import {
+  isOpenAIFlexPricedModel,
+  resolveOpenAIServiceTier,
+} from "integrations/openai/flexTier";
+import { getApproxPricePerImage } from "./imagePricing";
+
+interface ModelPricing {
+  inputPrice: number;
+  inputCacheHitPrice: number;
+  outputPrice: number;
+}
+
+interface Prices {
+  cybotInput: number;
+  cybotOutput: number;
+  serverInput: number;
+  serverOutput: number;
+}
+
+const MAX_OUTPUT_TOKENS = 8192; // 单次返回最大 token 数
+
+export const getModelPricing = (
+  provider: string,
+  modelName: string,
+  requestedServiceTier?: string
+): ModelPricing | null => {
+  let model;
+  try {
+    model = getModelConfig(provider as any, modelName);
+  } catch {
+    return null;
+  }
+
+  if (!model?.price) return null;
+
+  return getModelPricingForModel(provider, modelName, model, requestedServiceTier);
+};
+
+export const getModelPricingForModel = (
+  provider: string,
+  modelName: string,
+  model: { price?: { input: number; inputCacheHit?: number; output: number } | null },
+  requestedServiceTier?: string
+): ModelPricing | null => {
+  if (!model.price) return null;
+  const normalizedRequestedServiceTier =
+    requestedServiceTier === "flex" || requestedServiceTier === "default"
+      ? requestedServiceTier
+      : undefined;
+  const shouldApplyFlexDiscount =
+    provider === "openai" &&
+    (normalizedRequestedServiceTier === "flex" ||
+      resolveOpenAIServiceTier({
+        providerName: provider,
+        model: modelName,
+      }) === "flex") &&
+    !isOpenAIFlexPricedModel(modelName);
+  const priceMultiplier = shouldApplyFlexDiscount ? 0.5 : 1;
+
+  return {
+    inputPrice: model.price.input * priceMultiplier,
+    inputCacheHitPrice:
+      typeof model.price.inputCacheHit === "number"
+        ? model.price.inputCacheHit * priceMultiplier
+        : 0,
+    outputPrice: model.price.output * priceMultiplier,
+  };
+};
+
+export const getPrices = (config: any, serverPrices: any): Prices => ({
+  cybotInput: Number(config?.inputPrice ?? 0),
+  cybotOutput: Number(config?.outputPrice ?? 0),
+  serverInput: Number(serverPrices?.inputPrice ?? 0),
+  serverOutput: Number(serverPrices?.outputPrice ?? 0),
+});
+
+/**
+ * 计算最终价格：
+ * - 从所有价格中取「每百万 tokens 价格」的最大值
+ * - 再换算成单 token 价格
+ * - 再乘以最大输出 token 数
+ */
+export const getFinalPrice = (prices: Prices): number => {
+  // 1. 取出所有价格字段
+  const rawValues = Object.values(prices);
+
+  // 2. 过滤出合法数字（去掉 NaN、Infinity、null/undefined 等）
+  const validValues = rawValues.filter(
+    (value) => typeof value === "number" && Number.isFinite(value)
+  );
+
+  // 3. 没有合法值时返回 0，避免 Math.max(...[]) 抛错
+  if (validValues.length === 0) {
+    return 0;
+  }
+
+  // 4. 找到每百万 token 的最高单价
+  const maxPricePerMillion = Math.max(...validValues);
+
+  // 5. 换算成单 token 价格
+  const maxPricePerToken = maxPricePerMillion / 1_000_000;
+
+  // 6. 计算 8192 个 token 的费用
+  return maxPricePerToken * MAX_OUTPUT_TOKENS;
+};
+
+/**
+ * 每次对话估算的典型 token 用量（输入含历史上下文，输出为单次回复）
+ * 仅用于 UI 展示的估算，并非计费依据。
+ */
+const TYPICAL_INPUT_TOKENS = 500;
+const TYPICAL_OUTPUT_TOKENS = 300;
+
+export type AgentPriceHint =
+  | { type: "per_image"; amount: number }
+  | { type: "per_turn"; amount: number };
+
+export interface CompactTurnPriceDisplay {
+  amountText: string;
+  unitCount: number;
+}
+
+const trimTrailingZeros = (value: string): string => {
+  if (!value.includes(".")) return value;
+  return value.replace(/\.?0+$/, "");
+};
+
+/**
+ * 根据 agent 配置返回用户可读的价格提示：
+ * - 图像 agent（imageConfig.enabled）：从 model config 查 pricePerImage，以每张为单位
+ * - 文字 agent：用典型 token 用量估算每次对话费用
+ * - 无价格信息则返回 null
+ */
+export const getAgentPriceHint = (
+  agent: Pick<Agent, "inputPrice" | "outputPrice" | "model" | "provider" | "imageConfig">
+): AgentPriceHint | null => {
+  if (agent.imageConfig?.enabled && agent.provider && agent.model) {
+    try {
+      const model = getModelConfig(agent.provider as any, agent.model);
+      const pricePerImage = getApproxPricePerImage(
+        model,
+        agent.imageConfig?.imageSize
+      );
+      if (typeof pricePerImage === "number") {
+        return { type: "per_image", amount: pricePerImage };
+      }
+    } catch {
+      // Fall through to token-based pricing or null.
+    }
+  }
+
+  const inputPrice = agent.inputPrice ?? 0;
+  const outputPrice = agent.outputPrice ?? 0;
+  if (inputPrice === 0 && outputPrice === 0) return null;
+
+  const amount =
+    (inputPrice * TYPICAL_INPUT_TOKENS + outputPrice * TYPICAL_OUTPUT_TOKENS) /
+    1_000_000;
+  return { type: "per_turn", amount };
+};
+
+/** 将金额格式化为用户可读字符串（2 位有效数字，不使用科学计数法）*/
+export const formatPriceAmount = (amount: number): string => {
+  if (amount >= 0.1) return amount.toFixed(2);
+  if (amount >= 0.01) return amount.toFixed(3);
+  if (amount >= 0.001) return amount.toFixed(4);
+  if (amount >= 0.0001) return amount.toFixed(5);
+  return amount.toFixed(6);
+};
+
+export const formatCompactTurnPrice = (
+  amount: number
+): CompactTurnPriceDisplay => {
+  if (amount <= 0) {
+    return { amountText: "0", unitCount: 1 };
+  }
+
+  if (amount >= 0.01) {
+    return {
+      amountText: trimTrailingZeros(formatPriceAmount(amount)),
+      unitCount: 1,
+    };
+  }
+
+  const perHundred = amount * 100;
+  if (perHundred >= 0.01) {
+    return {
+      amountText: trimTrailingZeros(formatPriceAmount(perHundred)),
+      unitCount: 100,
+    };
+  }
+
+  return {
+    amountText: trimTrailingZeros(formatPriceAmount(amount * 1000)),
+    unitCount: 1000,
+  };
+};

package/ai/llm/hooks/useModelPricing.ts

@@ -0,0 +1,75 @@
+import { useState, useEffect, useRef } from "react";
+import { getModelsByProvider, getProviderByModelName } from "ai/llm/providers";
+import type { Model, ModelPrice } from "ai/llm/types";
+
+type ModelWithLegacyPricing = Partial<Model> & {
+  pricing?: ModelPrice | null;
+};
+
+export const resolveModelPrice = (
+  model?: ModelWithLegacyPricing | null
+): ModelPrice => {
+  const price = model?.price ?? model?.pricing;
+  return {
+    input: typeof price?.input === "number" ? price.input : 0,
+    output: typeof price?.output === "number" ? price.output : 0,
+    cachingWrite: price?.cachingWrite,
+    cachingRead: price?.cachingRead,
+    inputCacheHit: price?.inputCacheHit,
+  };
+};
+
+const useModelPricing = (
+  provider: string,
+  modelName: string,
+  setValue?: (name: string, value: number) => void
+) => {
+  const [models, setModels] = useState<Model[]>([]);
+  const [inputPrice, setInputPrice] = useState<number>(0);
+  const [outputPrice, setOutputPrice] = useState<number>(0);
+  const setValueRef = useRef(setValue);
+
+  useEffect(() => {
+    setValueRef.current = setValue;
+  }, [setValue]);
+
+  useEffect(() => {
+    // 若 provider 为空，从模型名反查（防止 provider 漏传导致定价缺失）
+    const resolvedProvider = provider || getProviderByModelName(modelName);
+    if (!resolvedProvider) return;
+    setModels(getModelsByProvider(resolvedProvider));
+  }, [provider, modelName]);
+
+  useEffect(() => {
+    const selectedModel = models.find((model) => model.name === modelName);
+    if (selectedModel) {
+      const price = resolveModelPrice(selectedModel as ModelWithLegacyPricing);
+      setInputPrice((current) => current === price.input ? current : price.input);
+      setOutputPrice((current) => current === price.output ? current : price.output);
+
+      if (setValueRef.current) {
+        setValueRef.current("inputPrice", price.input);
+        setValueRef.current("outputPrice", price.output);
+      }
+    }
+  }, [models, modelName]);
+
+  const updateInputPrice = (value: number) => {
+    setInputPrice(value);
+    if (setValue) setValue("inputPrice", value);
+  };
+
+  const updateOutputPrice = (value: number) => {
+    setOutputPrice(value);
+    if (setValue) setValue("outputPrice", value);
+  };
+
+  return {
+    inputPrice,
+    outputPrice,
+    setInputPrice: updateInputPrice,
+    setOutputPrice: updateOutputPrice,
+  };
+};
+
+export default useModelPricing;

package/ai/llm/imagePricing.ts

@@ -0,0 +1,40 @@
+import type { Model } from "./types";
+
+export type ImageSizeKey = "1K" | "2K" | "4K";
+
+type ImagePricingModel = Pick<
+  Model,
+  "pricePerImage" | "imageTokenPricePerMillion" | "imageOutputTokenEstimateBySize"
+>;
+
+const DEFAULT_IMAGE_SIZE: ImageSizeKey = "1K";
+
+export const getApproxPricePerImage = (
+  model: ImagePricingModel | null | undefined,
+  requestedSize?: ImageSizeKey
+): number | undefined => {
+  if (!model) return undefined;
+
+  if (typeof model.pricePerImage === "number") {
+    return model.pricePerImage;
+  }
+
+  if (typeof model.imageTokenPricePerMillion !== "number") {
+    return undefined;
+  }
+
+  const tokenEstimates = model.imageOutputTokenEstimateBySize;
+  if (!tokenEstimates) return undefined;
+
+  const resolvedSize =
+    requestedSize && typeof tokenEstimates[requestedSize] === "number"
+      ? requestedSize
+      : DEFAULT_IMAGE_SIZE;
+  const outputTokens = tokenEstimates[resolvedSize];
+
+  if (typeof outputTokens !== "number") {
+    return undefined;
+  }
+
+  return (model.imageTokenPricePerMillion * outputTokens) / 1_000_000;
+};

package/ai/llm/isResponseAPIModel.ts

@@ -0,0 +1,13 @@
+import { getModelConfig } from "./providers";
+
+export const isResponseAPIModel = (agentConfig) => {
+  if (agentConfig.provider !== "openai") return false;
+  if (agentConfig.endpointKey === "responses") return true;
+  if (!agentConfig.model) return false;
+
+  try {
+    return getModelConfig("openai", agentConfig.model).endpointKey === "responses";
+  } catch {
+    return false;
+  }
+};

package/ai/llm/mimo.ts

@@ -0,0 +1,71 @@
+import type { Model } from "./types";
+
+const MIMO_TEXT_PRICE = {
+  input: 1 * 8,
+  output: 3 * 8,
+  cachingRead: 0.2 * 8,
+  cachingWrite: 0,
+};
+
+const MIMO_TEXT_MODELS = [
+  {
+    name: "mimo-v2.5-pro",
+    displayName: "Xiaomi: MiMo V2.5 Pro",
+  },
+  {
+    name: "mimo-v2.5",
+    displayName: "Xiaomi: MiMo V2.5",
+  },
+  {
+    name: "mimo-v2-pro",
+    displayName: "Xiaomi: MiMo V2 Pro",
+  },
+  {
+    name: "mimo-v2-omni",
+    displayName: "Xiaomi: MiMo V2 Omni",
+    hasAudio: true,
+  },
+];
+
+const MIMO_TTS_MODELS = [
+  {
+    name: "mimo-v2.5-tts-voiceclone",
+    displayName: "Xiaomi: MiMo V2.5 TTS VoiceClone",
+  },
+  {
+    name: "mimo-v2.5-tts-voicedesign",
+    displayName: "Xiaomi: MiMo V2.5 TTS VoiceDesign",
+  },
+  {
+    name: "mimo-v2.5-tts",
+    displayName: "Xiaomi: MiMo V2.5 TTS",
+  },
+  {
+    name: "mimo-v2-tts",
+    displayName: "Xiaomi: MiMo V2 TTS",
+  },
+];
+
+export const mimoModels: Model[] = [
+  ...MIMO_TEXT_MODELS.map((model): Model => ({
+    ...model,
+    hasVision: true,
+    price: MIMO_TEXT_PRICE,
+    maxOutputTokens: 131072,
+    contextWindow: 1048576,
+    supportsTool: true,
+    supportsReasoningEffort: true,
+    description: "MiMo platform chat model.",
+  })),
+  ...MIMO_TTS_MODELS.map((model): Model => ({
+    ...model,
+    hasVision: false,
+    hasAudio: true,
+    price: MIMO_TEXT_PRICE,
+    maxOutputTokens: 32768,
+    contextWindow: 32768,
+    supportsTool: false,
+    supportsReasoningEffort: false,
+    description: "MiMo platform TTS model.",
+  })),
+];

package/ai/llm/mistral.ts

@@ -0,0 +1,22 @@
+// ai/llm/mistral.ts
+
+import type { Model } from "./types";
+
+export const mistralModels: Model[] = [
+    {
+        // 实际使用时建议改成你真实调用用的 model id
+        name: "devstral-2512",
+        displayName: "Mistral: Devstral 2",
+        hasVision: false,
+        price: {
+            // 单位：$/M tokens，按你给的文档：input 0.4，output 2
+            input: 0.1,
+            output: 0.1,
+        },
+        maxOutputTokens: 262144,
+        contextWindow: 262144, // 256k ≈ 256 * 1024
+        supportsTool: true,
+        // 如果后面确认支持 reasoning effort，可以加上：
+        // supportsReasoningEffort: true,
+    },
+];