Method GenerateSummary

pkg/memory/session_summary.go:128–179

GenerateSummary 生成会话摘要

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

// GenerateSummary 生成会话摘要
func (m *SessionSummaryManager) GenerateSummary(
	ctx context.Context,
	sessionID string,
	messages []types.Message,
) (*SessionSummary, error) {
	if !m.config.Enabled {
		return nil, errors.New("session summary is disabled")
	}

	// 构建提示词
	prompt := m.buildPrompt(messages)

	// 调用 LLM 生成摘要
	resp, err := m.provider.Complete(ctx, []types.Message{
		{
			Role:    "user",
			Content: prompt,
		},
	}, &provider.StreamOptions{
		MaxTokens:   1000,
		Temperature: 0.3, // 较低的温度以获得更一致的输出
	})
	if err != nil {
		return nil, fmt.Errorf("failed to generate summary: %w", err)
	}

	// 解析响应
	summary, err := m.parseSummaryResponse(resp.Message.Content)
	if err != nil {
		return nil, fmt.Errorf("failed to parse summary: %w", err)
	}

	// 设置基本信息
	summary.SessionID = sessionID
	summary.CreatedAt = time.Now()
	summary.UpdatedAt = time.Now()
	summary.MessageCount = len(messages)

	// 计算 token 数量（粗略估计）
	totalTokens := 0
	for _, msg := range messages {
		totalTokens += len(msg.Content) / 4 // 粗略估计：4个字符 ≈ 1个token
	}
	summary.TokenCount = totalTokens

	// 存储摘要
	m.mu.Lock()
	m.summaries[sessionID] = summary
	m.mu.Unlock()

	return summary, nil
}

// UpdateSummary 更新会话摘要
func (m *SessionSummaryManager) UpdateSummary(