+
+
diff --git a/electron/main.ts b/electron/main.ts
index 718579f..13ed5e4 100644
--- a/electron/main.ts
+++ b/electron/main.ts
@@ -21,7 +21,7 @@ import { videoService } from './services/videoService'
import { snsService, isVideoUrl } from './services/snsService'
import { contactExportService } from './services/contactExportService'
import { windowsHelloService } from './services/windowsHelloService'
-import { llamaService } from './services/llamaService'
+
import { registerNotificationHandlers, showNotification } from './windows/notificationWindow'
import { httpService } from './services/httpService'
@@ -825,63 +825,6 @@ function registerIpcHandlers() {
return await chatService.getContact(username)
})
- // Llama AI
- ipcMain.handle('llama:init', async () => {
- return await llamaService.init()
- })
-
- ipcMain.handle('llama:loadModel', async (_, modelPath: string) => {
- return llamaService.loadModel(modelPath)
- })
-
- ipcMain.handle('llama:createSession', async (_, systemPrompt?: string) => {
- return llamaService.createSession(systemPrompt)
- })
-
- ipcMain.handle('llama:chat', async (event, message: string, options?: { thinking?: boolean }) => {
- // We use a callback to stream back to the renderer
- const webContents = event.sender
- try {
- if (!webContents) return { success: false, error: 'No sender' }
-
- const response = await llamaService.chat(message, options, (token) => {
- if (!webContents.isDestroyed()) {
- webContents.send('llama:token', token)
- }
- })
- return { success: true, response }
- } catch (e) {
- return { success: false, error: String(e) }
- }
- })
-
- ipcMain.handle('llama:downloadModel', async (event, url: string, savePath: string) => {
- const webContents = event.sender
- try {
- await llamaService.downloadModel(url, savePath, (payload) => {
- if (!webContents.isDestroyed()) {
- webContents.send('llama:downloadProgress', payload)
- }
- })
- return { success: true }
- } catch (e) {
- return { success: false, error: String(e) }
- }
- })
-
- ipcMain.handle('llama:getModelsPath', async () => {
- return llamaService.getModelsPath()
- })
-
- ipcMain.handle('llama:checkFileExists', async (_, filePath: string) => {
- const { existsSync } = await import('fs')
- return existsSync(filePath)
- })
-
- ipcMain.handle('llama:getModelStatus', async (_, modelPath: string) => {
- return llamaService.getModelStatus(modelPath)
- })
-
ipcMain.handle('chat:getContactAvatar', async (_, username: string) => {
return await chatService.getContactAvatar(username)
@@ -951,6 +894,10 @@ function registerIpcHandlers() {
return snsService.getTimeline(limit, offset, usernames, keyword, startTime, endTime)
})
+ ipcMain.handle('sns:getSnsUsernames', async () => {
+ return snsService.getSnsUsernames()
+ })
+
ipcMain.handle('sns:debugResource', async (_, url: string) => {
return snsService.debugResource(url)
})
diff --git a/electron/preload.ts b/electron/preload.ts
index 3adee1c..b67d65a 100644
--- a/electron/preload.ts
+++ b/electron/preload.ts
@@ -276,6 +276,7 @@ contextBridge.exposeInMainWorld('electronAPI', {
sns: {
getTimeline: (limit: number, offset: number, usernames?: string[], keyword?: string, startTime?: number, endTime?: number) =>
ipcRenderer.invoke('sns:getTimeline', limit, offset, usernames, keyword, startTime, endTime),
+ getSnsUsernames: () => ipcRenderer.invoke('sns:getSnsUsernames'),
debugResource: (url: string) => ipcRenderer.invoke('sns:debugResource', url),
proxyImage: (payload: { url: string; key?: string | number }) => ipcRenderer.invoke('sns:proxyImage', payload),
downloadImage: (payload: { url: string; key?: string | number }) => ipcRenderer.invoke('sns:downloadImage', payload),
@@ -287,27 +288,6 @@ contextBridge.exposeInMainWorld('electronAPI', {
selectExportDir: () => ipcRenderer.invoke('sns:selectExportDir')
},
- // Llama AI
- llama: {
- loadModel: (modelPath: string) => ipcRenderer.invoke('llama:loadModel', modelPath),
- createSession: (systemPrompt?: string) => ipcRenderer.invoke('llama:createSession', systemPrompt),
- chat: (message: string, options?: any) => ipcRenderer.invoke('llama:chat', message, options),
- downloadModel: (url: string, savePath: string) => ipcRenderer.invoke('llama:downloadModel', url, savePath),
- getModelsPath: () => ipcRenderer.invoke('llama:getModelsPath'),
- checkFileExists: (filePath: string) => ipcRenderer.invoke('llama:checkFileExists', filePath),
- getModelStatus: (modelPath: string) => ipcRenderer.invoke('llama:getModelStatus', modelPath),
- onToken: (callback: (token: string) => void) => {
- const listener = (_: any, token: string) => callback(token)
- ipcRenderer.on('llama:token', listener)
- return () => ipcRenderer.removeListener('llama:token', listener)
- },
- onDownloadProgress: (callback: (payload: { downloaded: number; total: number; speed: number }) => void) => {
- const listener = (_: any, payload: { downloaded: number; total: number; speed: number }) => callback(payload)
- ipcRenderer.on('llama:downloadProgress', listener)
- return () => ipcRenderer.removeListener('llama:downloadProgress', listener)
- }
- },
-
// HTTP API 服务
http: {
start: (port?: number) => ipcRenderer.invoke('http:start', port),
diff --git a/electron/services/llamaService.ts b/electron/services/llamaService.ts
deleted file mode 100644
index 11426b8..0000000
--- a/electron/services/llamaService.ts
+++ /dev/null
@@ -1,371 +0,0 @@
-import fs from "fs";
-import { app, BrowserWindow } from "electron";
-import path from "path";
-import { ConfigService } from './config';
-
-// Define interfaces locally to avoid static import of types that might not be available or cause issues
-type LlamaModel = any;
-type LlamaContext = any;
-type LlamaChatSession = any;
-
-export class LlamaService {
- private _model: LlamaModel | null = null;
- private _context: LlamaContext | null = null;
- private _sequence: any = null;
- private _session: LlamaChatSession | null = null;
- private _llama: any = null;
- private _nodeLlamaCpp: any = null;
- private configService = new ConfigService();
- private _initialized = false;
-
- constructor() {
- // 延迟初始化，只在需要时初始化
- }
-
- public async init() {
- if (this._initialized) return;
-
- try {
- // Dynamic import to handle ESM module in CJS context
- this._nodeLlamaCpp = await import("node-llama-cpp");
- this._llama = await this._nodeLlamaCpp.getLlama();
- this._initialized = true;
- console.log("[LlamaService] Llama initialized");
- } catch (error) {
- console.error("[LlamaService] Failed to initialize Llama:", error);
- }
- }
-
- public async loadModel(modelPath: string) {
- if (!this._llama) await this.init();
-
- try {
- console.log("[LlamaService] Loading model from:", modelPath);
- if (!this._llama) {
- throw new Error("Llama not initialized");
- }
- this._model = await this._llama.loadModel({
- modelPath: modelPath,
- gpuLayers: 'max', // Offload all layers to GPU if possible
- useMlock: false // Disable mlock to avoid "VirtualLock" errors (common on Windows)
- });
-
- if (!this._model) throw new Error("Failed to load model");
-
- this._context = await this._model.createContext({
- contextSize: 8192, // Balanced context size for better performance
- batchSize: 2048 // Increase batch size for better prompt processing speed
- });
-
- if (!this._context) throw new Error("Failed to create context");
-
- this._sequence = this._context.getSequence();
-
- const { LlamaChatSession } = this._nodeLlamaCpp;
- this._session = new LlamaChatSession({
- contextSequence: this._sequence
- });
-
- console.log("[LlamaService] Model loaded successfully");
- return true;
- } catch (error) {
- console.error("[LlamaService] Failed to load model:", error);
- throw error;
- }
- }
-
- public async createSession(systemPrompt?: string) {
- if (!this._context) throw new Error("Model not loaded");
- if (!this._nodeLlamaCpp) await this.init();
-
- const { LlamaChatSession } = this._nodeLlamaCpp;
-
- if (!this._sequence) {
- this._sequence = this._context.getSequence();
- }
-
- this._session = new LlamaChatSession({
- contextSequence: this._sequence,
- systemPrompt: systemPrompt
- });
-
- return true;
- }
-
- public async chat(message: string, options: { thinking?: boolean } = {}, onToken: (token: string) => void) {
- if (!this._session) throw new Error("Session not initialized");
-
- const thinking = options.thinking ?? false;
-
- // Sampling parameters based on mode
- const samplingParams = thinking ? {
- temperature: 0.6,
- topP: 0.95,
- topK: 20,
- repeatPenalty: 1.5 // PresencePenalty=1.5
- } : {
- temperature: 0.7,
- topP: 0.8,
- topK: 20,
- repeatPenalty: 1.5
- };
-
- try {
- const response = await this._session.prompt(message, {
- ...samplingParams,
- onTextChunk: (chunk: string) => {
- onToken(chunk);
- }
- });
- return response;
- } catch (error) {
- console.error("[LlamaService] Chat error:", error);
- throw error;
- }
- }
-
- public async getModelStatus(modelPath: string) {
- try {
- const exists = fs.existsSync(modelPath);
- if (!exists) {
- return { exists: false, path: modelPath };
- }
- const stats = fs.statSync(modelPath);
- return {
- exists: true,
- path: modelPath,
- size: stats.size
- };
- } catch (error) {
- return { exists: false, error: String(error) };
- }
- }
-
- private resolveModelDir(): string {
- const configured = this.configService.get('whisperModelDir') as string | undefined;
- if (configured) return configured;
- return path.join(app.getPath('documents'), 'WeFlow', 'models');
- }
-
- public async downloadModel(url: string, savePath: string, onProgress: (payload: { downloaded: number; total: number; speed: number }) => void): Promise- {availableModels.length === 0 - ? "请先在设置页面下载模型" - : "输入消息开始对话，模型将自动加载" - } -
-