Method SummarizeSession

pkg/session/summary.go:57–117

SummarizeSession 生成 Session 摘要

(ctx context.Context, messages []types.Message)

// SummarizeSession 生成 Session 摘要
func (s *Summarizer) SummarizeSession(ctx context.Context, messages []types.Message) (*SessionSummary, error) {
	if len(messages) < s.minMessages {
		return nil, fmt.Errorf("not enough messages to summarize (need at least %d)", s.minMessages)
	}

	// 限制消息数量
	if len(messages) > s.maxMessages {
		messages = messages[len(messages)-s.maxMessages:]
	}

	// 构建摘要提示词
	prompt := s.buildSummaryPrompt(messages)

	// 调用 Provider 生成摘要
	summaryMessages := []types.Message{
		{
			Role: types.MessageRoleUser,
			ContentBlocks: []types.ContentBlock{
				&types.TextBlock{Text: prompt},
			},
		},
	}

	// 临时设置系统提示词
	originalSysPrompt := ""
	if caps := s.provider.Capabilities(); caps.SupportSystemPrompt {
		// 保存原始系统提示词（如果需要恢复）
		if err := s.provider.SetSystemPrompt(s.sysPrompt); err != nil {
			return nil, fmt.Errorf("set system prompt: %w", err)
		}
	}

	response, err := s.provider.Complete(ctx, summaryMessages, nil)
	if err != nil {
		return nil, fmt.Errorf("generate summary: %w", err)
	}

	// 恢复原始系统提示词
	if originalSysPrompt != "" {
		_ = s.provider.SetSystemPrompt(originalSysPrompt)
	}

	// 提取文本内容
	summaryText := ""
	if len(response.Message.ContentBlocks) > 0 {
		if textBlock, ok := response.Message.ContentBlocks[0].(*types.TextBlock); ok {
			summaryText = textBlock.Text
		}
	}

	summary := &SessionSummary{
		Text:         summaryText,
		MessageCount: len(messages),
		GeneratedAt:  time.Now(),
		TokensUsed:   int(response.Usage.TotalTokens),
		KeyTopics:    extractKeyTopics(summaryText),
		Participants: extractParticipants(messages),
	}