@huyooo/ai-chat-bridge-electron 0.2.14 → 0.2.16

package/dist/main/index.js

@@ -1,843 +1 @@
-// src/main/index.ts
-import { ipcMain as ipcMain2, shell } from "electron";
-import * as fs from "fs";
-import * as path from "path";
-import {
-  HybridAgent,
-  MODELS,
-  DEFAULT_MODEL
-} from "@huyooo/ai-chat-core";
-import { createSearchElectronBridge } from "@huyooo/ai-search/bridge/electron";
-import {
-  createStorage
-} from "@huyooo/ai-chat-storage";
-
-// src/main/asr/protocol.ts
-import * as zlib from "zlib";
-var PROTOCOL_VERSION = 1;
-var HEADER_SIZE = 1;
-var MessageType = {
-  /** 客户端发送：包含请求参数的完整请求 */
-  FULL_CLIENT_REQUEST: 1,
-  /** 客户端发送：仅包含音频数据 */
-  AUDIO_ONLY_REQUEST: 2,
-  /** 服务端响应：包含识别结果 */
-  FULL_SERVER_RESPONSE: 9,
-  /** 服务端响应：错误消息 */
-  ERROR_RESPONSE: 15
-};
-var MessageFlags = {
-  /** 无特殊标志 */
-  NONE: 0,
-  /** header 后 4 字节为正 sequence number */
-  HAS_SEQUENCE: 1,
-  /** 最后一包（负包） */
-  LAST_PACKET: 2,
-  /** 最后一包且有 sequence number */
-  LAST_PACKET_WITH_SEQUENCE: 3
-};
-var SerializationMethod = {
-  NONE: 0,
-  JSON: 1
-};
-var CompressionMethod = {
-  NONE: 0,
-  GZIP: 1
-};
-function buildHeader(messageType, messageFlags, serialization, compression) {
-  return [
-    PROTOCOL_VERSION << 4 | HEADER_SIZE,
-    messageType << 4 | messageFlags,
-    serialization << 4 | compression,
-    0
-    // reserved
-  ];
-}
-function encodeFullClientRequest(config, useGzip = true) {
-  const header = buildHeader(
-    MessageType.FULL_CLIENT_REQUEST,
-    MessageFlags.NONE,
-    SerializationMethod.JSON,
-    useGzip ? CompressionMethod.GZIP : CompressionMethod.NONE
-  );
-  const jsonPayload = JSON.stringify(config);
-  const payloadBuffer = Buffer.from(jsonPayload, "utf-8");
-  const compressed = useGzip ? zlib.gzipSync(payloadBuffer) : payloadBuffer;
-  const payloadSize = Buffer.alloc(4);
-  payloadSize.writeUInt32BE(compressed.length, 0);
-  return Buffer.concat([
-    Buffer.from(header),
-    payloadSize,
-    compressed
-  ]);
-}
-function encodeAudioOnlyRequest(audioData, isLast = false, useGzip = true) {
-  const header = buildHeader(
-    MessageType.AUDIO_ONLY_REQUEST,
-    isLast ? MessageFlags.LAST_PACKET : MessageFlags.NONE,
-    SerializationMethod.NONE,
-    useGzip ? CompressionMethod.GZIP : CompressionMethod.NONE
-  );
-  const compressed = useGzip ? zlib.gzipSync(audioData) : audioData;
-  const payloadSize = Buffer.alloc(4);
-  payloadSize.writeUInt32BE(compressed.length, 0);
-  return Buffer.concat([
-    Buffer.from(header),
-    payloadSize,
-    compressed
-  ]);
-}
-function decodeServerResponse(data) {
-  if (data.length < 4) {
-    throw new Error("Invalid response: too short");
-  }
-  const byte0 = data[0];
-  const byte1 = data[1];
-  const byte2 = data[2];
-  const version = byte0 >> 4 & 15;
-  const headerSize = (byte0 & 15) * 4;
-  const messageType = byte1 >> 4 & 15;
-  const messageFlags = byte1 & 15;
-  const serialization = byte2 >> 4 & 15;
-  const compression = byte2 & 15;
-  console.log("[ASR Protocol] \u89E3\u6790\u54CD\u5E94:", {
-    version,
-    headerSize,
-    messageType: messageType.toString(2).padStart(4, "0"),
-    messageFlags: messageFlags.toString(2).padStart(4, "0"),
-    serialization,
-    compression,
-    dataLength: data.length,
-    headerHex: data.slice(0, 4).toString("hex")
-  });
-  if (version !== PROTOCOL_VERSION) {
-    throw new Error(`Unsupported protocol version: ${version}`);
-  }
-  let offset = headerSize;
-  let sequence;
-  const hasSequence = (messageFlags & 1) === 1;
-  if (hasSequence && data.length >= offset + 4) {
-    sequence = data.readUInt32BE(offset);
-    offset += 4;
-    console.log("[ASR Protocol] sequence:", sequence);
-  }
-  const isLast = (messageFlags & 2) === 2;
-  if (messageType === MessageType.ERROR_RESPONSE) {
-    const errorCode = data.readUInt32BE(offset);
-    offset += 4;
-    const errorMsgSize = data.readUInt32BE(offset);
-    offset += 4;
-    const errorMsg = data.slice(offset, offset + errorMsgSize).toString("utf-8");
-    console.log("[ASR Protocol] \u9519\u8BEF\u54CD\u5E94:", { errorCode, errorMsg });
-    return {
-      type: "error",
-      sequence,
-      isLast: true,
-      data: { code: errorCode, message: errorMsg }
-    };
-  }
-  if (messageType === MessageType.FULL_SERVER_RESPONSE) {
-    const payloadSize = data.readUInt32BE(offset);
-    offset += 4;
-    console.log("[ASR Protocol] payloadSize:", payloadSize, "offset:", offset, "remaining:", data.length - offset);
-    if (payloadSize === 0) {
-      return {
-        type: "result",
-        sequence,
-        isLast,
-        data: {}
-      };
-    }
-    let payload = data.slice(offset, offset + payloadSize);
-    if (compression === CompressionMethod.GZIP) {
-      try {
-        payload = zlib.gunzipSync(payload);
-      } catch (e) {
-        console.error("[ASR Protocol] gzip \u89E3\u538B\u5931\u8D25:", e);
-        console.log("[ASR Protocol] \u539F\u59CB payload \u524D 32 \u5B57\u8282:", payload.slice(0, 32).toString("hex"));
-        throw e;
-      }
-    }
-    console.log("[ASR Protocol] \u89E3\u538B\u540E payload \u957F\u5EA6:", payload.length);
-    let result;
-    if (serialization === SerializationMethod.JSON) {
-      const jsonStr = payload.toString("utf-8");
-      console.log("[ASR Protocol] JSON \u5B57\u7B26\u4E32\u524D 200 \u5B57\u7B26:", jsonStr.slice(0, 200));
-      try {
-        result = JSON.parse(jsonStr);
-      } catch (e) {
-        console.error("[ASR Protocol] JSON \u89E3\u6790\u5931\u8D25:", e);
-        console.log("[ASR Protocol] \u5B8C\u6574 JSON:", jsonStr);
-        throw e;
-      }
-    } else {
-      result = {};
-    }
-    return {
-      type: "result",
-      sequence,
-      isLast,
-      data: result
-    };
-  }
-  throw new Error(`Unknown message type: ${messageType}`);
-}
-
-// src/main/asr/client.ts
-import WebSocket from "ws";
-import { v4 as uuidv4 } from "uuid";
-var AsrClient = class {
-  config;
-  ws = null;
-  callbacks = {};
-  connectId = "";
-  isConnected = false;
-  sessionConfig = {};
-  constructor(config) {
-    this.config = config;
-  }
-  /**
-   * 获取 WebSocket 连接地址
-   */
-  getWsUrl() {
-    if (this.config.useAsyncMode !== false) {
-      return "wss://openspeech.bytedance.com/api/v3/sauc/bigmodel_async";
-    }
-    return "wss://openspeech.bytedance.com/api/v3/sauc/bigmodel";
-  }
-  /**
-   * 获取资源 ID
-   */
-  getResourceId() {
-    return this.config.resourceId || "volc.bigasr.sauc.duration";
-  }
-  /**
-   * 连接到 ASR 服务
-   */
-  connect(callbacks, sessionConfig = {}) {
-    return new Promise((resolve2, reject) => {
-      this.callbacks = callbacks;
-      this.sessionConfig = sessionConfig;
-      this.connectId = uuidv4();
-      const wsUrl = this.getWsUrl();
-      this.ws = new WebSocket(wsUrl, {
-        headers: {
-          "X-Api-App-Key": this.config.appId,
-          "X-Api-Access-Key": this.config.accessKey,
-          "X-Api-Resource-Id": this.getResourceId(),
-          "X-Api-Connect-Id": this.connectId
-        }
-      });
-      this.ws.on("open", () => {
-        console.log("[ASR] WebSocket \u8FDE\u63A5\u6210\u529F, connectId:", this.connectId);
-        this.isConnected = true;
-        this.sendFullClientRequest();
-        this.callbacks.onConnected?.();
-        resolve2();
-      });
-      this.ws.on("message", (data) => {
-        try {
-          const response = decodeServerResponse(data);
-          if (response.type === "error") {
-            const errorData = response.data;
-            console.error("[ASR] \u670D\u52A1\u7AEF\u9519\u8BEF:", errorData);
-            this.callbacks.onError?.(new Error(`ASR Error ${errorData.code}: ${errorData.message}`));
-          } else {
-            const result = response.data;
-            this.callbacks.onResult?.(result, response.isLast);
-            if (response.isLast) {
-              console.log("[ASR] \u6536\u5230\u6700\u7EC8\u7ED3\u679C");
-            }
-          }
-        } catch (error) {
-          console.error("[ASR] \u89E3\u6790\u54CD\u5E94\u5931\u8D25:", error);
-          this.callbacks.onError?.(error instanceof Error ? error : new Error(String(error)));
-        }
-      });
-      this.ws.on("error", (error) => {
-        console.error("[ASR] WebSocket \u9519\u8BEF:", error);
-        this.callbacks.onError?.(error);
-        reject(error);
-      });
-      this.ws.on("close", () => {
-        console.log("[ASR] WebSocket \u8FDE\u63A5\u5173\u95ED");
-        this.isConnected = false;
-        this.ws = null;
-        this.callbacks.onClose?.();
-      });
-    });
-  }
-  /**
-   * 发送首包（Full Client Request）
-   */
-  sendFullClientRequest() {
-    if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-      console.error("[ASR] WebSocket \u672A\u8FDE\u63A5");
-      return;
-    }
-    const config = {
-      user: {
-        uid: "ai-chat-user"
-      },
-      audio: {
-        format: this.sessionConfig.format || "pcm",
-        rate: this.sessionConfig.sampleRate || 16e3,
-        bits: 16,
-        channel: 1
-      },
-      request: {
-        model_name: "bigmodel",
-        enable_itn: this.sessionConfig.enableItn ?? true,
-        enable_punc: this.sessionConfig.enablePunc ?? true,
-        enable_ddc: this.sessionConfig.enableDdc ?? false,
-        show_utterances: this.sessionConfig.showUtterances ?? true,
-        result_type: "full"
-      }
-    };
-    const packet = encodeFullClientRequest(config);
-    this.ws.send(packet);
-    console.log("[ASR] \u5DF2\u53D1\u9001\u9996\u5305");
-  }
-  /**
-   * 发送音频数据
-   * @param audioData PCM 音频数据（16bit, 16kHz, 单声道）
-   */
-  sendAudio(audioData) {
-    if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-      console.error("[ASR] WebSocket \u672A\u8FDE\u63A5\uFF0C\u65E0\u6CD5\u53D1\u9001\u97F3\u9891");
-      return;
-    }
-    const packet = encodeAudioOnlyRequest(audioData, false);
-    this.ws.send(packet);
-  }
-  /**
-   * 发送最后一包并结束识别
-   */
-  finish() {
-    if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-      console.error("[ASR] WebSocket \u672A\u8FDE\u63A5\uFF0C\u65E0\u6CD5\u7ED3\u675F");
-      return;
-    }
-    const packet = encodeAudioOnlyRequest(Buffer.alloc(0), true);
-    this.ws.send(packet);
-    console.log("[ASR] \u5DF2\u53D1\u9001\u7ED3\u675F\u5305");
-  }
-  /**
-   * 断开连接
-   */
-  disconnect() {
-    if (this.ws) {
-      this.ws.close();
-      this.ws = null;
-    }
-    this.isConnected = false;
-  }
-  /**
-   * 检查是否已连接
-   */
-  get connected() {
-    return this.isConnected && this.ws?.readyState === WebSocket.OPEN;
-  }
-};
-
-// src/main/asr/index.ts
-import { ipcMain } from "electron";
-function createAsrBridge(config) {
-  const { channelPrefix = "ai-chat", appId, accessKey, resourceId } = config;
-  const sessions = /* @__PURE__ */ new Map();
-  function getOrCreateSession(webContentsId, webContents) {
-    let session = sessions.get(webContentsId);
-    if (!session) {
-      const clientConfig = {
-        appId,
-        accessKey,
-        resourceId,
-        useAsyncMode: true
-        // 使用双向流式优化版
-      };
-      const client = new AsrClient(clientConfig);
-      session = { client, webContents };
-      sessions.set(webContentsId, session);
-      webContents.on("destroyed", () => {
-        const s = sessions.get(webContentsId);
-        if (s) {
-          s.client.disconnect();
-          sessions.delete(webContentsId);
-        }
-      });
-    }
-    return session;
-  }
-  ipcMain.handle(`${channelPrefix}:asr:start`, async (event, sessionConfig) => {
-    const webContents = event.sender;
-    const webContentsId = webContents.id;
-    const session = getOrCreateSession(webContentsId, webContents);
-    if (session.client.connected) {
-      session.client.disconnect();
-    }
-    try {
-      await session.client.connect({
-        onConnected: () => {
-          if (!webContents.isDestroyed()) {
-            webContents.send(`${channelPrefix}:asr:connected`);
-          }
-        },
-        onResult: (result, isLast) => {
-          if (!webContents.isDestroyed()) {
-            webContents.send(`${channelPrefix}:asr:result`, { result, isLast });
-          }
-        },
-        onError: (error) => {
-          if (!webContents.isDestroyed()) {
-            webContents.send(`${channelPrefix}:asr:error`, { message: error.message });
-          }
-        },
-        onClose: () => {
-          if (!webContents.isDestroyed()) {
-            webContents.send(`${channelPrefix}:asr:closed`);
-          }
-        }
-      }, sessionConfig);
-      return { success: true };
-    } catch (error) {
-      console.error("[ASR Bridge] \u542F\u52A8\u5931\u8D25:", error);
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : String(error)
-      };
-    }
-  });
-  ipcMain.handle(`${channelPrefix}:asr:sendAudio`, async (event, audioData) => {
-    const webContentsId = event.sender.id;
-    const session = sessions.get(webContentsId);
-    if (!session || !session.client.connected) {
-      return { success: false, error: "ASR \u4F1A\u8BDD\u672A\u542F\u52A8" };
-    }
-    try {
-      const payload = audioData instanceof ArrayBuffer ? Buffer.from(audioData) : Buffer.from(new Uint8Array(audioData));
-      session.client.sendAudio(payload);
-      return { success: true };
-    } catch (error) {
-      console.error("[ASR Bridge] \u53D1\u9001\u97F3\u9891\u5931\u8D25:", error);
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : String(error)
-      };
-    }
-  });
-  ipcMain.handle(`${channelPrefix}:asr:finish`, async (event) => {
-    const webContentsId = event.sender.id;
-    const session = sessions.get(webContentsId);
-    if (!session || !session.client.connected) {
-      return { success: false, error: "ASR \u4F1A\u8BDD\u672A\u542F\u52A8" };
-    }
-    try {
-      session.client.finish();
-      return { success: true };
-    } catch (error) {
-      console.error("[ASR Bridge] \u7ED3\u675F\u4F1A\u8BDD\u5931\u8D25:", error);
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : String(error)
-      };
-    }
-  });
-  ipcMain.handle(`${channelPrefix}:asr:stop`, async (event) => {
-    const webContentsId = event.sender.id;
-    const session = sessions.get(webContentsId);
-    if (session) {
-      session.client.disconnect();
-    }
-    return { success: true };
-  });
-  ipcMain.handle(`${channelPrefix}:asr:status`, async (event) => {
-    const webContentsId = event.sender.id;
-    const session = sessions.get(webContentsId);
-    return {
-      connected: session?.client.connected ?? false
-    };
-  });
-  ipcMain.handle(`${channelPrefix}:asr:warmup`, async (event, sessionConfig) => {
-    const webContents = event.sender;
-    const webContentsId = webContents.id;
-    const session = getOrCreateSession(webContentsId, webContents);
-    if (session.client.connected) return { success: true };
-    try {
-      await session.client.connect(
-        {
-          onConnected: () => {
-          },
-          onResult: () => {
-          },
-          onError: () => {
-          },
-          onClose: () => {
-          }
-        },
-        sessionConfig
-      );
-      session.client.disconnect();
-      return { success: true };
-    } catch (error) {
-      console.warn("[ASR Bridge] warmup \u5931\u8D25:", error);
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : String(error)
-      };
-    }
-  });
-  console.log("[ASR Bridge] IPC handlers \u5DF2\u6CE8\u518C");
-  return {
-    /** 清理所有会话 */
-    cleanup: () => {
-      for (const session of sessions.values()) {
-        session.client.disconnect();
-      }
-      sessions.clear();
-    }
-  };
-}
-
-// src/main/index.ts
-async function createElectronBridge(options) {
-  const {
-    channelPrefix = "ai-chat",
-    dataDir,
-    defaultContext = {},
-    ...agentConfig
-  } = options;
-  const resolvedStoragePath = path.join(dataDir, "db.sqlite");
-  createSearchElectronBridge({
-    ipcMain: ipcMain2,
-    channelPrefix
-  }).init();
-  const pendingApprovals = /* @__PURE__ */ new Map();
-  const toolApprovalCallback = async (toolCall) => {
-    console.log("[Main] \u5DE5\u5177\u6279\u51C6\u56DE\u8C03\u88AB\u8C03\u7528:", toolCall.name);
-    const currentWebContents = global.currentWebContents;
-    if (!currentWebContents) {
-      console.log("[Main] \u8B66\u544A: \u6CA1\u6709 webContents\uFF0C\u9ED8\u8BA4\u6279\u51C6");
-      return true;
-    }
-    return new Promise((resolve2, reject) => {
-      pendingApprovals.set(toolCall.id, {
-        resolve: resolve2,
-        reject,
-        webContents: currentWebContents
-      });
-      console.log("[Main] \u53D1\u9001\u5DE5\u5177\u6279\u51C6\u8BF7\u6C42\u5230\u524D\u7AEF:", toolCall.name);
-      currentWebContents.send(`${channelPrefix}:toolApprovalRequest`, {
-        id: toolCall.id,
-        name: toolCall.name,
-        args: toolCall.args
-      });
-    });
-  };
-  const storage = await createStorage({
-    type: "sqlite",
-    sqlitePath: resolvedStoragePath
-  });
-  const getContext = () => defaultContext;
-  const getAutoRunConfig = async () => {
-    try {
-      const configJson = await storage.getUserSetting("autoRunConfig", getContext());
-      if (configJson) {
-        return JSON.parse(configJson);
-      }
-    } catch (error) {
-      console.error("[Main] \u83B7\u53D6 autoRunConfig \u5931\u8D25:", error);
-    }
-    return void 0;
-  };
-  const agent = new HybridAgent({
-    ...agentConfig,
-    // tools 直接传递 ToolConfigItem[]，HybridAgent 会在 asyncInit 中解析
-    tools: options.tools,
-    onToolApprovalRequest: toolApprovalCallback,
-    getAutoRunConfig
-  });
-  ipcMain2.handle(`${channelPrefix}:models`, () => {
-    return MODELS;
-  });
-  ipcMain2.handle(`${channelPrefix}:getAllTools`, async () => {
-    console.log("[Main] getAllTools \u8C03\u7528\uFF0C\u5F53\u524D\u5DE5\u5177\u6570\u91CF:", agent["tools"].size);
-    console.log("[Main] toolConfig \u662F\u5426\u5B58\u5728:", !!agent["toolConfig"]);
-    console.log("[Main] toolConfig \u5185\u5BB9:", agent["toolConfig"] ? `${agent["toolConfig"].length} \u4E2A\u5DE5\u5177\u914D\u7F6E` : "undefined");
-    if (agent["tools"].size === 0 && agent["toolConfig"]) {
-      console.log("[Main] \u5DE5\u5177\u672A\u521D\u59CB\u5316\uFF0C\u5F00\u59CB asyncInit...");
-      await agent["asyncInit"]();
-      console.log("[Main] asyncInit \u5B8C\u6210\uFF0C\u5DE5\u5177\u6570\u91CF:", agent["tools"].size);
-    } else if (!agent["toolConfig"]) {
-      console.warn("[Main] \u26A0\uFE0F toolConfig \u4E0D\u5B58\u5728\uFF0C\u5DE5\u5177\u53EF\u80FD\u672A\u6CE8\u5165\uFF01");
-    }
-    const tools = agent.getAllTools();
-    console.log("[Main] getAllTools \u8FD4\u56DE:", tools.length, "\u4E2A\u5DE5\u5177", tools.map((t) => t.name));
-    return tools;
-  });
-  ipcMain2.handle(`${channelPrefix}:send`, async (event, params) => {
-    const webContents = event.sender;
-    const { message, images, options: options2 = {}, sessionId } = params;
-    global.currentWebContents = webContents;
-    console.log("[AI-Chat] \u6536\u5230\u6D88\u606F:", { message, options: options2, images: images?.length || 0, sessionId });
-    console.log("[AI-Chat] autoRunConfig:", JSON.stringify(options2.autoRunConfig, null, 2));
-    try {
-      for await (const chatEvent of agent.chat(message, options2, images)) {
-        console.log(
-          "[AI-Chat] \u4E8B\u4EF6:",
-          chatEvent.type,
-          chatEvent.type === "text_delta" ? chatEvent.data.content.slice(0, 20) : chatEvent.type === "thinking_delta" ? chatEvent.data.content.slice(0, 20) : chatEvent.type === "search_result" ? `\u641C\u7D22\u5B8C\u6210 ${chatEvent.data.results?.length || 0} \u6761` : JSON.stringify(chatEvent.data).slice(0, 100)
-        );
-        if (!webContents.isDestroyed()) {
-          webContents.send(`${channelPrefix}:progress`, { ...chatEvent, sessionId });
-        }
-      }
-      console.log("[AI-Chat] \u5B8C\u6210");
-    } catch (error) {
-      console.error("[AI-Chat] \u9519\u8BEF:", error);
-      if (error instanceof Error) {
-        console.error("[AI-Chat] \u9519\u8BEF\u8BE6\u60C5:", {
-          message: error.message,
-          stack: error.stack,
-          name: error.name
-        });
-      }
-      if (!webContents.isDestroyed()) {
-        const errorData = error instanceof Error ? {
-          category: "api",
-          message: error.message || String(error),
-          cause: error.stack
-        } : {
-          category: "api",
-          message: String(error)
-        };
-        webContents.send(`${channelPrefix}:progress`, {
-          type: "error",
-          data: errorData,
-          sessionId
-        });
-      }
-    } finally {
-      delete global.currentWebContents;
-    }
-  });
-  ipcMain2.handle(`${channelPrefix}:toolApprovalResponse`, (_event, params) => {
-    const { id, approved } = params;
-    const pending = pendingApprovals.get(id);
-    if (pending) {
-      pendingApprovals.delete(id);
-      pending.resolve(approved);
-    }
-  });
-  ipcMain2.handle(`${channelPrefix}:cancel`, () => {
-    agent.abort();
-  });
-  ipcMain2.handle(`${channelPrefix}:settings:get`, async (_event, key) => {
-    return storage.getUserSetting(key, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:settings:set`, async (_event, key, value) => {
-    await storage.setUserSetting(key, value, getContext());
-    return { success: true };
-  });
-  ipcMain2.handle(`${channelPrefix}:settings:getAll`, async () => {
-    return storage.getUserSettings(getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:settings:delete`, async (_event, key) => {
-    await storage.deleteUserSetting(key, getContext());
-    return { success: true };
-  });
-  ipcMain2.handle(`${channelPrefix}:setCwd`, (_event, dir) => {
-    agent.setCwd(dir);
-  });
-  ipcMain2.handle(`${channelPrefix}:config`, () => {
-    return agent.getConfig();
-  });
-  ipcMain2.handle(`${channelPrefix}:sessions:list`, async () => {
-    return storage.getSessions(getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:sessions:get`, async (_event, id) => {
-    return storage.getSession(id, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:sessions:create`, async (_event, params) => {
-    const input = {
-      id: params.id || crypto.randomUUID(),
-      title: params.title || "\u65B0\u5BF9\u8BDD",
-      model: params.model || DEFAULT_MODEL,
-      mode: params.mode || "agent",
-      webSearchEnabled: params.webSearchEnabled ?? true,
-      thinkingEnabled: params.thinkingEnabled ?? true,
-      hidden: params.hidden ?? false
-    };
-    return storage.createSession(input, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:sessions:update`, async (_event, id, data) => {
-    await storage.updateSession(id, data, getContext());
-    return storage.getSession(id, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:sessions:delete`, async (_event, id) => {
-    await storage.deleteSession(id, getContext());
-    return { success: true };
-  });
-  ipcMain2.handle(`${channelPrefix}:messages:list`, async (_event, sessionId) => {
-    return storage.getMessages(sessionId, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:messages:save`, async (_event, params) => {
-    const input = {
-      id: params.id || crypto.randomUUID(),
-      sessionId: params.sessionId,
-      role: params.role,
-      content: params.content,
-      images: params.images || [],
-      model: params.model || null,
-      mode: params.mode || null,
-      webSearchEnabled: params.webSearchEnabled ?? null,
-      thinkingEnabled: params.thinkingEnabled ?? null,
-      steps: params.steps || null,
-      operationIds: params.operationIds || null
-    };
-    return storage.saveMessage(input, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:messages:update`, async (_event, params) => {
-    await storage.updateMessage(params.id, {
-      content: params.content,
-      steps: params.steps
-    }, getContext());
-    return { success: true };
-  });
-  ipcMain2.handle(`${channelPrefix}:messages:deleteAfter`, async (_event, sessionId, timestamp) => {
-    await storage.deleteMessagesAfter(sessionId, new Date(timestamp), getContext());
-    return { success: true };
-  });
-  ipcMain2.handle(`${channelPrefix}:messages:deleteAfterMessageId`, async (_event, sessionId, messageId) => {
-    await storage.deleteMessagesAfterMessageId(sessionId, messageId, getContext());
-    return { success: true };
-  });
-  ipcMain2.handle(`${channelPrefix}:operations:list`, async (_event, sessionId) => {
-    return storage.getOperations(sessionId, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:trash:list`, async () => {
-    return storage.getTrashItems?.(getContext()) || [];
-  });
-  ipcMain2.handle(`${channelPrefix}:trash:restore`, async (_event, id) => {
-    return storage.restoreFromTrash?.(id, getContext());
-  });
-  ipcMain2.handle(`${channelPrefix}:openExternal`, async (_event, url) => {
-    return shell.openExternal(url);
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:listDir`, async (_event, dirPath) => {
-    try {
-      const entries = fs.readdirSync(dirPath, { withFileTypes: true });
-      const files = [];
-      for (const entry of entries) {
-        if (entry.name.startsWith(".")) continue;
-        const fullPath = path.join(dirPath, entry.name);
-        try {
-          const stats = fs.statSync(fullPath);
-          files.push({
-            name: entry.name,
-            path: fullPath,
-            isDirectory: entry.isDirectory(),
-            size: stats.size,
-            modifiedAt: stats.mtime,
-            extension: entry.isDirectory() ? "" : path.extname(entry.name).toLowerCase()
-          });
-        } catch {
-        }
-      }
-      return files.sort((a, b) => {
-        if (a.isDirectory && !b.isDirectory) return -1;
-        if (!a.isDirectory && b.isDirectory) return 1;
-        return a.name.localeCompare(b.name);
-      });
-    } catch (error) {
-      console.error("[AI-Chat] \u5217\u51FA\u76EE\u5F55\u5931\u8D25:", error);
-      return [];
-    }
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:exists`, async (_event, filePath) => {
-    return fs.existsSync(filePath);
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:stat`, async (_event, filePath) => {
-    try {
-      const stats = fs.statSync(filePath);
-      return {
-        name: path.basename(filePath),
-        path: filePath,
-        isDirectory: stats.isDirectory(),
-        size: stats.size,
-        modifiedAt: stats.mtime,
-        extension: stats.isDirectory() ? "" : path.extname(filePath).toLowerCase()
-      };
-    } catch {
-      return null;
-    }
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:readFile`, async (_event, filePath) => {
-    try {
-      return fs.readFileSync(filePath, "utf-8");
-    } catch {
-      return null;
-    }
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:readFileBase64`, async (_event, filePath) => {
-    try {
-      const buffer = fs.readFileSync(filePath);
-      return buffer.toString("base64");
-    } catch {
-      return null;
-    }
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:homeDir`, async () => {
-    return process.env.HOME || process.env.USERPROFILE || "/";
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:resolvePath`, async (_event, inputPath) => {
-    if (inputPath.startsWith("~")) {
-      const homeDir = process.env.HOME || process.env.USERPROFILE || "/";
-      return path.join(homeDir, inputPath.slice(1));
-    }
-    return path.resolve(inputPath);
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:parentDir`, async (_event, dirPath) => {
-    return path.dirname(dirPath);
-  });
-  const activeWatchers = /* @__PURE__ */ new Map();
-  ipcMain2.handle(`${channelPrefix}:fs:watchDir`, async (event, dirPath) => {
-    const webContents = event.sender;
-    if (activeWatchers.has(dirPath)) {
-      activeWatchers.get(dirPath)?.close();
-      activeWatchers.delete(dirPath);
-    }
-    try {
-      const watcher = fs.watch(dirPath, { persistent: false }, (eventType, filename) => {
-        if (!webContents.isDestroyed()) {
-          webContents.send(`${channelPrefix}:fs:dirChange`, {
-            dirPath,
-            eventType,
-            filename
-          });
-        }
-      });
-      watcher.on("error", (error) => {
-        console.error("[AI-Chat] Watch error:", error);
-        activeWatchers.delete(dirPath);
-      });
-      activeWatchers.set(dirPath, watcher);
-      return true;
-    } catch (error) {
-      console.error("[AI-Chat] Failed to watch directory:", error);
-      return false;
-    }
-  });
-  ipcMain2.handle(`${channelPrefix}:fs:unwatchDir`, async (_event, dirPath) => {
-    const watcher = activeWatchers.get(dirPath);
-    if (watcher) {
-      watcher.close();
-      activeWatchers.delete(dirPath);
-    }
-  });
-  return { agent, storage };
-}
-export {
-  MODELS,
-  createAsrBridge,
-  createElectronBridge
-};
+import{ipcMain as e,shell as s}from"electron";import*as t from"fs";import*as n from"path";import{HybridAgent as r,MODELS as o,DEFAULT_MODEL as a}from"@huyooo/ai-chat-core";import{createSearchElectronBridge as c}from"@huyooo/ai-search/bridge/electron";import{createStorage as i}from"@huyooo/ai-chat-storage";import*as l from"zlib";var d=1,u=2,h=9,f=15,g=0,y=2,p=0,m=1,w=0,$=1;function b(e,s,t,n){return[17,e<<4|s,t<<4|n,0]}function S(e,s=!1,t=!0){const n=b(u,s?y:g,p,t?$:w),r=t?l.gzipSync(e):e,o=Buffer.alloc(4);return o.writeUInt32BE(r.length,0),Buffer.concat([Buffer.from(n),o,r])}import E from"ws";import{v4 as C}from"uuid";var I=class{config;ws=null;callbacks={};connectId="";isConnected=!1;sessionConfig={};constructor(e){this.config=e}getWsUrl(){return!1!==this.config.useAsyncMode?"wss://openspeech.bytedance.com/api/v3/sauc/bigmodel_async":"wss://openspeech.bytedance.com/api/v3/sauc/bigmodel"}getResourceId(){return this.config.resourceId||"volc.bigasr.sauc.duration"}connect(e,s={}){return new Promise((t,n)=>{this.callbacks=e,this.sessionConfig=s,this.connectId=C();const r=this.getWsUrl();this.ws=new E(r,{headers:{"X-Api-App-Key":this.config.appId,"X-Api-Access-Key":this.config.accessKey,"X-Api-Resource-Id":this.getResourceId(),"X-Api-Connect-Id":this.connectId}}),this.ws.on("open",()=>{this.isConnected=!0,this.sendFullClientRequest(),this.callbacks.onConnected?.(),t()}),this.ws.on("message",e=>{try{const s=function(e){if(e.length<4)throw new Error("Invalid response: too short");const s=e[0],t=e[1],n=e[2],r=s>>4&15,o=4*(15&s),a=t>>4&15,c=15&t,i=n>>4&15,d=15&n;if(1!==r)throw new Error(`Unsupported protocol version: ${r}`);let u,g=o;!(1&~c)&&e.length>=g+4&&(u=e.readUInt32BE(g),g+=4);const y=!(2&~c);if(a===f){const s=e.readUInt32BE(g);g+=4;const t=e.readUInt32BE(g);return g+=4,{type:"error",sequence:u,isLast:!0,data:{code:s,message:e.slice(g,g+t).toString("utf-8")}}}if(a===h){const s=e.readUInt32BE(g);if(g+=4,0===s)return{type:"result",sequence:u,isLast:y,data:{}};let t,n=e.slice(g,g+s);if(d===$)try{n=l.gunzipSync(n)}catch(e){throw e}if(i===m){const e=n.toString("utf-8");try{t=JSON.parse(e)}catch(e){throw e}}else t={};return{type:"result",sequence:u,isLast:y,data:t}}throw new Error(`Unknown message type: ${a}`)}(e);if("error"===s.type){const e=s.data;this.callbacks.onError?.(new Error(`ASR Error ${e.code}: ${e.message}`))}else{const e=s.data;this.callbacks.onResult?.(e,s.isLast),s.isLast}}catch(e){this.callbacks.onError?.(e instanceof Error?e:new Error(String(e)))}}),this.ws.on("error",e=>{this.callbacks.onError?.(e),n(e)}),this.ws.on("close",()=>{this.isConnected=!1,this.ws=null,this.callbacks.onClose?.()})})}sendFullClientRequest(){if(!this.ws||this.ws.readyState!==E.OPEN)return;const e=function(e,s=!0){const t=b(d,g,m,s?$:w),n=JSON.stringify(e),r=Buffer.from(n,"utf-8"),o=s?l.gzipSync(r):r,a=Buffer.alloc(4);return a.writeUInt32BE(o.length,0),Buffer.concat([Buffer.from(t),a,o])}({user:{uid:"ai-chat-user"},audio:{format:this.sessionConfig.format||"pcm",rate:this.sessionConfig.sampleRate||16e3,bits:16,channel:1},request:{model_name:"bigmodel",enable_itn:this.sessionConfig.enableItn??!0,enable_punc:this.sessionConfig.enablePunc??!0,enable_ddc:this.sessionConfig.enableDdc??!1,show_utterances:this.sessionConfig.showUtterances??!0,result_type:"full"}});this.ws.send(e)}sendAudio(e){if(!this.ws||this.ws.readyState!==E.OPEN)return;const s=S(e,!1);this.ws.send(s)}finish(){if(!this.ws||this.ws.readyState!==E.OPEN)return;const e=S(Buffer.alloc(0),!0);this.ws.send(e)}disconnect(){this.ws&&(this.ws.close(),this.ws=null),this.isConnected=!1}get connected(){return this.isConnected&&this.ws?.readyState===E.OPEN}};import{ipcMain as A}from"electron";function D(e){const{channelPrefix:s="ai-chat",appId:t,accessKey:n,resourceId:r}=e,o=new Map;function a(e,s){let a=o.get(e);if(!a){a={client:new I({appId:t,accessKey:n,resourceId:r,useAsyncMode:!0}),webContents:s},o.set(e,a),s.on("destroyed",()=>{const s=o.get(e);s&&(s.client.disconnect(),o.delete(e))})}return a}return A.handle(`${s}:asr:start`,async(e,t)=>{const n=e.sender,r=a(n.id,n);r.client.connected&&r.client.disconnect();try{return await r.client.connect({onConnected:()=>{n.isDestroyed()||n.send(`${s}:asr:connected`)},onResult:(e,t)=>{n.isDestroyed()||n.send(`${s}:asr:result`,{result:e,isLast:t})},onError:e=>{n.isDestroyed()||n.send(`${s}:asr:error`,{message:e.message})},onClose:()=>{n.isDestroyed()||n.send(`${s}:asr:closed`)}},t),{success:!0}}catch(e){return{success:!1,error:e instanceof Error?e.message:String(e)}}}),A.handle(`${s}:asr:sendAudio`,async(e,s)=>{const t=e.sender.id,n=o.get(t);if(!n||!n.client.connected)return{success:!1,error:"ASR 会话未启动"};try{const e=s instanceof ArrayBuffer?Buffer.from(s):Buffer.from(new Uint8Array(s));return n.client.sendAudio(e),{success:!0}}catch(e){return{success:!1,error:e instanceof Error?e.message:String(e)}}}),A.handle(`${s}:asr:finish`,async e=>{const s=e.sender.id,t=o.get(s);if(!t||!t.client.connected)return{success:!1,error:"ASR 会话未启动"};try{return t.client.finish(),{success:!0}}catch(e){return{success:!1,error:e instanceof Error?e.message:String(e)}}}),A.handle(`${s}:asr:stop`,async e=>{const s=e.sender.id,t=o.get(s);return t&&t.client.disconnect(),{success:!0}}),A.handle(`${s}:asr:status`,async e=>{const s=e.sender.id,t=o.get(s);return{connected:t?.client.connected??!1}}),A.handle(`${s}:asr:warmup`,async(e,s)=>{const t=e.sender,n=a(t.id,t);if(n.client.connected)return{success:!0};try{return await n.client.connect({onConnected:()=>{},onResult:()=>{},onError:()=>{},onClose:()=>{}},s),n.client.disconnect(),{success:!0}}catch(e){return{success:!1,error:e instanceof Error?e.message:String(e)}}}),{cleanup:()=>{for(const e of o.values())e.client.disconnect();o.clear()}}}async function v(l){const{channelPrefix:d="ai-chat",dataDir:u,defaultContext:h={},...f}=l,g=n.join(u,"db.sqlite");c({ipcMain:e,channelPrefix:d}).init();const y=new Map,p=await i({type:"sqlite",sqlitePath:g}),m=()=>h,w=new r({...f,tools:l.tools,onToolApprovalRequest:async e=>{const s=global.currentWebContents;return!s||new Promise((t,n)=>{y.set(e.id,{resolve:t,reject:n,webContents:s}),s.send(`${d}:toolApprovalRequest`,{id:e.id,name:e.name,args:e.args})})},getAutoRunConfig:async()=>{try{const e=await p.getUserSetting("autoRunConfig",m());if(e)return JSON.parse(e)}catch(e){}}});e.handle(`${d}:models`,()=>o),e.handle(`${d}:getAllTools`,async()=>{0===w.tools.size&&w.toolConfig?await w.asyncInit():w.toolConfig;return w.getAllTools()}),e.handle(`${d}:send`,async(e,s)=>{const t=e.sender,{message:n,images:r,options:o={},sessionId:a}=s;global.currentWebContents=t;try{for await(const e of w.chat(n,o,r))t.isDestroyed()||t.send(`${d}:progress`,{...e,sessionId:a})}catch(e){if(Error,!t.isDestroyed()){const s=e instanceof Error?{category:"api",message:e.message||String(e),cause:e.stack}:{category:"api",message:String(e)};t.send(`${d}:progress`,{type:"error",data:s,sessionId:a})}}finally{delete global.currentWebContents}}),e.handle(`${d}:toolApprovalResponse`,(e,s)=>{const{id:t,approved:n}=s,r=y.get(t);r&&(y.delete(t),r.resolve(n))}),e.handle(`${d}:cancel`,()=>{w.abort()}),e.handle(`${d}:settings:get`,async(e,s)=>p.getUserSetting(s,m())),e.handle(`${d}:settings:set`,async(e,s,t)=>(await p.setUserSetting(s,t,m()),{success:!0})),e.handle(`${d}:settings:getAll`,async()=>p.getUserSettings(m())),e.handle(`${d}:settings:delete`,async(e,s)=>(await p.deleteUserSetting(s,m()),{success:!0})),e.handle(`${d}:setCwd`,(e,s)=>{w.setCwd(s)}),e.handle(`${d}:config`,()=>w.getConfig()),e.handle(`${d}:sessions:list`,async()=>p.getSessions(m())),e.handle(`${d}:sessions:get`,async(e,s)=>p.getSession(s,m())),e.handle(`${d}:sessions:create`,async(e,s)=>{const t={id:s.id||crypto.randomUUID(),title:s.title||"新对话",model:s.model||a,mode:s.mode||"agent",webSearchEnabled:s.webSearchEnabled??!0,thinkingEnabled:s.thinkingEnabled??!0,hidden:s.hidden??!1};return p.createSession(t,m())}),e.handle(`${d}:sessions:update`,async(e,s,t)=>(await p.updateSession(s,t,m()),p.getSession(s,m()))),e.handle(`${d}:sessions:delete`,async(e,s)=>(await p.deleteSession(s,m()),{success:!0})),e.handle(`${d}:messages:list`,async(e,s)=>p.getMessages(s,m())),e.handle(`${d}:messages:save`,async(e,s)=>{const t={id:s.id||crypto.randomUUID(),sessionId:s.sessionId,role:s.role,content:s.content,images:s.images||[],model:s.model||null,mode:s.mode||null,webSearchEnabled:s.webSearchEnabled??null,thinkingEnabled:s.thinkingEnabled??null,steps:s.steps||null,operationIds:s.operationIds||null};return p.saveMessage(t,m())}),e.handle(`${d}:messages:update`,async(e,s)=>(await p.updateMessage(s.id,{content:s.content,steps:s.steps},m()),{success:!0})),e.handle(`${d}:messages:deleteAfter`,async(e,s,t)=>(await p.deleteMessagesAfter(s,new Date(t),m()),{success:!0})),e.handle(`${d}:messages:deleteAfterMessageId`,async(e,s,t)=>(await p.deleteMessagesAfterMessageId(s,t,m()),{success:!0})),e.handle(`${d}:operations:list`,async(e,s)=>p.getOperations(s,m())),e.handle(`${d}:trash:list`,async()=>p.getTrashItems?.(m())||[]),e.handle(`${d}:trash:restore`,async(e,s)=>p.restoreFromTrash?.(s,m())),e.handle(`${d}:openExternal`,async(e,t)=>s.openExternal(t)),e.handle(`${d}:fs:listDir`,async(e,s)=>{try{const e=t.readdirSync(s,{withFileTypes:!0}),r=[];for(const o of e){if(o.name.startsWith("."))continue;const e=n.join(s,o.name);try{const s=t.statSync(e);r.push({name:o.name,path:e,isDirectory:o.isDirectory(),size:s.size,modifiedAt:s.mtime,extension:o.isDirectory()?"":n.extname(o.name).toLowerCase()})}catch{}}return r.sort((e,s)=>e.isDirectory&&!s.isDirectory?-1:!e.isDirectory&&s.isDirectory?1:e.name.localeCompare(s.name))}catch(e){return[]}}),e.handle(`${d}:fs:exists`,async(e,s)=>t.existsSync(s)),e.handle(`${d}:fs:stat`,async(e,s)=>{try{const e=t.statSync(s);return{name:n.basename(s),path:s,isDirectory:e.isDirectory(),size:e.size,modifiedAt:e.mtime,extension:e.isDirectory()?"":n.extname(s).toLowerCase()}}catch{return null}}),e.handle(`${d}:fs:readFile`,async(e,s)=>{try{return t.readFileSync(s,"utf-8")}catch{return null}}),e.handle(`${d}:fs:readFileBase64`,async(e,s)=>{try{return t.readFileSync(s).toString("base64")}catch{return null}}),e.handle(`${d}:fs:homeDir`,async()=>process.env.HOME||process.env.USERPROFILE||"/"),e.handle(`${d}:fs:resolvePath`,async(e,s)=>{if(s.startsWith("~")){const e=process.env.HOME||process.env.USERPROFILE||"/";return n.join(e,s.slice(1))}return n.resolve(s)}),e.handle(`${d}:fs:parentDir`,async(e,s)=>n.dirname(s));const $=new Map;return e.handle(`${d}:fs:watchDir`,async(e,s)=>{const n=e.sender;$.has(s)&&($.get(s)?.close(),$.delete(s));try{const e=t.watch(s,{persistent:!1},(e,t)=>{n.isDestroyed()||n.send(`${d}:fs:dirChange`,{dirPath:s,eventType:e,filename:t})});return e.on("error",e=>{$.delete(s)}),$.set(s,e),!0}catch(e){return!1}}),e.handle(`${d}:fs:unwatchDir`,async(e,s)=>{const t=$.get(s);t&&(t.close(),$.delete(s))}),{agent:w,storage:p}}export{o as MODELS,D as createAsrBridge,v as createElectronBridge};