Function createElectronRunAgentStream

apps/desktop/main/ai/agent-stream-runner.ts:65–300

(
  semanticIndexService?: SemanticIndexRuntime
)

}

export function createElectronRunAgentStream(
  semanticIndexService?: SemanticIndexRuntime
): (
  params: AgentStreamRequest,
  onEvent: (chunk: SharedAgentStreamChunk) => void,
  abortSignal: AbortSignal
) => Promise<void> {
  return async (params, onEvent, abortSignal) => {
    const {
      userMessage,
      aiChatId,
      historyLeafMessageId,
      sessionId,
      chatType,
      locale,
      assistantId,
      skillId,
      enableAutoSkill,
      chartAutoMode,
      compressionConfig,
      ownerInfo,
      mentionedMembers,
      thinkingLevel,
    } = params

    // 确保 tokenizer rank 表已加载（compression + agent 路径均依赖）
    await initTokenizer()

    const requestId = `internal_${Date.now()}`

    aiLogger.info('AgentStream', `Agent stream request: ${requestId}`, {
      userMessage: userMessage.slice(0, 100),
      sessionId,
      aiChatId,
      chatType: chatType ?? 'group',
      assistantId: assistantId ?? '(none)',
      skillId: skillId ?? '(none)',
      enableAutoSkill: enableAutoSkill ?? false,
    })

    const activeAIConfig = getDefaultAssistantConfig()
    if (!activeAIConfig) {
      onEvent({ type: 'error', error: { name: 'ConfigError', message: t('llm.notConfigured') } })
      return
    }
    const piModel = buildPiModel(activeAIConfig)

    if (compressionConfig?.enabled && aiChatId && historyLeafMessageId === undefined) {
      try {
        const tempAssistantConfig = assistantId
          ? (assistantManager.getAssistantConfig(assistantId) ?? undefined)
          : undefined
        const systemPromptForCompression = tempAssistantConfig?.systemPrompt || ''

        const compressionResult = await checkAndCompress(
          aiChatId,
          compressionConfig as CompressionConfig,
          systemPromptForCompression,