From b9daae8cb4b0adcb0a2fb5e87012f9259456a7a5 Mon Sep 17 00:00:00 2001 From: digua Date: Thu, 26 Feb 2026 21:05:08 +0800 Subject: [PATCH] chore: remove @ai-sdk dependencies and legacy LLM service implementations - Remove @ai-sdk/google, @ai-sdk/openai, ai packages from dependencies - Delete SDK-based service files: gemini.ts, openai-compatible.ts, sdkUtils.ts - Clean up unused types from llm/types.ts (ChatMessage, ChatOptions, LLMConfig, ILLMService, etc.) --- electron/main/ai/llm/gemini.ts | 266 ----------- electron/main/ai/llm/openai-compatible.ts | 543 ---------------------- electron/main/ai/llm/sdkUtils.ts | 150 ------ electron/main/ai/llm/types.ts | 156 ------- package.json | 3 - 5 files changed, 1118 deletions(-) delete mode 100644 electron/main/ai/llm/gemini.ts delete mode 100644 electron/main/ai/llm/openai-compatible.ts delete mode 100644 electron/main/ai/llm/sdkUtils.ts diff --git a/electron/main/ai/llm/gemini.ts b/electron/main/ai/llm/gemini.ts deleted file mode 100644 index ebc328e..0000000 --- a/electron/main/ai/llm/gemini.ts +++ /dev/null @@ -1,266 +0,0 @@ -/** - * Google Gemini LLM Provider - * 使用 AI SDK 的 Google Generative AI Provider，支持 Function Calling - */ - -import { createGoogleGenerativeAI } from '@ai-sdk/google' -import { generateText, streamText } from 'ai' -import type { ContentPart, ModelMessage, ToolSet, TypedToolCall } from 'ai' -import type { - ILLMService, - LLMProvider, - ChatMessage, - ChatOptions, - ChatResponse, - ChatStreamChunk, - ToolCall, - ProviderInfo, -} from './types' -import { aiLogger } from '../logger' -import { buildModelMessages, buildToolSet, mapFinishReason, mapUsage } from './sdkUtils' - -const DEFAULT_BASE_URL = 'https://generativelanguage.googleapis.com' -const DEFAULT_API_VERSION = '/v1beta' -const GEMINI_MAX_RETRIES = 1 - -const MODELS = [ - { id: 'gemini-3-flash-preview', name: 'Gemini 3 Flash Preview', description: '高速预览版' }, - { id: 'gemini-3-pro-preview', name: 'Gemini 3 Pro Preview', description: '专业预览版' }, -] - -export const GEMINI_INFO: ProviderInfo = { - id: 'gemini', - name: 'Gemini', - description: 'Google Gemini 大语言模型', - defaultBaseUrl: DEFAULT_BASE_URL, - models: MODELS, -} - -/** - * 统一处理 Gemini 的 baseUrl，确保包含 /v1beta - */ -function normalizeBaseUrl(baseUrl?: string): string { - const normalized = (baseUrl || DEFAULT_BASE_URL).replace(/\/+$/, '') - - if (normalized.endsWith(DEFAULT_API_VERSION)) { - return normalized - } - - return `${normalized}${DEFAULT_API_VERSION}` -} - -const GEMINI_PROVIDER_KEYS = ['google', 'vertex', 'gemini'] - -/** - * 从 AI SDK 的 providerMetadata 中提取 thoughtSignature - */ -function getThoughtSignature(metadata?: unknown): string | undefined { - if (!metadata || typeof metadata !== 'object') { - return undefined - } - - const record = metadata as Record - for (const key of GEMINI_PROVIDER_KEYS) { - const providerMeta = record[key] - if (providerMeta && typeof providerMeta === 'object') { - const signature = (providerMeta as Record).thoughtSignature - if (typeof signature === 'string' && signature) { - return signature - } - } - } - - const directSignature = record.thoughtSignature - if (typeof directSignature === 'string' && directSignature) { - return directSignature - } - - return undefined -} - -/** - * Gemini 工具调用需要 thoughtSignature - */ -function mapGeminiToolCalls(toolCalls: TypedToolCall[]): ToolCall[] { - return toolCalls.map((tc) => ({ - id: tc.toolCallId, - type: 'function' as const, - function: { - name: tc.toolName, - arguments: JSON.stringify(tc.input ?? {}), - }, - thoughtSignature: getThoughtSignature(tc.providerMetadata), - })) -} - -/** - * 构建 Gemini 的模型消息，补充 thoughtSignature - */ -function buildGeminiModelMessages(messages: ChatMessage[]): ModelMessage[] { - const modelMessages = buildModelMessages(messages) - const signatureMap = new Map() - - for (const message of messages) { - if (message.role !== 'assistant' || !message.tool_calls) continue - for (const toolCall of message.tool_calls) { - if (toolCall.thoughtSignature) { - signatureMap.set(toolCall.id, toolCall.thoughtSignature) - } - } - } - - if (signatureMap.size === 0) { - return modelMessages - } - - return modelMessages.map((message) => { - if (message.role !== 'assistant' || !Array.isArray(message.content)) { - return message - } - - const contentParts = message.content.map((part) => { - if (part.type !== 'tool-call') { - return part - } - - const signature = signatureMap.get(part.toolCallId) - if (!signature) { - return part - } - - const nextPart: ContentPart & { - providerOptions?: Record - } = { - ...part, - providerOptions: { - google: { thoughtSignature: signature }, - }, - } - - return nextPart - }) - - return { - ...message, - content: contentParts, - } - }) -} - -export class GeminiService implements ILLMService { - private apiKey: string - private baseUrl: string - private model: string - - private provider = createGoogleGenerativeAI() - - constructor(apiKey: string, model?: string, baseUrl?: string) { - this.apiKey = apiKey - this.baseUrl = normalizeBaseUrl(baseUrl) - this.model = model || 'gemini-3-flash-preview' - this.provider = createGoogleGenerativeAI({ - apiKey: this.apiKey, - baseURL: this.baseUrl, - name: 'gemini', - }) - } - - getProvider(): LLMProvider { - return 'gemini' - } - - getModels(): string[] { - return MODELS.map((m) => m.id) - } - - getDefaultModel(): string { - return 'gemini-3-flash-preview' - } - - async chat(messages: ChatMessage[], options?: ChatOptions): Promise { - const model = this.provider.chat(this.model) - const toolSet = buildToolSet(options?.tools) - - const result = await generateText({ - model, - messages: buildGeminiModelMessages(messages), - tools: toolSet, - temperature: options?.temperature ?? 0.7, - maxTokens: options?.maxTokens ?? 2048, - // 降低 Gemini 自动重试次数，避免触发免费配额限制 - maxRetries: GEMINI_MAX_RETRIES, - abortSignal: options?.abortSignal, - }) - - const toolCalls = result.toolCalls.length > 0 ? mapGeminiToolCalls(result.toolCalls) : undefined - - return { - content: result.text, - finishReason: mapFinishReason(result.finishReason), - tool_calls: toolCalls, - usage: mapUsage(result.usage), - } - } - - async *chatStream(messages: ChatMessage[], options?: ChatOptions): AsyncGenerator { - const model = this.provider.chat(this.model) - const toolSet = buildToolSet(options?.tools) - - const result = streamText({ - model, - messages: buildGeminiModelMessages(messages), - tools: toolSet, - temperature: options?.temperature ?? 0.7, - maxTokens: options?.maxTokens ?? 2048, - // 降低 Gemini 自动重试次数，避免触发免费配额限制 - maxRetries: GEMINI_MAX_RETRIES, - abortSignal: options?.abortSignal, - }) - - try { - for await (const chunk of result.textStream) { - if (options?.abortSignal?.aborted) { - yield { content: '', isFinished: true, finishReason: 'stop' } - return - } - if (chunk) { - yield { content: chunk, isFinished: false } - } - } - - const finishReason = mapFinishReason(await result.finishReason) - const toolCalls = await result.toolCalls - const usage = mapUsage(await result.totalUsage) - - yield { - content: '', - isFinished: true, - finishReason, - tool_calls: toolCalls.length > 0 ? mapGeminiToolCalls(toolCalls) : undefined, - usage, - } - } catch (error) { - if (error instanceof Error && error.name === 'AbortError') { - yield { content: '', isFinished: true, finishReason: 'stop' } - return - } - aiLogger.error('Gemini', 'Stream request failed', { error: String(error) }) - throw error - } - } - - async validateApiKey(): Promise<{ success: boolean; error?: string }> { - try { - const model = this.provider.chat(this.model) - await generateText({ - model, - prompt: 'Hi', - maxTokens: 1, - }) - return { success: true } - } catch (error) { - const errorMessage = error instanceof Error ? error.message : String(error) - return { success: false, error: errorMessage } - } - } -} diff --git a/electron/main/ai/llm/openai-compatible.ts b/electron/main/ai/llm/openai-compatible.ts deleted file mode 100644 index 7d35913..0000000 --- a/electron/main/ai/llm/openai-compatible.ts +++ /dev/null @@ -1,543 +0,0 @@ -/** - * OpenAI Compatible LLM Provider - * 支持任何兼容 OpenAI API 格式的服务（如 Ollama、LocalAI、vLLM 等） - */ - -import { createOpenAI } from '@ai-sdk/openai' -import { generateText, streamText, wrapLanguageModel, extractReasoningMiddleware } from 'ai' -import type { ModelMessage, ToolSet, TypedToolCall, LanguageModel } from 'ai' -import type { - ILLMService, - LLMProvider, - ChatMessage, - ChatOptions, - ChatResponse, - ChatStreamChunk, - ToolCall, - ProviderInfo, -} from './types' -import { aiLogger } from '../logger' -import { buildModelMessages, buildToolSet, mapFinishReason, mapToolCalls, mapUsage } from './sdkUtils' - -/** - * 从 AI SDK 错误中提取详细信息 - * 特别是从 responseBody 中获取模型/服务端返回的错误消息 - */ -interface DetailedError { - message: string - statusCode?: number - responseBody?: string - originalMessage?: string -} - -function extractDetailedError(error: unknown): DetailedError { - const result: DetailedError = { - message: '未知错误', - } - - if (!error) { - return result - } - - // 处理 Error 对象 - if (error instanceof Error) { - result.originalMessage = error.message - result.message = error.message - - // 尝试从错误对象中提取更多信息 - const anyError = error as Record - - // 检查状态码 - if (typeof anyError.statusCode === 'number') { - result.statusCode = anyError.statusCode - } - - // 检查是否有 responseBody（API 错误通常包含）- 直接使用原始内容 - if (typeof anyError.responseBody === 'string') { - result.responseBody = anyError.responseBody - result.message = anyError.responseBody - } - - // 检查是否是 RetryError（包含多个错误） - if (Array.isArray(anyError.errors)) { - const errors = anyError.errors as Array> - for (const innerError of errors) { - // 提取状态码 - if (typeof innerError.statusCode === 'number' && !result.statusCode) { - result.statusCode = innerError.statusCode - } - // 提取 responseBody - 直接使用原始内容 - if (typeof innerError.responseBody === 'string') { - result.responseBody = innerError.responseBody - result.message = innerError.responseBody - break - } - } - } - } else if (typeof error === 'string') { - result.message = error - } else { - result.message = String(error) - } - - return result -} - -const DEFAULT_BASE_URL = 'http://localhost:11434/v1' -const DEFAULT_THOUGHT_SIGNATURE = 'context_engineering_is_the_way_to_go' - -export const OPENAI_COMPATIBLE_INFO: ProviderInfo = { - id: 'openai-compatible', - name: 'OpenAI 兼容', - description: '支持任何兼容 OpenAI API 的服务（如 Ollama、LocalAI、vLLM 等）', - defaultBaseUrl: DEFAULT_BASE_URL, - models: [ - { id: 'llama3.2', name: 'Llama 3.2', description: 'Meta Llama 3.2 模型' }, - { id: 'qwen2.5', name: 'Qwen 2.5', description: '通义千问 2.5 模型' }, - { id: 'deepseek-r1', name: 'DeepSeek R1', description: 'DeepSeek R1 推理模型' }, - ], -} - -/** - * 统一处理 baseUrl：去掉尾部斜杠和多余路径 - */ -function normalizeBaseUrl(baseUrl?: string): string { - let processed = baseUrl || DEFAULT_BASE_URL - processed = processed.replace(/\/+$/, '') - if (processed.endsWith('/chat/completions')) { - processed = processed.slice(0, -'/chat/completions'.length) - } - return processed -} - -/** - * MiniMax 流式返回可能是累计文本，这里按前缀增量去重 - */ -function dedupeCumulativeStreamChunk(chunk: string, previousText: string): { delta: string; nextText: string } { - if (!previousText) { - return { delta: chunk, nextText: chunk } - } - - if (chunk.startsWith(previousText)) { - return { delta: chunk.slice(previousText.length), nextText: chunk } - } - - if (previousText.startsWith(chunk)) { - // 偶发回退或重复帧，保持已输出内容 - return { delta: '', nextText: previousText } - } - - // 无法判定为累计时，退化为增量追加 - return { delta: chunk, nextText: previousText + chunk } -} - -/** - * 包装 fetch：注入思考开关和 thought_signature - */ -function createCompatFetch(disableThinking: boolean): typeof fetch { - return async (input, init) => { - if (!init?.body || typeof init.body !== 'string') { - return fetch(input, init) - } - - let parsedBody: Record | null = null - try { - parsedBody = JSON.parse(init.body) as Record - } catch { - return fetch(input, init) - } - - if (!parsedBody) { - return fetch(input, init) - } - - let changed = false - - if (Array.isArray(parsedBody.messages)) { - const messages = parsedBody.messages as Array> - // 为 Gemini 兼容后端补充 thought_signature - for (const message of messages) { - if ( - message && - typeof message === 'object' && - (message as { role?: string }).role === 'assistant' && - Array.isArray((message as { tool_calls?: unknown[] }).tool_calls) - ) { - const toolCalls = (message as { tool_calls: Array> }).tool_calls - for (const toolCall of toolCalls) { - const typedCall = toolCall as Record & { - thought_signature?: string - thoughtSignature?: string - } - if (!typedCall.thought_signature && !typedCall.thoughtSignature) { - typedCall.thought_signature = DEFAULT_THOUGHT_SIGNATURE - changed = true - } - } - } - } - - // 禁用思考模式（用于本地模型） - if (disableThinking) { - const chatTemplate = parsedBody.chat_template_kwargs - if (!chatTemplate || typeof chatTemplate !== 'object') { - parsedBody.chat_template_kwargs = { enable_thinking: false } - changed = true - } else if (!(chatTemplate as { enable_thinking?: boolean }).enable_thinking) { - parsedBody.chat_template_kwargs = { - ...(chatTemplate as Record), - enable_thinking: false, - } - changed = true - } - } - } - - if (!changed) { - return fetch(input, init) - } - - const nextInit: RequestInit = { - ...init, - body: JSON.stringify(parsedBody), - } - - return fetch(input, nextInit) - } -} - -export class OpenAICompatibleService implements ILLMService { - private apiKey: string - private baseUrl: string - private model: string - private providerId: LLMProvider - private models: ProviderInfo['models'] - private defaultModel: string - private provider = createOpenAI() - private isReasoning: boolean = false - - constructor( - apiKey: string, - model?: string, - baseUrl?: string, - disableThinking?: boolean, - providerId?: LLMProvider, - models?: ProviderInfo['models'], - isReasoningModelConfig?: boolean - ) { - const normalizedBaseUrl = normalizeBaseUrl(baseUrl) - const resolvedApiKey = apiKey || 'sk-no-key-required' - const resolvedDisableThinking = disableThinking ?? true - const resolvedProviderId = providerId ?? 'openai-compatible' - const resolvedModels = models && models.length > 0 ? models : OPENAI_COMPATIBLE_INFO.models - const defaultModel = resolvedModels[0]?.id || 'llama3.2' - const resolvedModel = model || defaultModel - - this.apiKey = resolvedApiKey - this.baseUrl = normalizedBaseUrl - this.providerId = resolvedProviderId - this.models = resolvedModels - this.defaultModel = defaultModel - this.model = resolvedModel - - // 推理模型判断：完全依赖用户配置 - this.isReasoning = isReasoningModelConfig ?? false - if (this.isReasoning) { - aiLogger.info(this.providerId, `User configured reasoning model: ${resolvedModel}`) - } - - this.provider = createOpenAI({ - apiKey: resolvedApiKey, - baseURL: normalizedBaseUrl, - name: 'openai-compatible', - fetch: createCompatFetch(resolvedDisableThinking), - }) - } - - /** - * 获取语言模型实例 - * 对于推理模型，会应用 extractReasoningMiddleware 中间件 - */ - private getModel(): LanguageModel { - const baseModel = this.provider.chat(this.model) - - if (this.isReasoning) { - // 对推理模型应用中间件，提取 标签中的思考内容 - // startWithReasoning: true - 因为 Ollama 上的 DeepSeek-R1 响应总是以思考开始 - return wrapLanguageModel({ - model: baseModel, - middleware: extractReasoningMiddleware({ - tagName: 'think', - startWithReasoning: true, - }), - }) - } - - return baseModel - } - - getProvider(): LLMProvider { - return this.providerId - } - - getModels(): string[] { - return this.models.map((m) => m.id) - } - - getDefaultModel(): string { - return this.defaultModel - } - - // 统一处理消息映射，保持与旧实现一致 - private buildMessages(messages: ChatMessage[]): ModelMessage[] { - return buildModelMessages(messages) - } - - // 统一映射工具调用结果 - private mapToolCalls(toolCalls: TypedToolCall[]): ToolCall[] { - return mapToolCalls(toolCalls) - } - - // 仅 MiniMax 需要累计去重，避免其他模型缓存全文 - private shouldTrackStreamText(): boolean { - return this.providerId === 'minimax' - } - - // MiniMax 流式返回可能是累计文本，按前缀增量去重 - private getStreamChunkDelta(chunk: string, previousText: string): { delta: string; nextText: string } { - if (this.providerId !== 'minimax') { - return { delta: chunk, nextText: previousText + chunk } - } - return dedupeCumulativeStreamChunk(chunk, previousText) - } - - async chat(messages: ChatMessage[], options?: ChatOptions): Promise { - const model = this.getModel() - // 推理模型（如 DeepSeek-R1）不支持 tool-calling，需要禁用 tools - const toolSet = this.isReasoning ? undefined : buildToolSet(options?.tools) - - aiLogger.info(this.providerId, 'Starting non-streaming request', { - model: this.model, - isReasoning: this.isReasoning, - hasTools: !!toolSet, - messagesCount: messages.length, - }) - - const result = await generateText({ - model, - messages: this.buildMessages(messages), - tools: toolSet, - temperature: options?.temperature ?? 0.7, - maxTokens: options?.maxTokens ?? 2048, - abortSignal: options?.abortSignal, - }) - - const toolCalls = result.toolCalls.length > 0 ? this.mapToolCalls(result.toolCalls) : undefined - - // 对于推理模型，reasoningText 包含思考内容 - if (this.isReasoning && result.reasoningText) { - aiLogger.info(this.providerId, 'Reasoning model returned thinking content', { - reasoningLength: result.reasoningText.length, - textLength: result.text.length, - }) - } - - return { - content: result.text, - finishReason: mapFinishReason(result.finishReason), - tool_calls: toolCalls, - usage: mapUsage(result.usage), - } - } - - async *chatStream(messages: ChatMessage[], options?: ChatOptions): AsyncGenerator { - const model = this.getModel() - const shouldTrack = this.shouldTrackStreamText() - let streamedText = '' - - // 推理模型（如 DeepSeek-R1）不支持 tool-calling，需要禁用 tools - // 参考: https://sdk.vercel.ai/docs/guides/r1#limitations - const toolSet = this.isReasoning ? undefined : buildToolSet(options?.tools) - - aiLogger.info(this.providerId, 'Starting streaming request', { - model: this.model, - isReasoning: this.isReasoning, - hasTools: !!toolSet, - messagesCount: messages.length, - }) - - const result = streamText({ - model, - messages: this.buildMessages(messages), - tools: toolSet, - temperature: options?.temperature ?? 0.7, - maxTokens: options?.maxTokens ?? 2048, - abortSignal: options?.abortSignal, - }) - - let partCount = 0 - let textChunkCount = 0 - let reasoningChunkCount = 0 - - try { - // 使用 fullStream 替代 textStream，以支持 reasoning 模型（如 DeepSeek-R1） - // fullStream 包含 text/reasoning/tool-call 等多种类型的 chunk - for await (const part of result.fullStream) { - partCount++ - - if (options?.abortSignal?.aborted) { - yield { content: '', isFinished: true, finishReason: 'stop' } - return - } - - // 记录每个 part 的类型（用于调试） - if (partCount <= 5 || partCount % 50 === 0) { - aiLogger.info(this.providerId, `Stream part #${partCount}`, { type: part.type }) - } - - // 处理 error part - 获取详细错误信息并抛出更有意义的错误 - if (part.type === 'error') { - const errorPart = part as { type: 'error'; error: unknown } - const detailedError = extractDetailedError(errorPart.error) - - aiLogger.error(this.providerId, 'Stream received error part', { - detailedError, - originalError: String(errorPart.error), - }) - - // 抛出包含详细信息的错误，让前端能够显示 - throw new Error(detailedError.message) - } - - // 处理不同类型的 stream part - // 注意：fullStream 返回的文本增量类型可能是 'text' 或 'text-delta' - const textPart = part as { type: string; text?: string; textDelta?: string } - if (part.type === 'text' || part.type === 'text-delta') { - // 普通文本内容 - textChunkCount++ - const chunk = textPart.text || textPart.textDelta - if (chunk) { - const { delta, nextText } = this.getStreamChunkDelta(chunk, streamedText) - if (shouldTrack) { - streamedText = nextText - } - if (delta) { - yield { content: delta, isFinished: false } - } - } - } else if (part.type === 'reasoning') { - // 推理内容（DeepSeek-R1 等推理模型） - // 包装为 标签，让 Agent 的 createStreamParser 能够正确解析 - reasoningChunkCount++ - const reasoningText = part.text - if (reasoningText) { - yield { content: `${reasoningText}`, isFinished: false } - } - } else if (part.type === 'finish') { - // 流结束 - const finishReason = mapFinishReason(part.finishReason) - const toolCalls = await result.toolCalls - const usage = mapUsage(part.totalUsage) - - // 详细记录流式请求完成信息，包括工具调用 - aiLogger.info(this.providerId, 'Streaming request completed', { - partCount, - textChunkCount, - reasoningChunkCount, - finishReason: part.finishReason, - mappedFinishReason: finishReason, - toolCallsCount: toolCalls.length, - toolCallNames: toolCalls.map((tc) => tc.toolName), - }) - - yield { - content: '', - isFinished: true, - finishReason, - tool_calls: toolCalls.length > 0 ? this.mapToolCalls(toolCalls) : undefined, - usage, - } - return - } - // 其他类型（tool-call, tool-result, start, start-step, finish-step 等）暂时忽略 - } - - // 兜底：如果没有收到 finish 事件，手动获取结果 - aiLogger.warn(this.providerId, 'Stream ended without receiving finish event', { - partCount, - textChunkCount, - reasoningChunkCount, - }) - - const finishReason = mapFinishReason(await result.finishReason) - const toolCalls = await result.toolCalls - const usage = mapUsage(await result.totalUsage) - - yield { - content: '', - isFinished: true, - finishReason, - tool_calls: toolCalls.length > 0 ? this.mapToolCalls(toolCalls) : undefined, - usage, - } - } catch (error) { - aiLogger.error(this.providerId, 'Streaming request error', { - partCount, - textChunkCount, - reasoningChunkCount, - error: String(error), - }) - - if (error instanceof Error && error.name === 'AbortError') { - yield { content: '', isFinished: true, finishReason: 'stop' } - return - } - throw error - } - } - - async validateApiKey(): Promise<{ success: boolean; error?: string }> { - try { - const headers: Record = { - 'Content-Type': 'application/json', - } - if (this.apiKey && this.apiKey !== 'sk-no-key-required') { - headers['Authorization'] = `Bearer ${this.apiKey}` - } - - const response = await fetch(`${this.baseUrl}/chat/completions`, { - method: 'POST', - headers, - body: JSON.stringify({ - model: this.model, - messages: [{ role: 'user', content: 'Hi' }], - max_tokens: 1, - }), - }) - - if (response.ok) { - return { success: true } - } - - const errorText = await response.text() - let errorMessage = `HTTP ${response.status}` - try { - const errorJson = JSON.parse(errorText) as { error?: { message?: string }; message?: string } - errorMessage = errorJson.error?.message || errorJson.message || errorMessage - } catch { - if (errorText) { - errorMessage = errorText.slice(0, 200) - } - } - - if (response.status === 401 || response.status === 403) { - return { success: false, error: errorMessage } - } - - return { success: true } - } catch (error) { - const errorMessage = error instanceof Error ? error.message : String(error) - return { success: false, error: errorMessage } - } - } -} diff --git a/electron/main/ai/llm/sdkUtils.ts b/electron/main/ai/llm/sdkUtils.ts deleted file mode 100644 index 392835f..0000000 --- a/electron/main/ai/llm/sdkUtils.ts +++ /dev/null @@ -1,150 +0,0 @@ -/** - * AI SDK 适配通用工具 - */ - -import { jsonSchema } from 'ai' -import type { ContentPart, FinishReason, LanguageModelUsage, ModelMessage, ToolSet, TypedToolCall, JSONValue } from 'ai' -import type { ChatMessage, ChatResponse, ToolCall, ToolDefinition } from './types' - -const UNKNOWN_TOOL_NAME = 'unknown_tool' - -/** - * 将 OpenAI 风格的工具定义转换为 AI SDK ToolSet - */ -export function buildToolSet(tools?: ToolDefinition[]): ToolSet | undefined { - if (!tools || tools.length === 0) return undefined - - const toolSet: ToolSet = {} - for (const tool of tools) { - toolSet[tool.function.name] = { - description: tool.function.description, - inputSchema: jsonSchema(tool.function.parameters), - } - } - - return toolSet -} - -/** - * 解析 JSON 字符串，失败返回 null - */ -export function safeParseJson(value: string): JSONValue | null { - try { - return JSON.parse(value) as JSONValue - } catch { - return null - } -} - -/** - * 将 AI SDK finishReason 映射为现有的 LLM finishReason - */ -export function mapFinishReason(reason: FinishReason): ChatResponse['finishReason'] { - if (reason === 'stop') return 'stop' - if (reason === 'length') return 'length' - if (reason === 'tool-calls') return 'tool_calls' - return 'error' -} - -/** - * 将 AI SDK usage 映射为现有的 usage 结构 - */ -export function mapUsage(usage?: LanguageModelUsage): ChatResponse['usage'] | undefined { - if (!usage) return undefined - - const promptTokens = usage.inputTokens - const completionTokens = usage.outputTokens - const totalTokens = usage.totalTokens - - if (promptTokens == null && completionTokens == null && totalTokens == null) { - return undefined - } - - return { - promptTokens: promptTokens ?? 0, - completionTokens: completionTokens ?? 0, - totalTokens: totalTokens ?? 0, - } -} - -/** - * 将 AI SDK toolCalls 映射为现有 ToolCall 结构 - */ -export function mapToolCalls(toolCalls: TypedToolCall[]): ToolCall[] { - return toolCalls.map((tc) => ({ - id: tc.toolCallId, - type: 'function' as const, - function: { - name: tc.toolName, - arguments: JSON.stringify(tc.input ?? {}), - }, - })) -} - -/** - * 将现有 ChatMessage 转为 AI SDK 的 ModelMessage - */ -export function buildModelMessages(messages: ChatMessage[]): ModelMessage[] { - const toolCallNameMap = new Map() - - // 先建立 tool_call_id 到 toolName 的映射，供 tool 消息使用 - for (const message of messages) { - if (message.role !== 'assistant' || !message.tool_calls) continue - for (const toolCall of message.tool_calls) { - toolCallNameMap.set(toolCall.id, toolCall.function.name) - } - } - - return messages.map((message) => { - if (message.role === 'assistant') { - if (message.tool_calls && message.tool_calls.length > 0) { - const contentParts: Array> = [] - - if (message.content) { - contentParts.push({ type: 'text', text: message.content }) - } - - for (const toolCall of message.tool_calls) { - const input = safeParseJson(toolCall.function.arguments || '{}') ?? {} - contentParts.push({ - type: 'tool-call', - toolCallId: toolCall.id, - toolName: toolCall.function.name, - input, - }) - } - - return { - role: 'assistant', - content: contentParts, - } - } - - return { role: 'assistant', content: message.content } - } - - if (message.role === 'tool') { - const toolCallId = message.tool_call_id || '' - const toolName = toolCallNameMap.get(toolCallId) || UNKNOWN_TOOL_NAME - const parsed = safeParseJson(message.content) - - return { - role: 'tool', - content: [ - { - type: 'tool-result', - toolCallId, - toolName, - // 工具结果只允许 JSON/文本，保持与旧实现一致 - output: parsed ? { type: 'json', value: parsed } : { type: 'text', value: message.content }, - }, - ], - } - } - - return { - role: message.role, - content: message.content, - } - }) -} diff --git a/electron/main/ai/llm/types.ts b/electron/main/ai/llm/types.ts index 601f5d3..c86599a 100644 --- a/electron/main/ai/llm/types.ts +++ b/electron/main/ai/llm/types.ts @@ -7,162 +7,6 @@ */ export type LLMProvider = 'deepseek' | 'qwen' | 'minimax' | 'glm' | 'kimi' | 'gemini' | 'doubao' | 'openai-compatible' -/** - * LLM 配置 - */ -export interface LLMConfig { - provider: LLMProvider - apiKey: string - model?: string - baseUrl?: string - maxTokens?: number -} - -/** - * 聊天消息 - */ -export interface ChatMessage { - role: 'system' | 'user' | 'assistant' | 'tool' - content: string - /** tool 角色时的 tool_call_id */ - tool_call_id?: string - /** assistant 角色时的 tool_calls */ - tool_calls?: ToolCall[] -} - -/** - * 聊天请求选项 - */ -export interface ChatOptions { - temperature?: number - maxTokens?: number - stream?: boolean - /** 可用的工具列表 */ - tools?: ToolDefinition[] - /** 中止信号，用于取消请求 */ - abortSignal?: AbortSignal -} - -/** - * 非流式响应 - */ -export interface ChatResponse { - content: string - finishReason: 'stop' | 'length' | 'error' | 'tool_calls' - /** 如果 LLM 决定调用工具，返回 tool_calls */ - tool_calls?: ToolCall[] - usage?: { - promptTokens: number - completionTokens: number - totalTokens: number - } -} - -/** - * 流式响应 chunk - */ -export interface ChatStreamChunk { - content: string - isFinished: boolean - finishReason?: 'stop' | 'length' | 'error' | 'tool_calls' - /** 流式过程中的 tool_calls（增量） */ - tool_calls?: ToolCall[] - /** Token 使用量（通常在最后一个 chunk 中返回） */ - usage?: { - promptTokens: number - completionTokens: number - totalTokens: number - } -} - -// ==================== Function Calling 相关类型 ==================== - -/** - * 工具定义（OpenAI 兼容格式） - */ -export interface ToolDefinition { - type: 'function' - function: { - name: string - description: string - parameters: { - type: 'object' - properties: Record< - string, - { - type: string - description: string - enum?: string[] - items?: { type: string } - } - > - required?: string[] - } - } -} - -/** - * 工具调用 - */ -export interface ToolCall { - id: string - type: 'function' - function: { - name: string - arguments: string // JSON 字符串 - } - /** Gemini 3+ 模型需要的思考签名（用于工具调用验证） */ - thoughtSignature?: string -} - -/** - * 工具调用结果 - */ -export interface ToolCallResult { - tool_call_id: string - result: unknown -} - -/** - * LLM 服务接口 - */ -export interface ILLMService { - /** - * 获取提供商名称 - */ - getProvider(): LLMProvider - - /** - * 获取可用模型列表 - */ - getModels(): string[] - - /** - * 获取默认模型 - */ - getDefaultModel(): string - - /** - * 发送聊天请求（非流式） - * @param messages 消息列表 - * @param options 选项，可包含 tools 参数启用 Function Calling - */ - chat(messages: ChatMessage[], options?: ChatOptions): Promise - - /** - * 发送聊天请求（流式） - * @param messages 消息列表 - * @param options 选项，可包含 tools 参数启用 Function Calling - */ - chatStream(messages: ChatMessage[], options?: ChatOptions): AsyncGenerator - - /** - * 验证 API Key 是否有效 - * @returns 验证结果和可能的错误信息 - */ - validateApiKey(): Promise<{ success: boolean; error?: string }> -} - /** * 提供商信息 */ diff --git a/package.json b/package.json index ba0ab98..f685931 100644 --- a/package.json +++ b/package.json @@ -30,8 +30,6 @@ "postinstall": "electron-rebuild" }, "dependencies": { - "@ai-sdk/google": "^3.0.10", - "@ai-sdk/openai": "^3.0.12", "@aptabase/electron": "^0.3.1", "@electron-toolkit/preload": "^3.0.1", "@electron-toolkit/utils": "^4.0.0", @@ -41,7 +39,6 @@ "@tanstack/vue-virtual": "^3.13.18", "@types/markdown-it": "^14.1.2", "@zumer/snapdom": "^2.0.1", - "ai": "^6.0.41", "better-sqlite3": "^12.4.6", "echarts": "^6.0.0", "echarts-wordcloud": "^2.1.0",