tabby-ai-assistant 1.0.12 → 1.0.15

package/src/services/core/toast.service.ts

@@ -1,106 +1,106 @@
-import { Injectable } from '@angular/core';
-import { Subject } from 'rxjs';
-
-export interface ToastMessage {
-    id: string;
-    type: 'success' | 'error' | 'warning' | 'info';
-    message: string;
-    duration?: number;
-}
-
-@Injectable({ providedIn: 'root' })
-export class ToastService {
-    private toastSubject = new Subject<ToastMessage>();
-    toast$ = this.toastSubject.asObservable();
-
-    show(type: ToastMessage['type'], message: string, duration = 3000): void {
-        const id = `toast-${Date.now()}`;
-
-        // 确保容器存在
-        let container = document.getElementById('ai-toast-container');
-        if (!container) {
-            container = document.createElement('div');
-            container.id = 'ai-toast-container';
-            container.className = 'ai-toast-container';
-            container.style.cssText = `
-                position: fixed;
-                bottom: 20px;
-                right: 20px;
-                z-index: 99999;
-                display: flex;
-                flex-direction: column;
-                gap: 10px;
-                pointer-events: none;
-            `;
-            document.body.appendChild(container);
-        }
-
-        // 创建 Toast 元素
-        const toast = document.createElement('div');
-        toast.id = id;
-        toast.className = `ai-toast ai-toast-${type}`;
-        toast.style.cssText = `
-            padding: 12px 16px;
-            border-radius: 8px;
-            color: white;
-            font-size: 14px;
-            display: flex;
-            align-items: center;
-            gap: 8px;
-            animation: toastSlideIn 0.3s ease;
-            cursor: pointer;
-            box-shadow: 0 4px 12px rgba(0,0,0,0.3);
-            pointer-events: auto;
-            min-width: 200px;
-            max-width: 350px;
-            ${type === 'success' ? 'background: linear-gradient(135deg, #22c55e, #16a34a);' : ''}
-            ${type === 'error' ? 'background: linear-gradient(135deg, #ef4444, #dc2626);' : ''}
-            ${type === 'warning' ? 'background: linear-gradient(135deg, #f59e0b, #d97706);' : ''}
-            ${type === 'info' ? 'background: linear-gradient(135deg, #3b82f6, #2563eb);' : ''}
-        `;
-
-        const icon = type === 'success' ? '✓' : type === 'error' ? '✗' : type === 'warning' ? '⚠' : 'ℹ';
-        toast.innerHTML = `<span style="font-size: 16px;">${icon}</span><span>${message}</span>`;
-
-        toast.onclick = () => {
-            this.removeToast(toast);
-        };
-
-        container.appendChild(toast);
-
-        // 自动消失
-        setTimeout(() => {
-            this.removeToast(toast);
-        }, duration);
-
-        // 发射事件（兼容现有订阅）
-        this.toastSubject.next({ id, type, message, duration });
-    }
-
-    private removeToast(toast: HTMLElement): void {
-        if (toast && toast.parentNode) {
-            toast.style.animation = 'toastSlideOut 0.3s ease forwards';
-            setTimeout(() => {
-                if (toast.parentNode) {
-                    toast.remove();
-                }
-            }, 300);
-        }
-    }
-
-    success(message: string, duration = 3000): void {
-        this.show('success', message, duration);
-    }
-
-    error(message: string, duration = 5000): void {
-        this.show('error', message, duration);
-    }
-
-    warning(message: string, duration = 4000): void {
-        this.show('warning', message, duration);
-    }
-
-    info(message: string, duration = 3000): void {
-        this.show('info', message, duration);
-    }
-}
+import { Injectable } from '@angular/core';
+import { Subject } from 'rxjs';
+
+export interface ToastMessage {
+    id: string;
+    type: 'success' | 'error' | 'warning' | 'info';
+    message: string;
+    duration?: number;
+}
+
+@Injectable({ providedIn: 'root' })
+export class ToastService {
+    private toastSubject = new Subject<ToastMessage>();
+    toast$ = this.toastSubject.asObservable();
+
+    show(type: ToastMessage['type'], message: string, duration = 3000): void {
+        const id = `toast-${Date.now()}`;
+
+        // 确保容器存在
+        let container = document.getElementById('ai-toast-container');
+        if (!container) {
+            container = document.createElement('div');
+            container.id = 'ai-toast-container';
+            container.className = 'ai-toast-container';
+            container.style.cssText = `
+                position: fixed;
+                bottom: 20px;
+                right: 20px;
+                z-index: 99999;
+                display: flex;
+                flex-direction: column;
+                gap: 10px;
+                pointer-events: none;
+            `;
+            document.body.appendChild(container);
+        }
+
+        // 创建 Toast 元素
+        const toast = document.createElement('div');
+        toast.id = id;
+        toast.className = `ai-toast ai-toast-${type}`;
+        toast.style.cssText = `
+            padding: 12px 16px;
+            border-radius: 8px;
+            color: white;
+            font-size: 14px;
+            display: flex;
+            align-items: center;
+            gap: 8px;
+            animation: toastSlideIn 0.3s ease;
+            cursor: pointer;
+            box-shadow: 0 4px 12px rgba(0,0,0,0.3);
+            pointer-events: auto;
+            min-width: 200px;
+            max-width: 350px;
+            ${type === 'success' ? 'background: linear-gradient(135deg, #22c55e, #16a34a);' : ''}
+            ${type === 'error' ? 'background: linear-gradient(135deg, #ef4444, #dc2626);' : ''}
+            ${type === 'warning' ? 'background: linear-gradient(135deg, #f59e0b, #d97706);' : ''}
+            ${type === 'info' ? 'background: linear-gradient(135deg, #3b82f6, #2563eb);' : ''}
+        `;
+
+        const icon = type === 'success' ? '✓' : type === 'error' ? '✗' : type === 'warning' ? '⚠' : 'ℹ';
+        toast.innerHTML = `<span style="font-size: 16px;">${icon}</span><span>${message}</span>`;
+
+        toast.onclick = () => {
+            this.removeToast(toast);
+        };
+
+        container.appendChild(toast);
+
+        // 自动消失
+        setTimeout(() => {
+            this.removeToast(toast);
+        }, duration);
+
+        // 发射事件（兼容现有订阅）
+        this.toastSubject.next({ id, type, message, duration });
+    }
+
+    private removeToast(toast: HTMLElement): void {
+        if (toast && toast.parentNode) {
+            toast.style.animation = 'toastSlideOut 0.3s ease forwards';
+            setTimeout(() => {
+                if (toast.parentNode) {
+                    toast.remove();
+                }
+            }, 300);
+        }
+    }
+
+    success(message: string, duration = 3000): void {
+        this.show('success', message, duration);
+    }
+
+    error(message: string, duration = 5000): void {
+        this.show('error', message, duration);
+    }
+
+    warning(message: string, duration = 4000): void {
+        this.show('warning', message, duration);
+    }
+
+    info(message: string, duration = 3000): void {
+        this.show('info', message, duration);
+    }
+}

package/src/services/providers/anthropic-provider.service.ts

@@ -1,9 +1,9 @@
 import { Injectable } from '@angular/core';
-import { Observable, from } from 'rxjs';
+import { Observable, Observer } from 'rxjs';
 import { Anthropic } from '@anthropic-ai/sdk';
 import { BaseAiProvider } from './base-provider.service';
-import { ProviderCapability, HealthStatus, ValidationResult } from '../../types/provider.types';
-import { ChatRequest, ChatResponse, CommandRequest, CommandResponse, ExplainRequest, ExplainResponse, AnalysisRequest, AnalysisResponse, MessageRole } from '../../types/ai.types';
+import { ProviderCapability, ValidationResult } from '../../types/provider.types';
+import { ChatRequest, ChatResponse, CommandRequest, CommandResponse, ExplainRequest, ExplainResponse, AnalysisRequest, AnalysisResponse, MessageRole, StreamEvent } from '../../types/ai.types';
 import { LoggerService } from '../core/logger.service';
 
 /**
@@ -93,11 +93,119 @@ export class AnthropicProviderService extends BaseAiProvider {
     }
 
     /**
-     * 流式聊天功能 - 暂未实现，回退到非流式
+     * 流式聊天功能 - 支持工具调用事件
      */
-    chatStream(request: ChatRequest): Observable<any> {
-        // 回退到非流式
-        return from(this.chat(request));
+    chatStream(request: ChatRequest): Observable<StreamEvent> {
+        return new Observable<StreamEvent>((subscriber: Observer<StreamEvent>) => {
+            if (!this.client) {
+                const error = new Error('Anthropic client not initialized');
+                subscriber.next({ type: 'error', error: error.message });
+                subscriber.error(error);
+                return;
+            }
+
+            let currentToolId = '';
+            let currentToolName = '';
+            let currentToolInput = '';
+            let fullContent = '';
+
+            const abortController = new AbortController();
+
+            const runStream = async () => {
+                try {
+                    const stream = await this.client!.messages.stream({
+                        model: this.config?.model || 'claude-3-sonnet',
+                        max_tokens: request.maxTokens || 1000,
+                        system: request.systemPrompt || this.getDefaultSystemPrompt(),
+                        messages: this.transformMessages(request.messages),
+                        temperature: request.temperature || 1.0,
+                    });
+
+                    for await (const event of stream) {
+                        if (abortController.signal.aborted) break;
+
+                        const eventAny = event as any;
+                        this.logger.debug('Stream event', { type: event.type });
+
+                        // 处理文本增量
+                        if (event.type === 'content_block_delta' && eventAny.delta?.type === 'text_delta') {
+                            const textDelta = eventAny.delta.text;
+                            fullContent += textDelta;
+                            subscriber.next({
+                                type: 'text_delta',
+                                textDelta
+                            });
+                        }
+                        // 处理工具调用开始
+                        else if (event.type === 'content_block_start' && eventAny.content_block?.type === 'tool_use') {
+                            currentToolId = eventAny.content_block.id || `tool_${Date.now()}`;
+                            currentToolName = eventAny.content_block.name;
+                            currentToolInput = '';
+                            subscriber.next({
+                                type: 'tool_use_start',
+                                toolCall: {
+                                    id: currentToolId,
+                                    name: currentToolName,
+                                    input: {}
+                                }
+                            });
+                            this.logger.debug('Stream event', { type: 'tool_use_start', name: currentToolName });
+                        }
+                        // 处理工具调用参数
+                        else if (event.type === 'content_block_delta' && eventAny.delta?.type === 'input_json_delta') {
+                            currentToolInput += eventAny.delta.partial_json || '';
+                        }
+                        // 处理工具调用结束
+                        else if (event.type === 'content_block_stop') {
+                            if (currentToolId && currentToolName) {
+                                let parsedInput = {};
+                                try {
+                                    parsedInput = JSON.parse(currentToolInput || '{}');
+                                } catch (e) {
+                                    // 使用原始输入
+                                }
+                                subscriber.next({
+                                    type: 'tool_use_end',
+                                    toolCall: {
+                                        id: currentToolId,
+                                        name: currentToolName,
+                                        input: parsedInput
+                                    }
+                                });
+                                this.logger.debug('Stream event', { type: 'tool_use_end', name: currentToolName });
+                                currentToolId = '';
+                                currentToolName = '';
+                                currentToolInput = '';
+                            }
+                        }
+                    }
+
+                    subscriber.next({
+                        type: 'message_end',
+                        message: {
+                            id: this.generateId(),
+                            role: MessageRole.ASSISTANT,
+                            content: fullContent,
+                            timestamp: new Date()
+                        }
+                    });
+                    this.logger.debug('Stream event', { type: 'message_end', contentLength: fullContent.length });
+                    subscriber.complete();
+
+                } catch (error) {
+                    if ((error as any).name !== 'AbortError') {
+                        const errorMessage = `Anthropic stream failed: ${error instanceof Error ? error.message : String(error)}`;
+                        this.logger.error('Stream error', error);
+                        subscriber.next({ type: 'error', error: errorMessage });
+                        subscriber.error(new Error(errorMessage));
+                    }
+                }
+            };
+
+            runStream();
+
+            return () => abortController.abort();
+        });
     }
 
     async generateCommand(request: CommandRequest): Promise<CommandResponse> {
@@ -160,31 +268,19 @@ export class AnthropicProviderService extends BaseAiProvider {
         return this.parseAnalysisResponse(response.message.content);
     }
 
-    async healthCheck(): Promise<HealthStatus> {
-        try {
-            if (!this.client) {
-                return HealthStatus.UNHEALTHY;
-            }
-
-            const response = await this.client.messages.create({
-                model: this.config?.model || 'claude-3-sonnet',
-                max_tokens: 1,
-                messages: [
-                    {
-                        role: 'user',
-                        content: 'Hi'
-                    }
-                ]
-            });
+    protected async sendTestRequest(request: ChatRequest): Promise<ChatResponse> {
+        if (!this.client) {
+            throw new Error('Anthropic client not initialized');
+        }
 
-            this.lastHealthCheck = { status: HealthStatus.HEALTHY, timestamp: new Date() };
-            return HealthStatus.HEALTHY;
+        const response = await this.client.messages.create({
+            model: this.config?.model || 'claude-3-sonnet',
+            max_tokens: request.maxTokens || 1,
+            messages: this.transformMessages(request.messages),
+            temperature: request.temperature || 0
+        });
 
-        } catch (error) {
-            this.logger.error('Anthropic health check failed', error);
-            this.lastHealthCheck = { status: HealthStatus.UNHEALTHY, timestamp: new Date() };
-            return HealthStatus.UNHEALTHY;
-        }
+        return this.transformChatResponse(response);
     }
 
     validateConfig(): ValidationResult {
@@ -208,10 +304,6 @@ export class AnthropicProviderService extends BaseAiProvider {
         return result;
     }
 
-    protected getDefaultBaseURL(): string {
-        return 'https://api.anthropic.com';
-    }
-
     protected transformMessages(messages: any[]): any[] {
         return messages.map(msg => ({
             role: msg.role,
@@ -237,172 +329,4 @@ export class AnthropicProviderService extends BaseAiProvider {
             } : undefined
         };
     }
-
-    private buildCommandPrompt(request: CommandRequest): string {
-        let prompt = `请将以下自然语言描述转换为准确的终端命令：\n\n"${request.naturalLanguage}"\n\n`;
-
-        if (request.context) {
-            prompt += `当前环境：\n`;
-            if (request.context.currentDirectory) {
-                prompt += `- 当前目录：${request.context.currentDirectory}\n`;
-            }
-            if (request.context.operatingSystem) {
-                prompt += `- 操作系统：${request.context.operatingSystem}\n`;
-            }
-            if (request.context.shell) {
-                prompt += `- Shell：${request.context.shell}\n`;
-            }
-        }
-
-        prompt += `\n请直接返回JSON格式：\n`;
-        prompt += `{\n`;
-        prompt += `  "command": "具体命令",\n`;
-        prompt += `  "explanation": "命令解释",\n`;
-        prompt += `  "confidence": 0.95\n`;
-        prompt += `}\n`;
-
-        return prompt;
-    }
-
-    private buildExplainPrompt(request: ExplainRequest): string {
-        let prompt = `请详细解释以下终端命令：\n\n\`${request.command}\`\n\n`;
-
-        if (request.context?.currentDirectory) {
-            prompt += `当前目录：${request.context.currentDirectory}\n`;
-        }
-        if (request.context?.operatingSystem) {
-            prompt += `操作系统：${request.context.operatingSystem}\n`;
-        }
-
-        prompt += `\n请按以下JSON格式返回：\n`;
-        prompt += `{\n`;
-        prompt += `  "explanation": "整体解释",\n`;
-        prompt += `  "breakdown": [\n`;
-        prompt += `    {"part": "命令部分", "description": "说明"}\n`;
-        prompt += `  ],\n`;
-        prompt += `  "examples": ["使用示例"]\n`;
-        prompt += `}\n`;
-
-        return prompt;
-    }
-
-    private buildAnalysisPrompt(request: AnalysisRequest): string {
-        let prompt = `请分析以下命令执行结果：\n\n`;
-        prompt += `命令：${request.command}\n`;
-        prompt += `退出码：${request.exitCode}\n`;
-        prompt += `输出：\n${request.output}\n\n`;
-
-        if (request.context?.workingDirectory) {
-            prompt += `工作目录：${request.context.workingDirectory}\n`;
-        }
-
-        prompt += `\n请按以下JSON格式返回：\n`;
-        prompt += `{\n`;
-        prompt += `  "summary": "结果总结",\n`;
-        prompt += `  "insights": ["洞察1", "洞察2"],\n`;
-        prompt += `  "success": true/false,\n`;
-        prompt += `  "issues": [\n`;
-        prompt += `    {"severity": "warning|error|info", "message": "问题描述", "suggestion": "建议"}\n`;
-        prompt += `  ]\n`;
-        prompt += `}\n`;
-
-        return prompt;
-    }
-
-    private parseCommandResponse(content: string): CommandResponse {
-        try {
-            const match = content.match(/\{[\s\S]*\}/);
-            if (match) {
-                const parsed = JSON.parse(match[0]);
-                return {
-                    command: parsed.command || '',
-                    explanation: parsed.explanation || '',
-                    confidence: parsed.confidence || 0.5
-                };
-            }
-        } catch (error) {
-            this.logger.warn('Failed to parse command response as JSON', error);
-        }
-
-        const lines = content.split('\n').map(l => l.trim()).filter(l => l);
-        return {
-            command: lines[0] || '',
-            explanation: lines.slice(1).join(' ') || 'AI生成的命令',
-            confidence: 0.5
-        };
-    }
-
-    private parseExplainResponse(content: string): ExplainResponse {
-        try {
-            const match = content.match(/\{[\s\S]*\}/);
-            if (match) {
-                const parsed = JSON.parse(match[0]);
-                return {
-                    explanation: parsed.explanation || '',
-                    breakdown: parsed.breakdown || [],
-                    examples: parsed.examples || []
-                };
-            }
-        } catch (error) {
-            this.logger.warn('Failed to parse explain response as JSON', error);
-        }
-
-        return {
-            explanation: content,
-            breakdown: []
-        };
-    }
-
-    private parseAnalysisResponse(content: string): AnalysisResponse {
-        try {
-            const match = content.match(/\{[\s\S]*\}/);
-            if (match) {
-                const parsed = JSON.parse(match[0]);
-                return {
-                    summary: parsed.summary || '',
-                    insights: parsed.insights || [],
-                    success: parsed.success !== false,
-                    issues: parsed.issues || []
-                };
-            }
-        } catch (error) {
-            this.logger.warn('Failed to parse analysis response as JSON', error);
-        }
-
-        return {
-            summary: content,
-            insights: [],
-            success: true
-        };
-    }
-
-    private getDefaultSystemPrompt(): string {
-        return `你是一个专业的终端命令助手，运行在 Tabby 终端中。
-
-## 核心能力
-你可以通过以下工具直接操作终端：
-- write_to_terminal: 向终端写入并执行命令
-- read_terminal_output: 读取终端输出
-- get_terminal_list: 获取所有终端列表
-- get_terminal_cwd: 获取当前工作目录
-- focus_terminal: 切换到指定索引的终端（需要参数 terminal_index）
-- get_terminal_selection: 获取终端中选中的文本
-
-## 重要规则
-1. 当用户请求执行命令（如"查看当前目录"、"列出文件"等），你必须使用 write_to_terminal 工具来执行
-2. **当用户请求切换终端（如"切换到终端0"、"打开终端4"等），你必须使用 focus_terminal 工具**
-3. 不要只是描述你"将要做什么"，而是直接调用工具执行
-4. 执行命令后，使用 read_terminal_output 读取结果并报告给用户
-5. 如果不确定当前目录或终端状态，先使用 get_terminal_cwd 或 get_terminal_list 获取信息
-6. **永远不要假装执行了操作，必须真正调用工具**
-
-## 示例
-用户："查看当前目录的文件"
-正确做法：调用 write_to_terminal 工具，参数 { "command": "dir", "execute": true }
-错误做法：仅回复文字"我将执行 dir 命令"
-
-用户："切换到终端4"
-正确做法：调用 focus_terminal 工具，参数 { "terminal_index": 4 }
-错误做法：仅回复文字"已切换到终端4"（不调用工具）`;
-    }
 }