tabby-ai-assistant 1.0.4 → 1.0.6

package/src/types/ai.types.ts

@@ -26,6 +26,8 @@ export interface ChatRequest {
     temperature?: number;
     stream?: boolean;
     model?: string;
+    tools?: any[];  // 工具定义列表
+    enableTools?: boolean;  // 是否启用工具调用
 }
 
 // 聊天响应
@@ -131,3 +133,145 @@ export interface ValidationResult {
     errors?: string[];
     warnings?: string[];
 }
+
+// ============================================================================
+// 上下文工程相关类型定义
+// ============================================================================
+
+// API消息接口（支持压缩标记）
+export interface ApiMessage {
+    role: 'user' | 'assistant' | 'system';
+    content: string | ContentBlock[];
+    ts: number;  // 时间戳（毫秒）
+
+    // 压缩相关元数据
+    isSummary?: boolean;        // 是否为摘要消息
+    condenseId?: string;        // 摘要ID
+    condenseParent?: string;    // 被哪个摘要压缩
+
+    // 截断相关元数据
+    isTruncationMarker?: boolean;  // 是否为截断标记
+    truncationId?: string;         // 截断ID
+    truncationParent?: string;     // 被哪个截断隐藏
+}
+
+// 内容块类型（用于支持工具调用）
+export interface ContentBlock {
+    type: 'text' | 'tool_use' | 'tool_result';
+    text?: string;
+    id?: string;
+    name?: string;
+    input?: Record<string, any>;
+    tool_use_id?: string;
+    content?: string;
+}
+
+// Token使用统计
+export interface TokenUsage {
+    input: number;        // 输入token数
+    output: number;       // 输出token数
+    cacheRead: number;    // 缓存读取token数
+    cacheWrite: number;   // 缓存写入token数
+}
+
+// 会话状态
+export interface SessionState {
+    id: string;
+    messages: ApiMessage[];
+    tokens: TokenUsage;
+    createdAt: number;    // 创建时间戳（毫秒）
+    updatedAt: number;    // 更新时间戳（毫秒）
+    checkpoints: string[]; // 检查点ID列表
+}
+
+// 上下文管理配置
+export interface ContextConfig {
+    maxContextTokens: number;      // 最大上下文窗口大小
+    reservedOutputTokens: number;  // 输出预留token数
+    compactThreshold: number;      // 压缩触发阈值 (0-1)
+    pruneThreshold: number;        // 裁剪触发阈值 (0-1)
+    messagesToKeep: number;        // 保留的最近消息数
+    bufferPercentage: number;      // 安全缓冲区百分比
+}
+
+// 默认配置
+export const DEFAULT_CONTEXT_CONFIG: ContextConfig = {
+    maxContextTokens: 200000,
+    reservedOutputTokens: 16000,
+    compactThreshold: 0.85,
+    pruneThreshold: 0.70,
+    messagesToKeep: 3,
+    bufferPercentage: 0.10
+};
+
+// 压缩结果
+export interface CompactionResult {
+    success: boolean;
+    messages: ApiMessage[];
+    summary?: string;
+    condenseId?: string;
+    tokensSaved: number;
+    cost: number;  // API调用成本
+    error?: string;
+}
+
+// 裁剪结果
+export interface PruneResult {
+    pruned: boolean;
+    tokensSaved: number;
+    partsCompacted: number;
+}
+
+// 截断结果
+export interface TruncationResult {
+    messages: ApiMessage[];
+    truncationId: string;
+    messagesRemoved: number;
+}
+
+// 扩展现有 ChatMessage，添加压缩标记支持
+export interface ExtendedChatMessage extends ChatMessage {
+    // 压缩相关元数据（可选）
+    isSummary?: boolean;        // 是否为摘要消息
+    condenseId?: string;        // 摘要ID
+    condenseParent?: string;    // 被哪个摘要压缩
+
+    // 截断相关元数据（可选）
+    isTruncationMarker?: boolean;  // 是否为截断标记
+    truncationId?: string;         // 截断ID
+    truncationParent?: string;     // 被哪个截断隐藏
+
+    // Token使用统计（可选）
+    tokenUsage?: TokenUsage;
+}
+
+// 检查点接口
+export interface Checkpoint {
+    id: string;
+    sessionId: string;
+    messages: ApiMessage[];
+    summary: string;
+    createdAt: number;  // 时间戳（毫秒）
+    tokenUsage: TokenUsage;
+}
+
+// ============================================================================
+// 流式响应相关类型定义
+// ============================================================================
+
+// 流式事件类型
+export interface StreamEvent {
+    type: 'text_delta' | 'tool_use_start' | 'tool_use_delta' | 'tool_use_end' | 'message_end' | 'error';
+    // 文本增量
+    textDelta?: string;
+    // 工具调用（完整时才有）
+    toolCall?: {
+        id: string;
+        name: string;
+        input: any;
+    };
+    // 错误信息
+    error?: string;
+    // 最终消息（message_end 时）
+    message?: ChatMessage;
+}

package/src/types/provider.types.ts

@@ -11,7 +11,7 @@ export { ChatRequest, ChatResponse, CommandRequest, CommandResponse, ExplainRequ
 
 // 认证配置
 export interface AuthConfig {
-    type: 'apiKey' | 'bearer' | 'basic' | 'oauth';
+    type: 'apiKey' | 'bearer' | 'basic' | 'oauth' | 'none';
     credentials: Record<string, string>;
     requiresEncryption?: boolean;
 }

package/src/utils/encryption.utils.ts

@@ -26,12 +26,15 @@ export function generateIV(): string {
  * 使用AES加密字符串
  */
 export function encrypt(text: string, key: string): string {
-    if (!text || !key) {
+    if (text === null || text === undefined || key === null || key === undefined) {
         throw new Error('文本和密钥不能为空');
     }
 
     try {
-        const encrypted = CryptoJS.AES.encrypt(text, key);
+        // 确保输入是字符串
+        const strText = String(text);
+        const strKey = String(key);
+        const encrypted = CryptoJS.AES.encrypt(strText, strKey);
         return encrypted.toString();
     } catch (error) {
         throw new Error(`加密失败: ${error instanceof Error ? error.message : String(error)}`);
@@ -42,12 +45,15 @@ export function encrypt(text: string, key: string): string {
  * 使用AES解密字符串
  */
 export function decrypt(encryptedText: string, key: string): string {
-    if (!encryptedText || !key) {
+    if (encryptedText === null || encryptedText === undefined || key === null || key === undefined) {
         throw new Error('加密文本和密钥不能为空');
     }
 
     try {
-        const decrypted = CryptoJS.AES.decrypt(encryptedText, key);
+        // 确保输入是字符串
+        const strEncryptedText = String(encryptedText);
+        const strKey = String(key);
+        const decrypted = CryptoJS.AES.decrypt(strEncryptedText, strKey);
         return decrypted.toString(CryptoJS.enc.Utf8);
     } catch (error) {
         throw new Error(`解密失败: ${error instanceof Error ? error.message : String(error)}`);
@@ -151,12 +157,14 @@ export function generateToken(length: number = 32): string {
  * Base64编码
  */
 export function base64Encode(text: string): string {
-    if (!text) {
+    if (text === null || text === undefined) {
         throw new Error('文本不能为空');
     }
 
     try {
-        return CryptoJS.enc.Base64.stringify(CryptoJS.enc.Utf8.parse(text));
+        // 确保输入是字符串
+        const str = String(text);
+        return CryptoJS.enc.Base64.stringify(CryptoJS.enc.Utf8.parse(str));
     } catch (error) {
         throw new Error(`Base64编码失败: ${error instanceof Error ? error.message : String(error)}`);
     }
@@ -166,12 +174,14 @@ export function base64Encode(text: string): string {
  * Base64解码
  */
 export function base64Decode(encodedText: string): string {
-    if (!encodedText) {
+    if (encodedText === null || encodedText === undefined) {
         throw new Error('编码文本不能为空');
     }
 
     try {
-        return CryptoJS.enc.Base64.parse(encodedText).toString(CryptoJS.enc.Utf8);
+        // 确保输入是字符串
+        const str = String(encodedText);
+        return CryptoJS.enc.Base64.parse(str).toString(CryptoJS.enc.Utf8);
     } catch (error) {
         throw new Error(`Base64解码失败: ${error instanceof Error ? error.message : String(error)}`);
     }

package/tsconfig.json

@@ -6,20 +6,20 @@
         "declaration": true,
         "outDir": "./dist",
         "rootDir": "./src",
-        "strict": false,
+        "strict": true,
         "esModuleInterop": true,
         "skipLibCheck": true,
-        "forceConsistentCasingInFileNames": false,
+        "forceConsistentCasingInFileNames": true,
         "noImplicitAny": false,
-        "noImplicitReturns": false,
-        "noImplicitThis": false,
+        "noImplicitReturns": true,
+        "noImplicitThis": true,
         "noUnusedLocals": false,
         "noUnusedParameters": false,
-        "strictNullChecks": false,
-        "strictFunctionTypes": false,
-        "strictBindCallApply": false,
-        "strictPropertyInitialization": false,
-        "noFallthroughCasesInSwitch": false,
+        "strictNullChecks": true,
+        "strictFunctionTypes": true,
+        "strictBindCallApply": true,
+        "strictPropertyInitialization": true,
+        "noFallthroughCasesInSwitch": true,
         "moduleResolution": "node",
         "experimentalDecorators": true,
         "emitDecoratorMetadata": true,

package/webpack.config.js

@@ -2,14 +2,14 @@ const path = require('path');
 const webpack = require('webpack');
 
 module.exports = {
-    mode: 'production',
+    target: 'node',
+    mode: 'development',
+    devtool: 'eval-source-map',
     entry: './src/index.ts',
     output: {
         path: path.resolve(__dirname, 'dist'),
         filename: 'index.js',
-        library: 'AiAssistantModule',
         libraryTarget: 'commonjs2',
-        libraryExport: 'default',
         devtoolModuleFilenameTemplate: 'webpack-tabby-ai-assistant:///[resource-path]',
     },
     resolve: {
@@ -36,8 +36,30 @@ module.exports = {
                 exclude: /node_modules/
             },
             {
-                test: /\.(html|css|scss)$/,
-                use: ['raw-loader']
+                test: /\.html$/,
+                use: {
+                    loader: 'raw-loader',
+                    options: {
+                        esModule: false
+                    }
+                }
+            },
+            {
+                test: /\.scss$/,
+                use: [
+                    'to-string-loader',
+                    'css-loader',
+                    'sass-loader',
+                ],
+            },
+            {
+                test: /\.css$/,
+                use: {
+                    loader: 'raw-loader',
+                    options: {
+                        esModule: false
+                    }
+                }
             }
         ]
     },
@@ -53,7 +75,7 @@ module.exports = {
     ],
     plugins: [
         new webpack.DefinePlugin({
-            'process.env.NODE_ENV': JSON.stringify('production')
+            'process.env.NODE_ENV': JSON.stringify('development')
         })
     ]
 };