tabby-ai-assistant 1.0.12 → 1.0.15

package/src/types/ai.types.ts

@@ -6,7 +6,8 @@
 export enum MessageRole {
     USER = 'user',
     ASSISTANT = 'assistant',
-    SYSTEM = 'system'
+    SYSTEM = 'system',
+    TOOL = 'tool'      // 工具结果角色（部分 AI 需要）
 }
 
 // 聊天消息
@@ -140,7 +141,7 @@ export interface ValidationResult {
 
 // API消息接口（支持压缩标记）
 export interface ApiMessage {
-    role: 'user' | 'assistant' | 'system';
+    role: 'user' | 'assistant' | 'system' | 'tool' | 'function';
     content: string | ContentBlock[];
     ts: number;  // 时间戳（毫秒）
 
@@ -148,6 +149,11 @@ export interface ApiMessage {
     isSummary?: boolean;        // 是否为摘要消息
     condenseId?: string;        // 摘要ID
     condenseParent?: string;    // 被哪个摘要压缩
+    summaryMeta?: {             // 摘要元数据
+        originalMessageCount: number;
+        tokensCost: number;
+        compressionRatio: number;
+    };
 
     // 截断相关元数据
     isTruncationMarker?: boolean;  // 是否为截断标记
@@ -245,6 +251,16 @@ export interface ExtendedChatMessage extends ChatMessage {
     tokenUsage?: TokenUsage;
 }
 
+// 压缩后的检查点数据接口
+export interface CompressedCheckpointData {
+    compressed: boolean;
+    compressionRatio: number;
+    originalSize: number;
+    compressedSize: number;
+    messages?: ApiMessage[]; // 可选，用于即时访问
+    messagesJson: string; // 压缩后的JSON字符串
+}
+
 // 检查点接口
 export interface Checkpoint {
     id: string;
@@ -253,6 +269,11 @@ export interface Checkpoint {
     summary: string;
     createdAt: number;  // 时间戳（毫秒）
     tokenUsage: TokenUsage;
+    compressedData?: CompressedCheckpointData; // 压缩数据（可选）
+
+    // 新增字段
+    tags?: string[];      // 标签列表
+    isArchived?: boolean; // 是否已归档
 }
 
 // ============================================================================
@@ -261,7 +282,7 @@ export interface Checkpoint {
 
 // 流式事件类型
 export interface StreamEvent {
-    type: 'text_delta' | 'tool_use_start' | 'tool_use_delta' | 'tool_use_end' | 'message_end' | 'error';
+    type: 'text_delta' | 'tool_use_start' | 'tool_use_delta' | 'tool_use_end' | 'tool_result' | 'tool_error' | 'message_end' | 'error';
     // 文本增量
     textDelta?: string;
     // 工具调用（完整时才有）
@@ -270,8 +291,140 @@ export interface StreamEvent {
         name: string;
         input: any;
     };
+    // 工具结果（tool_result 事件）
+    result?: {
+        tool_use_id: string;
+        content: string;
+        is_error?: boolean;
+    };
     // 错误信息
     error?: string;
     // 最终消息（message_end 时）
     message?: ChatMessage;
 }
+
+// ============================================================================
+// Agent 循环相关类型定义
+// ============================================================================
+
+// 工具调用接口
+export interface ToolCall {
+    id: string;
+    name: string;
+    input: any;
+}
+
+// 工具结果接口
+export interface ToolResult {
+    tool_use_id: string;
+    name?: string;        // 工具名称
+    content: string;
+    is_error?: boolean;
+}
+
+// Agent 事件类型
+export type AgentEventType =
+    | 'text_delta'           // 文本增量
+    | 'tool_use_start'       // 工具开始
+    | 'tool_use_end'         // 工具调用结束（收集参数）
+    | 'tool_executing'       // 工具正在执行
+    | 'tool_executed'        // 工具执行完成（带结果）
+    | 'tool_error'           // 工具执行错误
+    | 'round_start'          // 新一轮开始
+    | 'round_end'            // 一轮结束
+    | 'agent_complete'       // Agent 循环完成
+    | 'error';               // 错误
+
+// Agent 流式事件
+export interface AgentStreamEvent {
+    type: AgentEventType;
+
+    // text_delta 事件
+    textDelta?: string;
+
+    // 工具相关事件
+    toolCall?: {
+        id: string;
+        name: string;
+        input: any;
+    };
+
+    // tool_executed/tool_error 事件
+    toolResult?: {
+        tool_use_id: string;
+        content: string;
+        is_error?: boolean;
+        duration?: number;
+    };
+
+    // round_start/round_end 事件
+    round?: number;
+
+    // agent_complete 事件
+    reason?: TerminationReason;
+    totalRounds?: number;
+    terminationMessage?: string;  // 可选的终止详情消息
+
+    // error 事件
+    error?: string;
+
+    // message_end 保留
+    message?: ChatMessage;
+}
+
+// Agent 循环配置
+export interface AgentLoopConfig {
+    maxRounds?: number;           // 最大轮数，默认 15
+    timeoutMs?: number;           // 默认 120000 (2分钟)
+    repeatThreshold?: number;     // 默认 3 次
+    failureThreshold?: number;    // 默认 2 次
+    enableTaskComplete?: boolean; // 默认 true
+    onRoundStart?: (round: number) => void;
+    onRoundEnd?: (round: number) => void;
+    onAgentComplete?: (reason: string, totalRounds: number) => void;
+}
+
+// ============================================================================
+// 智能 Agent 终止相关类型定义
+// ============================================================================
+
+// 终止原因枚举
+export type TerminationReason =
+    | 'task_complete'      // AI 主动调用 task_complete 工具
+    | 'no_tools'           // 本轮无工具调用
+    | 'summarizing'        // 检测到 AI 正在总结
+    | 'repeated_tool'      // 重复调用相同工具
+    | 'high_failure_rate'  // 连续失败率过高
+    | 'timeout'            // 总时间超时
+    | 'max_rounds'         // 达到最大轮数（安全保底）
+    | 'user_cancel';       // 用户取消
+
+// Agent 状态追踪
+export interface AgentState {
+    currentRound: number;
+    startTime: number;
+    toolCallHistory: ToolCallRecord[];
+    lastAiResponse: string;
+    isActive: boolean;
+}
+
+// 工具调用记录
+export interface ToolCallRecord {
+    name: string;
+    input: any;
+    inputHash: string;  // 用于快速比较
+    success: boolean;
+    timestamp: number;
+}
+
+// 终止检测结果
+export interface TerminationResult {
+    shouldTerminate: boolean;
+    reason: TerminationReason;
+    message?: string;
+}
+
+// 扩展 ToolResult 添加任务完成标记
+export interface ExtendedToolResult extends ToolResult {
+    isTaskComplete?: boolean;  // 特殊标记：task_complete 工具调用
+}

package/src/types/provider.types.ts

@@ -2,6 +2,7 @@
  * AI提供商相关类型定义
  */
 
+import { Observable } from 'rxjs';
 import { ProviderCapability, HealthStatus, ValidationResult } from './ai.types';
 import { ChatRequest, ChatResponse, CommandRequest, CommandResponse, ExplainRequest, ExplainResponse, AnalysisRequest, AnalysisResponse } from './ai.types';
 
@@ -9,14 +10,18 @@ import { ChatRequest, ChatResponse, CommandRequest, CommandResponse, ExplainRequ
 export { ProviderCapability, HealthStatus, ValidationResult };
 export { ChatRequest, ChatResponse, CommandRequest, CommandResponse, ExplainRequest, ExplainResponse, AnalysisRequest, AnalysisResponse };
 
-// 认证配置
+// ==================== 认证配置 ====================
+
+export type AuthType = 'apiKey' | 'bearer' | 'basic' | 'oauth' | 'none';
+
 export interface AuthConfig {
-    type: 'apiKey' | 'bearer' | 'basic' | 'oauth' | 'none';
+    type: AuthType;
     credentials: Record<string, string>;
     requiresEncryption?: boolean;
 }
 
-// 提供商配置
+// ==================== 提供商配置 ====================
+
 export interface ProviderConfig {
     name: string;
     displayName: string;
@@ -32,111 +37,234 @@ export interface ProviderConfig {
     contextWindow?: number;  // 供应商上下文窗口限制
 }
 
-// 提供商信息
-export interface ProviderInfo {
-    name: string;
-    displayName: string;
-    version: string;
-    description: string;
-    capabilities: ProviderCapability[];
+// 提供商默认配置
+export interface ProviderDefaults {
+    baseURL: string;
+    model: string;
+    maxTokens: number;
+    temperature: number;
+    timeout: number;
+    retries: number;
+    contextWindow: number;
     authConfig: AuthConfig;
-    supportedModels: string[];
-    pricing?: {
-        type: 'free' | 'paid' | 'freemium';
-        currency: string;
-        unit: string;
-        costPerUnit?: number;
-    };
-    documentation?: string;
 }
 
-// 基础AI提供商接口
-export abstract class BaseAiProvider {
-    abstract readonly name: string;
-    abstract readonly displayName: string;
-    abstract readonly capabilities: ProviderCapability[];
-    abstract readonly authConfig: AuthConfig;
-
-    protected config: ProviderConfig | null = null;
-
-    /**
-     * 配置提供商
-     */
-    abstract configure(config: ProviderConfig): void;
-
-    /**
-     * 聊天功能
-     */
-    abstract chat(request: ChatRequest): Promise<ChatResponse>;
-
-    /**
-     * 生成命令
-     */
-    abstract generateCommand(request: CommandRequest): Promise<CommandResponse>;
+// 所有已知提供商及其默认配置
+export const PROVIDER_DEFAULTS: Record<string, ProviderDefaults> = {
+    openai: {
+        baseURL: 'https://api.openai.com/v1',
+        model: 'gpt-4',
+        maxTokens: 1000,
+        temperature: 0.7,
+        timeout: 30000,
+        retries: 3,
+        contextWindow: 128000,
+        authConfig: { type: 'bearer', credentials: {} }
+    },
+    anthropic: {
+        baseURL: 'https://api.anthropic.com',
+        model: 'claude-3-sonnet',
+        maxTokens: 1000,
+        temperature: 1.0,
+        timeout: 30000,
+        retries: 3,
+        contextWindow: 200000,
+        authConfig: { type: 'bearer', credentials: {} }
+    },
+    minimax: {
+        baseURL: 'https://api.minimaxi.com/anthropic',
+        model: 'MiniMax-M2',
+        maxTokens: 1000,
+        temperature: 1.0,
+        timeout: 30000,
+        retries: 3,
+        contextWindow: 128000,
+        authConfig: { type: 'bearer', credentials: {} }
+    },
+    glm: {
+        baseURL: 'https://open.bigmodel.cn/api/anthropic',
+        model: 'glm-4.6',
+        maxTokens: 1000,
+        temperature: 0.95,
+        timeout: 30000,
+        retries: 3,
+        contextWindow: 128000,
+        authConfig: { type: 'bearer', credentials: {} }
+    },
+    ollama: {
+        baseURL: 'http://localhost:11434/v1',
+        model: 'llama3.1',
+        maxTokens: 1000,
+        temperature: 0.7,
+        timeout: 30000,
+        retries: 3,
+        contextWindow: 8192,
+        authConfig: { type: 'none', credentials: {} }
+    },
+    vllm: {
+        baseURL: 'http://localhost:8000/v1',
+        model: 'meta-llama/Llama-3.1-8B',
+        maxTokens: 1000,
+        temperature: 0.7,
+        timeout: 30000,
+        retries: 3,
+        contextWindow: 8192,
+        authConfig: { type: 'bearer', credentials: {} }
+    },
+    'openai-compatible': {
+        baseURL: 'http://localhost:11434/v1',
+        model: 'gpt-3.5-turbo',
+        maxTokens: 1000,
+        temperature: 0.7,
+        timeout: 30000,
+        retries: 3,
+        contextWindow: 128000,
+        authConfig: { type: 'bearer', credentials: {} }
+    }
+};
 
+// 配置工具函数
+export namespace ProviderConfigUtils {
     /**
-     * 解释命令
+     * 使用默认值填充配置
      */
-    abstract explainCommand(request: ExplainRequest): Promise<ExplainResponse>;
+    export function fillDefaults(config: Partial<ProviderConfig>, providerName: string): ProviderConfig {
+        const defaults = PROVIDER_DEFAULTS[providerName];
+        if (!defaults) {
+            throw new Error(`Unknown provider: ${providerName}`);
+        }
 
-    /**
-     * 分析结果
-     */
-    abstract analyzeResult(request: AnalysisRequest): Promise<AnalysisResponse>;
+        return {
+            name: config.name || providerName,
+            displayName: config.displayName || providerName,
+            apiKey: config.apiKey,
+            baseURL: config.baseURL || defaults.baseURL,
+            model: config.model || defaults.model,
+            maxTokens: config.maxTokens ?? defaults.maxTokens,
+            temperature: config.temperature ?? defaults.temperature,
+            timeout: config.timeout ?? defaults.timeout,
+            retries: config.retries ?? defaults.retries,
+            authConfig: config.authConfig || defaults.authConfig,
+            enabled: config.enabled ?? true,
+            contextWindow: config.contextWindow ?? defaults.contextWindow
+        };
+    }
 
     /**
-     * 健康检查
+     * 检查配置是否完整（可用于API调用）
      */
-    abstract healthCheck(): Promise<HealthStatus>;
+    export function isConfigComplete(config: ProviderConfig): boolean {
+        return !!(
+            config.name &&
+            config.displayName &&
+            // API key 不是必需的（如本地服务）
+            (config.apiKey || config.authConfig?.type === 'none') &&
+            config.baseURL
+        );
+    }
 
     /**
-     * 验证配置
+     * 克隆配置（深拷贝，移除敏感信息）
      */
-    abstract validateConfig(): ValidationResult;
+    export function cloneConfig(config: ProviderConfig, maskApiKey = true): ProviderConfig {
+        const clone = { ...config };
+        if (maskApiKey && clone.apiKey) {
+            clone.apiKey = '***MASKED***';
+        }
+        return clone;
+    }
 
     /**
-     * 获取提供商信息
+     * 获取提供商默认配置
      */
-    getInfo(): ProviderInfo {
-        return {
-            name: this.name,
-            displayName: this.displayName,
-            version: '1.0.0',
-            description: `${this.displayName} AI Provider`,
-            capabilities: this.capabilities,
-            authConfig: this.authConfig,
-            supportedModels: this.config?.model ? [this.config.model] : []
-        };
+    export function getDefaults(providerName: string): ProviderDefaults | undefined {
+        return PROVIDER_DEFAULTS[providerName];
     }
 
     /**
-     * 检查是否支持指定能力
+     * 获取所有已知提供商名称
      */
-    supportsCapability(capability: ProviderCapability): boolean {
-        return this.capabilities.includes(capability);
+    export function getKnownProviders(): string[] {
+        return Object.keys(PROVIDER_DEFAULTS);
     }
 
     /**
-     * 获取当前配置
+     * 检查是否为已知提供商
      */
-    getConfig(): ProviderConfig | null {
-        return this.config;
+    export function isKnownProvider(name: string): boolean {
+        return name in PROVIDER_DEFAULTS;
     }
 }
 
-// 提供商管理器
+// ==================== 提供商信息 ====================
+
+export interface ProviderPricing {
+    type: 'free' | 'paid' | 'freemium';
+    currency: string;
+    unit: string;
+    costPerUnit?: number;
+}
+
+export interface ProviderInfo {
+    name: string;
+    displayName: string;
+    version: string;
+    description: string;
+    capabilities: ProviderCapability[];
+    authConfig: AuthConfig;
+    supportedModels: string[];
+    configured?: boolean;
+    lastHealthCheck?: { status: HealthStatus; timestamp: Date };
+    pricing?: ProviderPricing;
+    documentation?: string;
+    defaults?: ProviderDefaults;
+}
+
+// ==================== 基础AI提供商接口 ====================
+
+export interface IBaseAiProvider {
+    readonly name: string;
+    readonly displayName: string;
+    readonly capabilities: ProviderCapability[];
+    readonly authConfig: AuthConfig;
+
+    // 配置与状态
+    configure(config: ProviderConfig): void;
+    getConfig(): ProviderConfig | null;
+    isConfigured(): boolean;
+    isEnabled(): boolean;
+
+    // 核心功能
+    chat(request: ChatRequest): Promise<ChatResponse>;
+    chatStream(request: ChatRequest): Observable<any>;
+    generateCommand(request: CommandRequest): Promise<CommandResponse>;
+    explainCommand(request: ExplainRequest): Promise<ExplainResponse>;
+    analyzeResult(request: AnalysisRequest): Promise<AnalysisResponse>;
+
+    // 健康与验证
+    healthCheck(): Promise<HealthStatus>;
+    validateConfig(): ValidationResult;
+
+    // 信息查询
+    getInfo(): ProviderInfo;
+    supportsCapability(capability: ProviderCapability): boolean;
+}
+
+// ==================== 提供商管理器 ====================
+
 export interface ProviderManager {
-    registerProvider(provider: BaseAiProvider): void;
+    registerProvider(provider: IBaseAiProvider): void;
     unregisterProvider(name: string): void;
-    getProvider(name: string): BaseAiProvider | undefined;
-    getAllProviders(): BaseAiProvider[];
-    getActiveProvider(): BaseAiProvider | undefined;
+    getProvider(name: string): IBaseAiProvider | undefined;
+    getAllProviders(): IBaseAiProvider[];
+    getActiveProvider(): IBaseAiProvider | undefined;
     setActiveProvider(name: string): boolean;
     getProviderInfo(name: string): ProviderInfo | undefined;
     getAllProviderInfo(): ProviderInfo[];
 }
 
-// 提供商事件
+// ==================== 提供商事件 ====================
+
 export interface ProviderEvent {
     type: 'connected' | 'disconnected' | 'error' | 'config_changed' | 'health_changed';
     provider: string;
@@ -144,5 +272,8 @@ export interface ProviderEvent {
     data?: any;
 }
 
-// 提供商事件监听器
 export type ProviderEventListener = (event: ProviderEvent) => void;
+
+// ==================== 便利类型 ====================
+
+export type BaseAiProvider = IBaseAiProvider;