Function generateSessionSummary

packages/node-runtime/src/ai/summary/index.ts:235–317

(
  deps: SummaryDeps,
  segmentId: number,
  options: SummaryOptions = {}
)

// ==================== Public API ====================

export async function generateSessionSummary(
  deps: SummaryDeps,
  segmentId: number,
  options: SummaryOptions = {}
): Promise<SummaryResult> {
  const { locale = 'zh-CN', forceRegenerate = false, strategy = 'standard' } = options
  const log = deps.logger

  try {
    if (!forceRegenerate) {
      const existing = deps.getSummary(segmentId)
      if (existing) return { success: true, summary: existing }
    }

    const rawMessages = deps.loadMessages(segmentId)
    if (!rawMessages) {
      return { success: false, error: deps.t('summary.sessionNotFound') }
    }

    if (rawMessages.length < MIN_MESSAGE_COUNT) {
      return { success: false, error: deps.t('summary.tooFewMessages', { count: MIN_MESSAGE_COUNT }) }
    }

    const validMessages = filterValidMessages(rawMessages)
    if (validMessages.length < MIN_MESSAGE_COUNT) {
      return { success: false, error: deps.t('summary.tooFewValidMessages', { count: MIN_MESSAGE_COUNT }) }
    }

    const lengthLimit = getSummaryLengthLimit(validMessages.length, strategy)
    const maxTokens = strategy === 'brief' ? 300 : 600
    const subMaxTokens = strategy === 'brief' ? 100 : 200
    const content = formatMessages(validMessages)

    log?.info(
      'Summary',
      `Generating summary: sessionId=${segmentId}, strategy=${strategy}, raw=${rawMessages.length}, valid=${validMessages.length}, chars=${content.length}`
    )

    let summary: string
    if (content.length <= SEGMENT_THRESHOLD) {
      summary = await deps.llmComplete(
        deps.t('summary.systemPromptDirect'),
        buildSummaryPrompt(content, lengthLimit, locale, strategy),
        { temperature: 0.3, maxTokens }
      )
      summary = summary.trim()
    } else {
      const segments = splitIntoSegments(validMessages, MAX_CONTENT_PER_CALL)
      log?.info('Summary', `Long session segmented: ${segments.length} segments`)

      const subSummaries: string[] = []
      for (const segment of segments) {
        const segContent = formatMessages(segment)
        const sub = await deps.llmComplete(
          deps.t('summary.systemPromptDirect'),
          buildSubSummaryPrompt(segContent, locale, strategy),
          { temperature: 0.3, maxTokens: subMaxTokens }
        )