tabby-ai-assistant 1.0.0

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

@@ -0,0 +1,104 @@
+import { Injectable } from '@angular/core';
+
+@Injectable({ providedIn: 'root' })
+export class LoggerService {
+    private logLevel: 'debug' | 'info' | 'warn' | 'error' = 'info';
+    private logs: Array<{ level: string; message: string; timestamp: Date; data?: any }> = [];
+
+    constructor() {
+        this.loadLogLevel();
+    }
+
+    private loadLogLevel(): void {
+        // 从配置中加载日志级别
+        try {
+            const config = localStorage.getItem('ai-assistant-config');
+            if (config) {
+                const parsed = JSON.parse(config);
+                this.logLevel = parsed.logLevel || 'info';
+            }
+        } catch (e) {
+            // 忽略错误，使用默认级别
+        }
+    }
+
+    debug(message: string, data?: any): void {
+        if (this.shouldLog('debug')) {
+            this.log('DEBUG', message, data);
+        }
+    }
+
+    info(message: string, data?: any): void {
+        if (this.shouldLog('info')) {
+            this.log('INFO', message, data);
+        }
+    }
+
+    warn(message: string, data?: any): void {
+        if (this.shouldLog('warn')) {
+            this.log('WARN', message, data);
+        }
+    }
+
+    error(message: string, error?: any): void {
+        if (this.shouldLog('error')) {
+            this.log('ERROR', message, error);
+        }
+    }
+
+    private shouldLog(level: string): boolean {
+        const levels = ['debug', 'info', 'warn', 'error'];
+        const currentIndex = levels.indexOf(this.logLevel);
+        const messageIndex = levels.indexOf(level);
+        return messageIndex >= currentIndex;
+    }
+
+    private log(level: string, message: string, data?: any): void {
+        const entry = {
+            level,
+            message,
+            timestamp: new Date(),
+            data
+        };
+
+        this.logs.push(entry);
+
+        // 限制日志数量，避免内存溢出
+        if (this.logs.length > 1000) {
+            this.logs = this.logs.slice(-500);
+        }
+
+        // 输出到控制台
+        switch (level) {
+            case 'DEBUG':
+                console.debug(`[AI Assistant] ${message}`, data);
+                break;
+            case 'INFO':
+                console.info(`[AI Assistant] ${message}`, data);
+                break;
+            case 'WARN':
+                console.warn(`[AI Assistant] ${message}`, data);
+                break;
+            case 'ERROR':
+                console.error(`[AI Assistant] ${message}`, data);
+                break;
+        }
+    }
+
+    getLogs(): Array<{ level: string; message: string; timestamp: Date; data?: any }> {
+        return [...this.logs];
+    }
+
+    clearLogs(): void {
+        this.logs = [];
+    }
+
+    setLogLevel(level: 'debug' | 'info' | 'warn' | 'error'): void {
+        this.logLevel = level;
+        localStorage.setItem('ai-assistant-log-level', level);
+    }
+
+    getLogLevel(): string {
+        return this.logLevel;
+    }
+}

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

@@ -0,0 +1,373 @@
+import { Injectable } from '@angular/core';
+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 { LoggerService } from '../core/logger.service';
+
+/**
+ * Anthropic Claude AI提供商
+ * 基于Anthropic Claude API
+ */
+@Injectable()
+export class AnthropicProviderService extends BaseAiProvider {
+    readonly name = 'anthropic';
+    readonly displayName = 'Anthropic Claude';
+    readonly capabilities = [
+        ProviderCapability.CHAT,
+        ProviderCapability.COMMAND_GENERATION,
+        ProviderCapability.COMMAND_EXPLANATION,
+        ProviderCapability.REASONING,
+        ProviderCapability.STREAMING
+    ];
+    readonly authConfig = {
+        type: 'bearer' as const,
+        credentials: {
+            apiKey: ''
+        }
+    };
+
+    private client: Anthropic | null = null;
+
+    constructor(logger: LoggerService) {
+        super(logger);
+    }
+
+    configure(config: any): void {
+        super.configure(config);
+        this.authConfig.credentials.apiKey = config.apiKey || '';
+        this.initializeClient();
+    }
+
+    private initializeClient(): void {
+        if (!this.config?.apiKey) {
+            this.logger.warn('Anthropic API key not provided');
+            return;
+        }
+
+        try {
+            this.client = new Anthropic({
+                apiKey: this.config.apiKey,
+                baseURL: this.getBaseURL()
+            });
+
+            this.logger.info('Anthropic client initialized', {
+                baseURL: this.getBaseURL(),
+                model: this.config.model || 'claude-3-sonnet'
+            });
+        } catch (error) {
+            this.logger.error('Failed to initialize Anthropic client', error);
+            throw error;
+        }
+    }
+
+    async chat(request: ChatRequest): Promise<ChatResponse> {
+        if (!this.client) {
+            throw new Error('Anthropic client not initialized');
+        }
+
+        this.logRequest(request);
+
+        try {
+            const response = await this.withRetry(async () => {
+                const result = await this.client!.messages.create({
+                    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,
+                    stream: request.stream || false
+                });
+
+                this.logResponse(result);
+                return result;
+            });
+
+            return this.transformChatResponse(response);
+
+        } catch (error) {
+            this.logError(error, { request });
+            throw new Error(`Anthropic chat failed: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+
+    async generateCommand(request: CommandRequest): Promise<CommandResponse> {
+        const prompt = this.buildCommandPrompt(request);
+
+        const chatRequest: ChatRequest = {
+            messages: [
+                {
+                    id: this.generateId(),
+                    role: MessageRole.USER,
+                    content: prompt,
+                    timestamp: new Date()
+                }
+            ],
+            maxTokens: 500,
+            temperature: 0.3
+        };
+
+        const response = await this.chat(chatRequest);
+        return this.parseCommandResponse(response.message.content);
+    }
+
+    async explainCommand(request: ExplainRequest): Promise<ExplainResponse> {
+        const prompt = this.buildExplainPrompt(request);
+
+        const chatRequest: ChatRequest = {
+            messages: [
+                {
+                    id: this.generateId(),
+                    role: MessageRole.USER,
+                    content: prompt,
+                    timestamp: new Date()
+                }
+            ],
+            maxTokens: 1000,
+            temperature: 0.5
+        };
+
+        const response = await this.chat(chatRequest);
+        return this.parseExplainResponse(response.message.content);
+    }
+
+    async analyzeResult(request: AnalysisRequest): Promise<AnalysisResponse> {
+        const prompt = this.buildAnalysisPrompt(request);
+
+        const chatRequest: ChatRequest = {
+            messages: [
+                {
+                    id: this.generateId(),
+                    role: MessageRole.USER,
+                    content: prompt,
+                    timestamp: new Date()
+                }
+            ],
+            maxTokens: 1000,
+            temperature: 0.7
+        };
+
+        const response = await this.chat(chatRequest);
+        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'
+                    }
+                ]
+            });
+
+            this.lastHealthCheck = { status: HealthStatus.HEALTHY, timestamp: new Date() };
+            return HealthStatus.HEALTHY;
+
+        } catch (error) {
+            this.logger.error('Anthropic health check failed', error);
+            this.lastHealthCheck = { status: HealthStatus.UNHEALTHY, timestamp: new Date() };
+            return HealthStatus.UNHEALTHY;
+        }
+    }
+
+    validateConfig(): ValidationResult {
+        const result = super.validateConfig();
+
+        if (!this.config?.apiKey) {
+            return {
+                valid: false,
+                errors: [...(result.errors || []), 'Anthropic API key is required']
+            };
+        }
+
+        const supportedModels = ['claude-3-opus', 'claude-3-sonnet', 'claude-3-haiku'];
+        if (this.config.model && !supportedModels.includes(this.config.model)) {
+            result.warnings = [
+                ...(result.warnings || []),
+                `Model ${this.config.model} might not be supported. Supported models: ${supportedModels.join(', ')}`
+            ];
+        }
+
+        return result;
+    }
+
+    protected getDefaultBaseURL(): string {
+        return 'https://api.anthropic.com';
+    }
+
+    protected transformMessages(messages: any[]): any[] {
+        return messages.map(msg => ({
+            role: msg.role,
+            content: [{ type: 'text', text: msg.content }]
+        }));
+    }
+
+    private transformChatResponse(response: any): ChatResponse {
+        const content = response.content[0];
+        const text = content.type === 'text' ? content.text : '';
+
+        return {
+            message: {
+                id: this.generateId(),
+                role: MessageRole.ASSISTANT,
+                content: text,
+                timestamp: new Date()
+            },
+            usage: response.usage ? {
+                promptTokens: response.usage.input_tokens || 0,
+                completionTokens: response.usage.output_tokens || 0,
+                totalTokens: (response.usage.input_tokens || 0) + (response.usage.output_tokens || 0)
+            } : 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 `你是一个专业的终端命令助手，擅长将自然语言转换为准确的命令，并提供详细的解释。`;
+    }
+}