Function createCliRunAgentStream

apps/cli/src/ai/agent-stream-runner.ts:65–227

(
  dbManager: DatabaseManager,
  aiChatManager: AIChatManager,
  semanticIndexService?: SemanticIndexRuntime
)

}

export function createCliRunAgentStream(
  dbManager: DatabaseManager,
  aiChatManager: AIChatManager,
  semanticIndexService?: SemanticIndexRuntime
): (params: AgentStreamRequest, onEvent: (chunk: AgentStreamChunk) => 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

    const aiDataDir = getAiDir(dbManager)

    let assistantSystemPrompt: string | undefined
    let assistantAllowedTools: string[] | undefined
    if (assistantId) {
      const assistantConfig = loadAssistantConfig(aiDataDir, assistantId)
      if (assistantConfig?.systemPrompt) {
        assistantSystemPrompt = assistantConfig.systemPrompt
      }
      assistantAllowedTools = assistantConfig?.allowedBuiltinTools
    }

    const llmConfig = getDefaultAssistantConfig(aiDataDir)
    const maxToolResultPercent = compressionConfig?.maxToolResultPercent ?? 50
    const contextWindow = llmConfig ? (buildPiModel(llmConfig).contextWindow ?? 128000) : 128000
    const maxToolResultTokens = Math.floor(contextWindow * (maxToolResultPercent / 100))

    const db = (dbManager as any).open?.(sessionId)
    const resolvedChartAutoMode: ChartAutoMode = chartAutoMode ?? 'suggest'
    const chartRuntime = resolveChartRuntimeForRequest({
      skillId,
      userMessage,
      locale,
      assistantAllowedTools,
      enableAutoDetection: enableAutoSkill === true,
      chartAutoMode: resolvedChartAutoMode,
    })
    const isChartCapability = chartRuntime.isChartCapability
    const autoSkillAllowedTools =
      !skillId && enableAutoSkill && resolvedChartAutoMode !== 'explicit'
        ? getAllowedBuiltinToolsForChartAutoSkill(assistantAllowedTools)
        : assistantAllowedTools
    const allowedToolSet = getAllowedToolSet(
      isChartCapability,
      isChartCapability ? chartRuntime.allowedBuiltinTools : autoSkillAllowedTools