tabby-ai-assistant 1.0.12 → 1.0.15

package/src/utils/cost.utils.ts

@@ -0,0 +1,249 @@
+/**
+ * API成本计算工具类
+ * 提供各AI提供商的API调用成本估算功能
+ */
+
+/**
+ * AI提供商类型
+ */
+export type AIProvider = 'openai' | 'anthropic' | 'minimax' | 'glm' | 'openai-compatible';
+
+/**
+ * 模型定价信息
+ */
+export interface ModelPricing {
+    provider: AIProvider;
+    model: string;
+    inputPricePerMillion: number;  // 每百万输入token的价格（美元）
+    outputPricePerMillion: number; // 每百万输出token的价格（美元）
+}
+
+/**
+ * Token使用信息
+ */
+export interface TokenUsage {
+    inputTokens: number;
+    outputTokens: number;
+}
+
+/**
+ * 成本计算结果
+ */
+export interface CostResult {
+    inputCost: number;       // 输入成本（美元）
+    outputCost: number;      // 输出成本（美元）
+    totalCost: number;       // 总成本（美元）
+    inputPricePerMillion: number;
+    outputPricePerMillion: number;
+}
+
+/**
+ * 默认模型定价表（2024年最新价格）
+ */
+const DEFAULT_PRICING: ModelPricing[] = [
+    // OpenAI
+    { provider: 'openai', model: 'gpt-4', inputPricePerMillion: 30, outputPricePerMillion: 60 },
+    { provider: 'openai', model: 'gpt-4-turbo', inputPricePerMillion: 10, outputPricePerMillion: 30 },
+    { provider: 'openai', model: 'gpt-4o', inputPricePerMillion: 5, outputPricePerMillion: 15 },
+    { provider: 'openai', model: 'gpt-3.5-turbo', inputPricePerMillion: 0.5, outputPricePerMillion: 1.5 },
+
+    // Anthropic
+    { provider: 'anthropic', model: 'claude-3-5-sonnet-20241022', inputPricePerMillion: 3, outputPricePerMillion: 15 },
+    { provider: 'anthropic', model: 'claude-3-opus-20240229', inputPricePerMillion: 15, outputPricePerMillion: 75 },
+    { provider: 'anthropic', model: 'claude-3-haiku-20240307', inputPricePerMillion: 0.25, outputPricePerMillion: 1.25 },
+
+    // Minimax (智谱AI)
+    { provider: 'minimax', model: 'abab6.5s-chat', inputPricePerMillion: 0.3, outputPricePerMillion: 0.3 },
+    { provider: 'minimax', model: 'abab6.5-chat', inputPricePerMillion: 0.5, outputPricePerMillion: 0.5 },
+    { provider: 'minimax', model: 'abab5.5-chat', inputPricePerMillion: 1, outputPricePerMillion: 1 },
+
+    // GLM (智谱AI)
+    { provider: 'glm', model: 'glm-4', inputPricePerMillion: 0.5, outputPricePerMillion: 1.5 },
+    { provider: 'glm', model: 'glm-4v', inputPricePerMillion: 0.5, outputPricePerMillion: 1.5 },
+    { provider: 'glm', model: 'glm-3-turbo', inputPricePerMillion: 0.1, outputPricePerMillion: 0.1 },
+];
+
+// 自定义定价表（可扩展）
+let customPricing: ModelPricing[] = [];
+
+/**
+ * 设置自定义模型定价
+ */
+export function setCustomPricing(pricing: ModelPricing[]): void {
+    customPricing = [...pricing];
+}
+
+/**
+ * 获取模型定价信息
+ */
+export function getModelPricing(provider: AIProvider, model: string): ModelPricing | undefined {
+    // 首先查找自定义定价
+    const custom = customPricing.find(p => p.provider === provider && p.model === model);
+    if (custom) {
+        return custom;
+    }
+
+    // 然后查找默认定价
+    const defaultPricing = DEFAULT_PRICING.find(p => p.provider === provider && p.model === model);
+
+    if (defaultPricing) {
+        return defaultPricing;
+    }
+
+    // 返回该提供商的通用定价
+    return getDefaultPricingForProvider(provider);
+}
+
+/**
+ * 获取提供商的默认定价
+ */
+function getDefaultPricingForProvider(provider: AIProvider): ModelPricing | undefined {
+    const providerDefaults: Record<AIProvider, Partial<ModelPricing>> = {
+        'openai': { inputPricePerMillion: 5, outputPricePerMillion: 15 },
+        'anthropic': { inputPricePerMillion: 3, outputPricePerMillion: 15 },
+        'minimax': { inputPricePerMillion: 0.5, outputPricePerMillion: 0.5 },
+        'glm': { inputPricePerMillion: 0.5, outputPricePerMillion: 1 },
+        'openai-compatible': { inputPricePerMillion: 1, outputPricePerMillion: 2 }
+    };
+
+    const defaults = providerDefaults[provider];
+    if (defaults) {
+        return {
+            provider,
+            model: 'default',
+            inputPricePerMillion: defaults.inputPricePerMillion ?? 1,
+            outputPricePerMillion: defaults.outputPricePerMillion ?? 2
+        };
+    }
+
+    return undefined;
+}
+
+/**
+ * 计算API调用成本
+ * @param provider AI提供商
+ * @param model 模型名称
+ * @param usage Token使用情况
+ * @returns 成本计算结果
+ */
+export function calculateCost(
+    provider: AIProvider,
+    model: string,
+    usage: TokenUsage
+): CostResult {
+    const pricing = getModelPricing(provider, model);
+
+    if (!pricing) {
+        // 未知提供商，返回零成本
+        return {
+            inputCost: 0,
+            outputCost: 0,
+            totalCost: 0,
+            inputPricePerMillion: 0,
+            outputPricePerMillion: 0
+        };
+    }
+
+    const inputCost = (usage.inputTokens / 1000000) * pricing.inputPricePerMillion;
+    const outputCost = (usage.outputTokens / 1000000) * pricing.outputPricePerMillion;
+
+    return {
+        inputCost: Math.round(inputCost * 1000000) / 1000000, // 保留6位小数
+        outputCost: Math.round(outputCost * 1000000) / 1000000,
+        totalCost: Math.round((inputCost + outputCost) * 1000000) / 1000000,
+        inputPricePerMillion: pricing.inputPricePerMillion,
+        outputPricePerMillion: pricing.outputPricePerMillion
+    };
+}
+
+/**
+ * 计算摘要生成成本
+ */
+export function calculateSummaryCost(
+    provider: AIProvider,
+    model: string,
+    originalMessageCount: number,
+    tokensUsed: number
+): CostResult {
+    // 摘要生成主要是输入成本
+    const pricing = getModelPricing(provider, model);
+
+    if (!pricing) {
+        return {
+            inputCost: 0,
+            outputCost: 0,
+            totalCost: 0,
+            inputPricePerMillion: 0,
+            outputPricePerMillion: 0
+        };
+    }
+
+    // 估算摘要的输入和输出token（假设输出占输入的5%）
+    const estimatedInputTokens = tokensUsed;
+    const estimatedOutputTokens = Math.floor(tokensUsed * 0.05);
+
+    return calculateCost(provider, model, {
+        inputTokens: estimatedInputTokens,
+        outputTokens: estimatedOutputTokens
+    });
+}
+
+/**
+ * 格式化成本为可读字符串
+ */
+export function formatCost(cost: number): string {
+    if (cost < 0.001) {
+        return `$${(cost * 1000000).toFixed(2)}`;
+    } else if (cost < 1) {
+        return `$${cost.toFixed(4)}`;
+    } else {
+        return `$${cost.toFixed(2)}`;
+    }
+}
+
+/**
+ * 格式化成本详细信息
+ */
+export function formatCostDetail(result: CostResult): string {
+    const parts: string[] = [];
+
+    if (result.inputCost > 0) {
+        parts.push(`输入: ${formatCost(result.inputCost)}`);
+    }
+    if (result.outputCost > 0) {
+        parts.push(`输出: ${formatCost(result.outputCost)}`);
+    }
+
+    return parts.join(', ') + ` (总计: ${formatCost(result.totalCost)})`;
+}
+
+/**
+ * 估算消息的Token数量
+ */
+export function estimateTokenCount(text: string): number {
+    // 粗略估算：1个Token约等于4个字符（英文）
+    // 中文：1个Token约等于1.5个字符
+    const chineseCharCount = (text.match(/[\u4e00-\u9fa5]/g) || []).length;
+    const englishCharCount = text.length - chineseCharCount;
+
+    return Math.ceil(chineseCharCount / 1.5 + englishCharCount / 4);
+}
+
+/**
+ * 计算批量请求的总成本
+ */
+export function calculateBatchCost(
+    provider: AIProvider,
+    model: string,
+    requests: TokenUsage[]
+): CostResult {
+    const totalUsage = requests.reduce(
+        (acc, usage) => ({
+            inputTokens: acc.inputTokens + usage.inputTokens,
+            outputTokens: acc.outputTokens + usage.outputTokens
+        }),
+        { inputTokens: 0, outputTokens: 0 }
+    );
+
+    return calculateCost(provider, model, totalUsage);
+}

package/src/utils/validation.utils.ts

@@ -121,8 +121,7 @@ export function validateModel(model: string, _provider: string): { valid: boolea
         return { valid: false, error: '模型名称包含非法字符' };
     }
 
-    // TODO: 根据提供商类型进行特定验证
-    // _provider 参数保留用于未来扩展
+    // 注意：更完整的提供商特定验证请使用 validateProviderModel()
 
     return { valid: true };
 }
@@ -312,3 +311,308 @@ export function validateFilePath(path: string): { valid: boolean; error?: string
 
     return { valid: true };
 }
+
+// ==================== AI提供商特定验证 ====================
+
+/**
+ * AI提供商类型
+ */
+export type AIProviderType = 'openai' | 'anthropic' | 'minimax' | 'glm' | 'openai-compatible' | 'ollama' | 'vllm';
+
+/**
+ * OpenAI模型列表
+ */
+export const OPENAI_MODELS = [
+    'gpt-4',
+    'gpt-4-turbo',
+    'gpt-4o',
+    'gpt-4o-mini',
+    'gpt-3.5-turbo',
+    'gpt-3.5-turbo-16k'
+];
+
+/**
+ * Anthropic模型列表
+ */
+export const ANTHROPIC_MODELS = [
+    'claude-3-5-sonnet-20241022',
+    'claude-3-5-sonnet-20240620',
+    'claude-3-opus-20240229',
+    'claude-3-haiku-20240307'
+];
+
+/**
+ * GLM/智谱模型列表
+ */
+export const GLM_MODELS = [
+    'glm-4',
+    'glm-4-plus',
+    'glm-4v',
+    'glm-3-turbo'
+];
+
+/**
+ * Minimax模型列表
+ */
+export const MINIMAX_MODELS = [
+    'MiniMax-M2',
+    'MiniMax-M2-16k',
+    'abab6.5s-chat',
+    'abab6.5-chat',
+    'abab5.5-chat'
+];
+
+/**
+ * 验证AI提供商配置
+ */
+export function validateProviderConfig(
+    provider: AIProviderType,
+    config: {
+        apiKey?: string;
+        baseURL?: string;
+        model?: string;
+    }
+): { valid: boolean; errors?: string[]; warnings?: string[] } {
+    const errors: string[] = [];
+    const warnings: string[] = [];
+
+    // 验证API密钥
+    if (!config.apiKey || config.apiKey.trim().length === 0) {
+        errors.push('API密钥不能为空');
+    } else {
+        const keyValidation = validateProviderApiKey(provider, config.apiKey);
+        if (!keyValidation.valid) {
+            errors.push(keyValidation.error || 'API密钥格式无效');
+        }
+    }
+
+    // 验证模型
+    if (config.model) {
+        const modelValidation = validateProviderModel(provider, config.model);
+        if (!modelValidation.valid) {
+            errors.push(modelValidation.error || '模型名称无效');
+        } else if (modelValidation.warning) {
+            warnings.push(modelValidation.warning);
+        }
+    } else {
+        warnings.push(`未指定模型，${provider}将使用默认模型`);
+    }
+
+    // 验证基础URL（对于需要自定义URL的提供商）
+    if (needsCustomBaseURL(provider)) {
+        if (!config.baseURL || config.baseURL.trim().length === 0) {
+            warnings.push(`未指定基础URL，将使用${provider}的默认端点`);
+        } else {
+            const urlValidation = validateUrl(config.baseURL);
+            if (!urlValidation.valid) {
+                errors.push(`基础URL无效: ${urlValidation.error}`);
+            }
+        }
+    }
+
+    return {
+        valid: errors.length === 0,
+        errors: errors.length > 0 ? errors : undefined,
+        warnings: warnings.length > 0 ? warnings : undefined
+    };
+}
+
+/**
+ * 验证提供商API密钥格式
+ */
+export function validateProviderApiKey(
+    provider: AIProviderType,
+    apiKey: string
+): { valid: boolean; error?: string } {
+    const trimmedKey = apiKey.trim();
+
+    switch (provider) {
+        case 'openai':
+            if (!trimmedKey.startsWith('sk-')) {
+                return { valid: false, error: 'OpenAI API密钥应以 sk- 开头' };
+            }
+            if (trimmedKey.length < 50) {
+                return { valid: false, error: 'OpenAI API密钥长度不足' };
+            }
+            break;
+
+        case 'anthropic':
+            if (!trimmedKey.startsWith('sk-ant-')) {
+                return { valid: false, error: 'Anthropic API密钥应以 sk-ant- 开头' };
+            }
+            if (trimmedKey.length < 50) {
+                return { valid: false, error: 'Anthropic API密钥长度不足' };
+            }
+            break;
+
+        case 'minimax':
+            if (trimmedKey.length < 20) {
+                return { valid: false, error: 'Minimax API密钥长度不足' };
+            }
+            // Minimax密钥通常以sk-开头
+            if (!trimmedKey.startsWith('sk-')) {
+                return { valid: false, error: 'Minimax API密钥应以 sk- 开头' };
+            }
+            break;
+
+        case 'glm':
+            if (trimmedKey.length < 20) {
+                return { valid: false, error: 'GLM API密钥长度不足' };
+            }
+            break;
+
+        case 'openai-compatible':
+            // 兼容模式不验证具体格式
+            if (trimmedKey.length < 10) {
+                return { valid: false, error: 'API密钥长度不足' };
+            }
+            break;
+
+        case 'ollama':
+            // Ollama本地服务通常不需要API密钥
+            break;
+
+        case 'vllm':
+            // vLLM可能有basic auth或无认证
+            break;
+    }
+
+    return { valid: true };
+}
+
+/**
+ * 验证提供商模型名称
+ */
+export function validateProviderModel(
+    provider: AIProviderType,
+    model: string
+): { valid: boolean; error?: string; warning?: string } {
+    const trimmedModel = model.trim();
+
+    if (!trimmedModel) {
+        return { valid: false, error: '模型名称不能为空' };
+    }
+
+    // 检查模型名称格式
+    if (!/^[a-zA-Z0-9._-/]+$/.test(trimmedModel)) {
+        return { valid: false, error: '模型名称包含非法字符' };
+    }
+
+    // 检查是否在已知模型列表中
+    const isKnownModel = isKnownModelForProvider(provider, trimmedModel);
+    if (!isKnownModel) {
+        return {
+            valid: true,
+            warning: `模型 "${trimmedModel}" 不在${provider}的官方模型列表中`
+        };
+    }
+
+    return { valid: true };
+}
+
+/**
+ * 检查模型是否在提供商的已知模型列表中
+ */
+function isKnownModelForProvider(provider: AIProviderType, model: string): boolean {
+    const normalizedModel = model.toLowerCase();
+
+    switch (provider) {
+        case 'openai':
+            return OPENAI_MODELS.some(m => m.toLowerCase() === normalizedModel);
+
+        case 'anthropic':
+            return ANTHROPIC_MODELS.some(m => m.toLowerCase() === normalizedModel);
+
+        case 'glm':
+            return GLM_MODELS.some(m => m.toLowerCase() === normalizedModel);
+
+        case 'minimax':
+            return MINIMAX_MODELS.some(m => m.toLowerCase() === normalizedModel);
+
+        case 'openai-compatible':
+        case 'ollama':
+        case 'vllm':
+            // 这些提供商支持自定义模型名称
+            return true;
+
+        default:
+            return true;
+    }
+}
+
+/**
+ * 检查提供商是否需要自定义基础URL
+ */
+export function needsCustomBaseURL(provider: AIProviderType): boolean {
+    return ['openai-compatible', 'ollama', 'vllm'].includes(provider);
+}
+
+/**
+ * 获取提供商的默认基础URL
+ */
+export function getProviderDefaultBaseURL(provider: AIProviderType): string {
+    switch (provider) {
+        case 'openai':
+            return 'https://api.openai.com/v1';
+
+        case 'anthropic':
+            return 'https://api.anthropic.com';
+
+        case 'minimax':
+            return 'https://api.minimax.chat';
+
+        case 'glm':
+            return 'https://open.bigmodel.cn/api/paas/v4';
+
+        case 'openai-compatible':
+            return '';
+
+        case 'ollama':
+            return 'http://localhost:11434';
+
+        case 'vllm':
+            return 'http://localhost:8000';
+
+        default:
+            return '';
+    }
+}
+
+/**
+ * 验证本地服务连接（用于Ollama、vLLM等本地提供商）
+ */
+export async function validateLocalServiceConnection(
+    baseURL: string,
+    timeout: number = 5000
+): Promise<{ valid: boolean; error?: string; latency?: number }> {
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), timeout);
+
+    try {
+        const start = Date.now();
+        const response = await fetch(`${baseURL}/models`, {
+            method: 'GET',
+            signal: controller.signal
+        });
+        const latency = Date.now() - start;
+
+        clearTimeout(timeoutId);
+
+        if (response.ok) {
+            return { valid: true, latency };
+        }
+
+        return { valid: false, error: `服务返回状态码 ${response.status}` };
+    } catch (error) {
+        clearTimeout(timeoutId);
+
+        if (error instanceof Error) {
+            if (error.name === 'AbortError') {
+                return { valid: false, error: '连接超时' };
+            }
+            return { valid: false, error: error.message };
+        }
+
+        return { valid: false, error: '无法连接到服务' };
+    }
+}

package/dist/index.js.LICENSE.txt

@@ -1,18 +0,0 @@
-/*! Axios v1.13.2 Copyright (c) 2025 Matt Zabriskie and contributors */
-
-/** @preserve
-	 * Counter block mode compatible with  Dr Brian Gladman fileenc.c
-	 * derived from CryptoJS.mode.CTR
-	 * Jan Hruby jhruby.web@gmail.com
-	 */
-
-/** @preserve
-	(c) 2012 by Cédric Mesnil. All rights reserved.
-
-	Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
-
-	    - Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
-	    - Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
-
-	THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-	*/