ChatLab/apps/desktop/preload/apis/ai.ts

/**
 * 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.
}