mirror of
https://github.com/hellodigua/ChatLab.git
synced 2026-10-26 02:04:37 +08:00
62 lines
1.8 KiB
TypeScript
62 lines
1.8 KiB
TypeScript
/**
|
|
* AI 相关 API — 仅保留 IPC 必须的能力
|
|
*
|
|
* 此处只保留需要 worker、native shell、工具注册表等 IPC 才能提供的功能。
|
|
*/
|
|
import { ipcRenderer } from 'electron'
|
|
import type { ExportProgress } from '../../../../src/types/base'
|
|
|
|
export type { TokenUsage, AgentRuntimeStatus, SerializedErrorInfo } from '../../shared/types'
|
|
|
|
// ==================== 类型定义 ====================
|
|
|
|
export interface DesensitizeRule {
|
|
id: string
|
|
label: string
|
|
pattern: string
|
|
replacement: string
|
|
enabled: boolean
|
|
builtin: boolean
|
|
locales: string[]
|
|
group?: string
|
|
}
|
|
|
|
export interface PreprocessConfig {
|
|
dataCleaning: boolean
|
|
mergeConsecutive: boolean
|
|
mergeWindowSeconds?: number
|
|
blacklistKeywords: string[]
|
|
denoise: boolean
|
|
desensitize: boolean
|
|
desensitizeRulesSchemaVersion?: number
|
|
desensitizeBuiltinRuleOverrides?: Record<string, boolean>
|
|
desensitizeRules: DesensitizeRule[]
|
|
anonymizeNames: boolean
|
|
}
|
|
|
|
// ==================== AI API (IPC-only subset) ====================
|
|
|
|
export const aiApi = {
|
|
exportFilterResultToFile: (params: {
|
|
sessionId: string
|
|
sessionName: string
|
|
outputDir: string
|
|
format?: 'txt' | 'json' | 'markdown'
|
|
timeFilter?: { startTs: number; endTs: number }
|
|
}): Promise<{ success: boolean; filePath?: string; error?: string }> => {
|
|
return ipcRenderer.invoke('ai:exportFilterResultToFile', params)
|
|
},
|
|
|
|
onExportProgress: (callback: (progress: ExportProgress) => void) => {
|
|
const handler = (_event: Electron.IpcRendererEvent, progress: ExportProgress) => {
|
|
callback(progress)
|
|
}
|
|
ipcRenderer.on('ai:exportProgress', handler)
|
|
return () => {
|
|
ipcRenderer.removeListener('ai:exportProgress', handler)
|
|
}
|
|
},
|
|
|
|
// Desensitize rules, tool testing, estimateContextTokens have been migrated to shared HTTP routes.
|
|
}
|