mirror of
https://github.com/hellodigua/ChatLab.git
synced 2026-10-25 17:54:34 +08:00
digua
1af644e4ab
Squash Phase 2-2 streaming migration: frontend stream service, LLM SSE route, Agent SSE route, and IPC/preload cleanup.
68 lines
2.2 KiB
TypeScript
68 lines
2.2 KiB
TypeScript
/**
|
|
* Electron Preload Script
|
|
* 将主进程 API 暴露给渲染进程
|
|
*/
|
|
import { contextBridge } from 'electron'
|
|
import { electronAPI } from '@electron-toolkit/preload'
|
|
|
|
// 从拆分的模块导入 API
|
|
import { extendedApi } from './apis/core'
|
|
import { chatApi } from './apis/chat'
|
|
import { aiApi } from './apis/ai'
|
|
import { networkApi, cacheApi, sessionApi } from './apis/utils'
|
|
import { apiServerApi } from './apis/api-server'
|
|
import { internalApi } from './apis/internal-api'
|
|
|
|
// 为渲染进程提供统一的类型入口，避免 type-only import 指向无导出的运行时代码。
|
|
export type { PreprocessConfig } from './apis/ai'
|
|
export type {
|
|
ProviderDefinition,
|
|
ProviderKind,
|
|
ModelDefinition,
|
|
ModelCapability,
|
|
ModelStatus,
|
|
ModelRecommendedFor,
|
|
LLMConnectionConfig,
|
|
LLMConnectionConfigCompat,
|
|
ModelUsage,
|
|
ModelSelectionState,
|
|
} from '../main/ai/llm/model-types'
|
|
|
|
// 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('aiApi', aiApi)
|
|
contextBridge.exposeInMainWorld('cacheApi', cacheApi)
|
|
contextBridge.exposeInMainWorld('networkApi', networkApi)
|
|
contextBridge.exposeInMainWorld('sessionApi', sessionApi)
|
|
contextBridge.exposeInMainWorld('apiServerApi', apiServerApi)
|
|
contextBridge.exposeInMainWorld('internalApi', internalApi)
|
|
} 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.aiApi = aiApi
|
|
// @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.apiServerApi = apiServerApi
|
|
// @ts-ignore (define in dts)
|
|
window.internalApi = internalApi
|
|
}
|