mirror of
https://github.com/hellodigua/ChatLab.git
synced 2026-05-16 11:29:24 +08:00
digua
60 lines
2.1 KiB
TypeScript
60 lines
2.1 KiB
TypeScript
/**
|
|
* Electron Preload Script
|
|
* 将主进程 API 暴露给渲染进程
|
|
*/
|
|
import { contextBridge } from 'electron'
|
|
import { electronAPI } from '@electron-toolkit/preload'
|
|
|
|
// 从拆分的模块导入 API
|
|
import { extendedApi } from './apis/core'
|
|
import { chatApi, mergeApi } from './apis/chat'
|
|
import { aiApi, llmApi, agentApi, embeddingApi } from './apis/ai'
|
|
import { nlpApi, networkApi, cacheApi, sessionApi } from './apis/utils'
|
|
|
|
// Use `contextBridge` APIs to expose Electron APIs to
|
|
// renderer only if context isolation is enabled, otherwise
|
|
// just add to the DOM global.
|
|
if (process.contextIsolated) {
|
|
try {
|
|
contextBridge.exposeInMainWorld('electron', electronAPI)
|
|
contextBridge.exposeInMainWorld('api', extendedApi)
|
|
contextBridge.exposeInMainWorld('chatApi', chatApi)
|
|
contextBridge.exposeInMainWorld('mergeApi', mergeApi)
|
|
contextBridge.exposeInMainWorld('aiApi', aiApi)
|
|
contextBridge.exposeInMainWorld('llmApi', llmApi)
|
|
contextBridge.exposeInMainWorld('agentApi', agentApi)
|
|
contextBridge.exposeInMainWorld('embeddingApi', embeddingApi)
|
|
contextBridge.exposeInMainWorld('cacheApi', cacheApi)
|
|
contextBridge.exposeInMainWorld('networkApi', networkApi)
|
|
contextBridge.exposeInMainWorld('sessionApi', sessionApi)
|
|
contextBridge.exposeInMainWorld('nlpApi', nlpApi)
|
|
} catch (error) {
|
|
console.error(error)
|
|
}
|
|
} else {
|
|
// @ts-ignore (define in dts)
|
|
window.electron = electronAPI
|
|
// @ts-ignore (define in dts)
|
|
window.api = extendedApi
|
|
// @ts-ignore (define in dts)
|
|
window.chatApi = chatApi
|
|
// @ts-ignore (define in dts)
|
|
window.mergeApi = mergeApi
|
|
// @ts-ignore (define in dts)
|
|
window.aiApi = aiApi
|
|
// @ts-ignore (define in dts)
|
|
window.llmApi = llmApi
|
|
// @ts-ignore (define in dts)
|
|
window.agentApi = agentApi
|
|
// @ts-ignore (define in dts)
|
|
window.embeddingApi = embeddingApi
|
|
// @ts-ignore (define in dts)
|
|
window.cacheApi = cacheApi
|
|
// @ts-ignore (define in dts)
|
|
window.networkApi = networkApi
|
|
// @ts-ignore (define in dts)
|
|
window.sessionApi = sessionApi
|
|
// @ts-ignore (define in dts)
|
|
window.nlpApi = nlpApi
|
|
}
|