@huyooo/ai-chat-bridge-electron 0.2.13 → 0.2.15

package/src/main/index.ts

@@ -46,6 +46,8 @@ import {
 export interface ElectronBridgeOptions extends Omit<AgentConfig, 'tools'> {
   /** IPC channel 前缀 */
   channelPrefix?: string;
+  /** Tavily API Key（用于统一 Web Search；可选） */
+  tavilyApiKey?: string;
   /** 
    * 统一数据目录（必填）
    * 所有数据都存储在此目录下：
@@ -88,6 +90,8 @@ interface MessageParams {
   sessionId: string;
   role: 'user' | 'assistant';
   content: string;
+  /** 用户消息附带的图片（data URL 或 base64） */
+  images?: string[];
   /** 生成此消息时使用的模型 */
   model?: string;
   /** 生成此消息时使用的模式 (ask/agent) */
@@ -225,6 +229,28 @@ export async function createElectronBridge(options: ElectronBridgeOptions) {
     return MODELS;
   });
 
+  // 获取所有工具列表（用于设置面板）
+  ipcMain.handle(`${channelPrefix}:getAllTools`, async () => {
+    // 确保工具已初始化（通过调用 chat 方法触发 asyncInit）
+    // 如果工具还未初始化，先触发一次初始化
+    console.log('[Main] getAllTools 调用，当前工具数量:', agent['tools'].size);
+    console.log('[Main] toolConfig 是否存在:', !!agent['toolConfig']);
+    console.log('[Main] toolConfig 内容:', agent['toolConfig'] ? `${agent['toolConfig'].length} 个工具配置` : 'undefined');
+    
+    // 如果工具未初始化，尝试初始化
+    if (agent['tools'].size === 0 && agent['toolConfig']) {
+      console.log('[Main] 工具未初始化，开始 asyncInit...');
+      await agent['asyncInit']();
+      console.log('[Main] asyncInit 完成，工具数量:', agent['tools'].size);
+    } else if (!agent['toolConfig']) {
+      console.warn('[Main] ⚠️ toolConfig 不存在，工具可能未注入！');
+    }
+    
+    const tools = (agent as any).getAllTools();
+    console.log('[Main] getAllTools 返回:', tools.length, '个工具', tools.map((t: any) => t.name));
+    return tools;
+  });
+
   // 发送消息
   ipcMain.handle(`${channelPrefix}:send`, async (event, params: SendMessageParams) => {
     const webContents = event.sender;
@@ -387,6 +413,7 @@ export async function createElectronBridge(options: ElectronBridgeOptions) {
       sessionId: params.sessionId,
       role: params.role,
       content: params.content,
+      images: params.images || [],
       model: params.model || null,
       mode: params.mode || null,
       webSearchEnabled: params.webSearchEnabled ?? null,
@@ -412,6 +439,12 @@ export async function createElectronBridge(options: ElectronBridgeOptions) {
     return { success: true };
   });
 
+  // 删除指定消息之后的所有消息（用于分叉）
+  ipcMain.handle(`${channelPrefix}:messages:deleteAfterMessageId`, async (_event, sessionId: string, messageId: string) => {
+    await storage.deleteMessagesAfterMessageId(sessionId, messageId, getContext());
+    return { success: true };
+  });
+
   // 获取操作日志
   ipcMain.handle(`${channelPrefix}:operations:list`, async (_event, sessionId: string) => {
     return storage.getOperations(sessionId, getContext());

package/src/preload/index.ts

@@ -25,12 +25,14 @@ interface SendMessageParams {
 interface ModelOption {
   modelId: string;
   displayName: string;
-  /** 分组名称（由后端决定，前端只负责渲染，必填） */
-  group: string;
-  /** 是否来自 OpenRouter（保留用于兼容，后续可能移除） */
-  isOpenRouter?: boolean;
-  /** 提供商名称（保留用于兼容，后续可能移除） */
-  provider?: string;
+  supportsThinking: boolean;
+  supportsVision: boolean;
+  tooltip?: {
+    features?: string[];
+    /** 开销信息（数组，分行显示） */
+    cost?: string[];
+    description?: string;
+  };
 }
 
 /** 会话记录 */
@@ -55,6 +57,8 @@ interface MessageRecord {
   sessionId: string;
   role: 'user' | 'assistant';
   content: string;
+  /** 用户消息附带的图片（data URL 或 base64） */
+  images: string[];
   /** 生成此消息时使用的模型 */
   model?: string | null;
   /** 生成此消息时使用的模式 (ask/agent) */
@@ -66,7 +70,10 @@ interface MessageRecord {
   /** 执行步骤列表 JSON */
   steps?: string | null;
   operationIds?: string | null;
-  timestamp: Date;
+  /** 消息序号（会话内递增，用于分叉删除，必须） */
+  sequence: number;
+  /** 时间戳（毫秒，Unix 时间戳，不受 JSON 序列化影响） */
+  timestamp: number;
 }
 
 /** 操作记录 */
@@ -137,6 +144,8 @@ export interface AiChatBridge {
   // ============ Chat API ============
   /** 获取可用模型 */
   getModels(): Promise<ModelOption[]>;
+  /** 获取所有工具列表（用于设置面板） */
+  getAllTools(): Promise<Array<{ name: string; description: string }>>;
   /** 发送消息 */
   send(params: SendMessageParams): Promise<void>;
   /** 取消当前请求 */
@@ -171,6 +180,8 @@ export interface AiChatBridge {
     sessionId: string;
     role: 'user' | 'assistant';
     content: string;
+    /** 用户消息附带的图片（data URL 或 base64） */
+    images?: string[];
     /** 生成此消息时使用的模型 */
     model?: string;
     /** 生成此消息时使用的模式 (ask/agent) */
@@ -191,6 +202,8 @@ export interface AiChatBridge {
   }): Promise<{ success: boolean }>;
   /** 删除指定时间之后的消息（用于分叉） */
   deleteMessagesAfter(sessionId: string, timestamp: number): Promise<{ success: boolean }>;
+  /** 删除指定消息之后的所有消息（用于分叉） */
+  deleteMessagesAfterMessageId(sessionId: string, messageId: string): Promise<{ success: boolean }>;
 
   // ============ Operations API ============
   /** 获取操作日志 */
@@ -323,6 +336,9 @@ export function exposeElectronBridge(options: ExposeOptions = {}) {
     getModels: () => 
       ipcRenderer.invoke(`${channelPrefix}:models`),
 
+    getAllTools: () =>
+      ipcRenderer.invoke(`${channelPrefix}:getAllTools`),
+
     send: (params: SendMessageParams) => 
       ipcRenderer.invoke(`${channelPrefix}:send`, params),
 
@@ -390,6 +406,8 @@ export function exposeElectronBridge(options: ExposeOptions = {}) {
 
     deleteMessagesAfter: (sessionId: string, timestamp: number) =>
       ipcRenderer.invoke(`${channelPrefix}:messages:deleteAfter`, sessionId, timestamp),
+    deleteMessagesAfterMessageId: (sessionId: string, messageId: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:messages:deleteAfterMessageId`, sessionId, messageId),
 
     // ============ Operations API ============
     getOperations: (sessionId: string) =>

package/src/renderer/index.ts

@@ -16,6 +16,7 @@ export type ChatEventType =
   | 'search_end'
   | 'tool_approval_request'
   | 'tool_call_start'
+  | 'tool_call_output'
   | 'tool_call_result'
   | 'text_delta'
   | 'image'
@@ -69,6 +70,8 @@ export interface ChatOptions {
   enableWebSearch?: boolean;
   /** 深度思考开关（每个 provider 内部使用最优参数） */
   thinkingMode?: ThinkingMode;
+  /** 启用的工具名称列表（undefined 表示全部启用） */
+  enabledTools?: string[];
   /** 自动运行配置 */
   autoRunConfig?: AutoRunConfig;
   /** 对话历史（无状态架构，历史从前端传入） */
@@ -83,12 +86,17 @@ export interface ModelOption {
   modelId: string;
   /** 显示名称 */
   displayName: string;
-  /** 分组名称（由后端决定，前端只负责渲染，必填） */
-  group: string;
-  /** 是否来自 OpenRouter（保留用于兼容，后续可能移除） */
-  isOpenRouter?: boolean;
-  /** 提供商名称（保留用于兼容，后续可能移除） */
-  provider?: string;
+  /** 是否支持深度思考 */
+  supportsThinking: boolean;
+  /** 是否支持图片理解 */
+  supportsVision: boolean;
+  /** 模型项 hover 提示（由后端定义，前端只渲染） */
+  tooltip?: {
+    features?: string[];
+    /** 开销信息（数组，分行显示） */
+    cost?: string[];
+    description?: string;
+  };
 }
 
 // 会话记录
@@ -113,6 +121,8 @@ export interface MessageRecord {
   sessionId: string;
   role: 'user' | 'assistant';
   content: string;
+  /** 用户消息附带的图片（data URL 或 base64） */
+  images: string[];
   /** 生成此消息时使用的模型 */
   model?: string | null;
   /** 生成此消息时使用的模式 (ask/agent) */
@@ -124,7 +134,10 @@ export interface MessageRecord {
   /** 执行步骤列表 JSON */
   steps?: string | null;
   operationIds?: string | null;
-  timestamp: Date;
+  /** 消息序号（会话内递增，用于分叉删除，必须） */
+  sequence: number;
+  /** 时间戳（毫秒，Unix 时间戳，不受 JSON 序列化影响） */
+  timestamp: number;
 }
 
 // 操作记录
@@ -196,6 +209,8 @@ export interface ChatAdapter {
   // ============ Chat API ============
   /** 获取可用模型 */
   getModels(): Promise<ModelOption[]>;
+  /** 获取所有工具列表（用于设置面板） */
+  getAllTools?(): Promise<Array<{ name: string; description: string }>>;
   /** 发送消息，返回异步迭代器 */
   sendMessage(message: string, options?: ChatOptions, images?: string[], sessionId?: string): AsyncIterable<ChatEvent>;
   /** 取消当前请求 */
@@ -226,6 +241,8 @@ export interface ChatAdapter {
     sessionId: string;
     role: 'user' | 'assistant';
     content: string;
+    /** 用户消息附带的图片（data URL 或 base64） */
+    images?: string[];
     /** 生成此消息时使用的模型 */
     model?: string;
     /** 生成此消息时使用的模式 (ask/agent) */
@@ -246,6 +263,8 @@ export interface ChatAdapter {
   }): Promise<void>;
   /** 删除指定时间之后的消息（用于分叉） */
   deleteMessagesAfter(sessionId: string, timestamp: number): Promise<void>;
+  /** 删除指定消息之后的所有消息（用于分叉） */
+  deleteMessagesAfterMessageId(sessionId: string, messageId: string): Promise<void>;
 
   // ============ Operations API ============
   /** 获取操作日志 */
@@ -353,6 +372,15 @@ export function createElectronAdapter(options: CreateAdapterOptions = {}): ChatA
       return bridge.getModels();
     },
 
+    async getAllTools(): Promise<Array<{ name: string; description: string }>> {
+      const bridge = getBridge();
+      // preload 已保证存在；这里保留一层容错，避免宿主未升级时直接崩
+      if (typeof (bridge as unknown as { getAllTools?: unknown }).getAllTools === 'function') {
+        return bridge.getAllTools();
+      }
+      return [];
+    },
+
     async *sendMessage(
       message: string, 
       options?: ChatOptions,
@@ -451,6 +479,10 @@ export function createElectronAdapter(options: CreateAdapterOptions = {}): ChatA
       await getBridge().deleteMessagesAfter(sessionId, timestamp);
     },
 
+    async deleteMessagesAfterMessageId(sessionId: string, messageId: string): Promise<void> {
+      await getBridge().deleteMessagesAfterMessageId(sessionId, messageId);
+    },
+
     // ============ Operations API ============
     async getOperations(sessionId: string): Promise<OperationRecord[]> {
       return getBridge().getOperations(sessionId);