nolo-cli 0.1.39 → 0.1.41

package/ai/llm/getModelContextWindow.ts

@@ -0,0 +1,93 @@
+// ai/llm/getModelContextWindow.ts
+
+/**
+ * 获取指定模型的 Context Window 大小。
+ * 
+ * 复用 ALL_MODELS 数据源，保持与模型选择器的一致性。
+ */
+
+import { ALL_MODELS, type ModelWithProvider } from "./models";
+
+// 默认 Context Window（用于未知模型）
+const DEFAULT_CONTEXT_WINDOW = 128000;
+const QWEN_3_6_CONTEXT_WINDOW = 262144;
+
+// 缓存模型映射表
+let modelMap: Map<string, ModelWithProvider> | null = null;
+
+const getModelMap = (): Map<string, ModelWithProvider> => {
+    if (!modelMap) {
+        modelMap = new Map();
+        for (const model of ALL_MODELS) {
+            modelMap.set(model.name, model);
+            // 也按 displayName 索引（如果有）
+            if (model.displayName) {
+                modelMap.set(model.displayName.toLowerCase(), model);
+            }
+        }
+    }
+    return modelMap;
+};
+
+/**
+ * 根据模型名称获取 Context Window 大小
+ * @param modelName 模型名称（如 "gpt-4o", "claude-3-5-sonnet"）
+ * @returns Context Window 大小（tokens）
+ */
+export const getModelContextWindow = (modelName: string): number => {
+  if (!modelName) return DEFAULT_CONTEXT_WINDOW;
+
+  const normalizedName = modelName.toLowerCase();
+  const map = getModelMap();
+  const model =
+    map.get(modelName) ||
+    map.get(normalizedName);
+
+    if (model?.contextWindow) {
+        return typeof model.contextWindow === "number"
+            ? model.contextWindow
+            : DEFAULT_CONTEXT_WINDOW;
+    }
+
+    if (
+        normalizedName.includes("qwen3.6") ||
+        normalizedName.includes("qwen3p6")
+    ) {
+        return QWEN_3_6_CONTEXT_WINDOW;
+    }
+
+    return DEFAULT_CONTEXT_WINDOW;
+};
+
+/**
+ * 获取模型的完整信息
+ */
+export const getModelInfo = (modelName: string): ModelWithProvider | null => {
+  if (!modelName) return null;
+
+  const map = getModelMap();
+  return (
+    map.get(modelName) ||
+    map.get(modelName.toLowerCase()) ||
+    null
+  );
+};
+
+/**
+ * 获取模型的 Max Output Tokens
+ */
+export const getModelMaxOutputTokens = (modelName: string): number => {
+    const model = getModelInfo(modelName);
+    if (model?.maxOutputTokens && typeof model.maxOutputTokens === "number") {
+        return model.maxOutputTokens;
+    }
+    return 4096; // 保守默认值
+};
+
+/**
+ * 获取模型是否支持视觉
+ */
+export const getModelHasVision = (modelName: string): boolean => {
+    const model = getModelInfo(modelName);
+    return model?.hasVision ?? false;
+};

package/ai/llm/getPricing.ts

@@ -0,0 +1,252 @@
+import type { Agent } from "../../app/types";
+import { getModelConfig } from "../llm/providers";
+import {
+  isOpenAIFlexPricedModel,
+  resolveOpenAIServiceTier,
+} from "../../integrations/openai/flexTier";
+import { getPublicImageAgentDefaultProfile } from "../agent/utils/publicImageAgentMode";
+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 hasExplicitAgentPricing = (config: any): boolean =>
+  [config?.inputPrice, config?.outputPrice].some(
+    (value) =>
+      typeof value === "number" && Number.isFinite(value) && value > 0
+  );
+
+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;
+      labelKey?: "defaultImageProfileEstimate";
+      profileLabel?: string;
+    }
+  | { 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"
+    | "imageModel"
+    | "provider"
+    | "imageConfig"
+    | "imageWorkflow"
+  >
+): AgentPriceHint | null => {
+  const imagePriceModel = agent.imageModel ?? agent.model;
+
+  if (agent.imageConfig?.enabled && agent.provider && imagePriceModel) {
+    try {
+      const model = getModelConfig(agent.provider as any, imagePriceModel);
+      const mode = agent.imageWorkflow;
+      
+      if (mode === "generate") {
+        // For generator agents, always use the default profile price
+        const defaultProfile = getPublicImageAgentDefaultProfile("generate");
+        const pricePerImage = getApproxPricePerImage(model, "1K");
+        if (typeof pricePerImage === "number") {
+          return {
+            type: "per_image",
+            amount: pricePerImage,
+            labelKey: "defaultImageProfileEstimate",
+            profileLabel: `${defaultProfile.quality} · ${defaultProfile.size}`,
+          };
+        }
+      } else {
+        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 formatModelCostPerMillion = (
+  value?: number,
+  creditsUnit = "积分"
+): string => {
+  if (typeof value !== "number" || !Number.isFinite(value)) return "未知";
+  if (value === 0) return "免费";
+  const decimals = Math.abs(value) < 1 ? 3 : 2;
+  const amount = value
+    .toFixed(decimals)
+    .replace(/\.0+$/, "")
+    .replace(/(\.\d*?[1-9])0+$/, "$1");
+  return `${amount} ${creditsUnit}`;
+};
+
+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/imagePricing.ts

@@ -0,0 +1,66 @@
+import type { Agent } from "../../app/types";
+import { findModelConfig } from "./providers";
+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;
+};
+
+const OPENAI_IMAGE_FALLBACK_MODEL = "gpt-image-2";
+
+export const getApproxAgentPricePerImage = (
+  agent: Pick<Agent, "provider" | "model" | "imageConfig">
+): number | undefined => {
+  if (!agent.imageConfig?.enabled || !agent.provider || !agent.model) {
+    return undefined;
+  }
+
+  const requestedSize = agent.imageConfig.imageSize as ImageSizeKey | undefined;
+  const directModel = findModelConfig(agent.provider, agent.model);
+  const directPrice = getApproxPricePerImage(directModel, requestedSize);
+  if (typeof directPrice === "number") {
+    return directPrice;
+  }
+
+  if (String(agent.provider).toLowerCase() !== "openai") {
+    return undefined;
+  }
+
+  const fallbackModel = findModelConfig("openai", OPENAI_IMAGE_FALLBACK_MODEL);
+  return getApproxPricePerImage(fallbackModel, requestedSize);
+};

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/kimi.ts

@@ -0,0 +1,23 @@
+export const FIREWORKS_KIMI_LATEST_MODEL = "accounts/fireworks/models/kimi-latest";
+export const FIREWORKS_KIMI_CURRENT_MODEL = "accounts/fireworks/models/kimi-k2p6";
+export const DEEPINFRA_KIMI_FALLBACK_MODEL = "moonshotai/Kimi-K2.6";
+export const OPENROUTER_KIMI_FALLBACK_MODEL = "moonshotai/kimi-k2.6";
+export const KIMI_PLATFORM_FALLBACK_STATUSES = [401, 402, 429, 500, 502, 503, 504];
+
+export const isFireworksKimiModel = (model?: string | null): boolean =>
+  model === FIREWORKS_KIMI_LATEST_MODEL || model === FIREWORKS_KIMI_CURRENT_MODEL;
+
+export const isDeepInfraKimiModel = (model?: string | null): boolean =>
+  model === DEEPINFRA_KIMI_FALLBACK_MODEL;
+
+export const resolveFireworksKimiModel = (model?: string | null): string => {
+  if (model === FIREWORKS_KIMI_LATEST_MODEL) {
+    return FIREWORKS_KIMI_CURRENT_MODEL;
+  }
+  return model ?? "";
+};
+
+export const shouldHideKimiAliasFromPricing = (
+  provider?: string | null,
+  model?: string | null
+): boolean => provider === "fireworks" && model === FIREWORKS_KIMI_LATEST_MODEL;

package/ai/llm/mimo.ts

@@ -0,0 +1,99 @@
+import type { Model } from "./types";
+
+// 备注: 因为是中文(国内服务直接人民币计价)，所以不用乘以 7
+const MIMO_V25_PRO_PRICE = {
+  input: 3,
+  output: 6,
+  cachingRead: 0.025,
+  cachingWrite: 0,
+};
+
+const MIMO_V25_PRICE = {
+  input: 1,
+  output: 2,
+  cachingRead: 0.02,
+  cachingWrite: 0,
+};
+
+const MIMO_LEGACY_PRICE = {
+  input: 7,
+  output: 21,
+  cachingRead: 1.4,
+  cachingWrite: 0,
+};
+
+const MIMO_TTS_FREE = {
+  input: 0,
+  output: 0,
+  cachingRead: 0,
+  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 => {
+    let price = MIMO_LEGACY_PRICE;
+    if (model.name === "mimo-v2.5-pro") price = MIMO_V25_PRO_PRICE;
+    else if (model.name === "mimo-v2.5") price = MIMO_V25_PRICE;
+    
+    return {
+      ...model,
+      hasVision: true,
+      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: model.name.includes("v2.5") ? MIMO_TTS_FREE : MIMO_LEGACY_PRICE,
+    maxOutputTokens: 32768,
+    contextWindow: 32768,
+    supportsTool: false,
+    supportsReasoningEffort: false,
+    description: "MiMo platform TTS model.",
+  })),
+];

package/ai/llm/mistral.ts

@@ -0,0 +1,21 @@
+// 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: {
+            input: 0.1 * 7,
+            output: 0.1 * 7,
+        },
+        maxOutputTokens: 262144,
+        contextWindow: 262144, // 256k ≈ 256 * 1024
+        supportsTool: true,
+        // 如果后面确认支持 reasoning effort，可以加上：
+        // supportsReasoningEffort: true,
+    },
+];

package/ai/llm/models.ts

@@ -0,0 +1,45 @@
+// ai/llm/models.ts
+
+import type { Model } from "./types";
+
+// 导入所有提供商的模型数据
+import { deepSeekModels } from "../../integrations/deepseek/models";
+import { googleModels } from "../../integrations/google/models";
+import { openAIModels } from "../../integrations/openai/models";
+import { deepinfraModels } from "./deepinfra";
+import { openrouterModels } from "./openrouterModels";
+import { fireworksModels } from "./fireworks";
+import { mistralModels } from "./mistral";
+import { mimoModels } from "./mimo";
+
+/**
+ * @interface ModelWithProvider
+ * 扩展基础 Model 类型，增加了 provider 字段，用于UI显示和逻辑处理。
+ */
+export interface ModelWithProvider extends Model {
+  provider: string;
+}
+
+/**
+ * 将一组模型打上 provider 标记的纯函数，方便复用和测试
+ */
+const withProvider =
+  (provider: ModelWithProvider["provider"]) =>
+    (models: Model[]): ModelWithProvider[] =>
+      models.map((model) => ({ ...model, provider }));
+
+/**
+ * @const ALL_MODELS
+ * 聚合了所有来源的模型数据，并为每个模型附加了其提供商信息。
+ * 这是整个应用中模型选择器的唯一数据源。
+ */
+export const ALL_MODELS: ModelWithProvider[] = [
+  ...withProvider("google")(googleModels),
+  ...withProvider("openai")(openAIModels),
+  ...withProvider("openrouter")(openrouterModels),
+  ...withProvider("deepseek")(deepSeekModels),
+  ...withProvider("deepinfra")(deepinfraModels),
+  ...withProvider("fireworks")(fireworksModels),
+  ...withProvider("mistral")(mistralModels),
+  ...withProvider("mimo")(mimoModels),
+];

package/ai/llm/openrouterModels.ts

@@ -0,0 +1,6 @@
+import { ModelDefinition } from '../types';
+
+export const OPENROUTER_MODELS: ModelDefinition[] = [
+];
+
+export const openrouterModels = OPENROUTER_MODELS;