Method UpdateSummary

pkg/memory/session_summary.go:182–237

UpdateSummary 更新会话摘要

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

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

	// 获取现有摘要
	m.mu.RLock()
	existingSummary, exists := m.summaries[sessionID]
	m.mu.RUnlock()

	// 如果不存在，生成新摘要
	if !exists {
		return m.GenerateSummary(ctx, sessionID, newMessages)
	}

	// 构建增量更新提示词
	prompt := m.buildIncrementalPrompt(existingSummary, newMessages)

	// 调用 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 update summary: %w", err)
	}

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

	// 更新基本信息
	updatedSummary.SessionID = sessionID
	updatedSummary.CreatedAt = existingSummary.CreatedAt
	updatedSummary.UpdatedAt = time.Now()
	updatedSummary.MessageCount = existingSummary.MessageCount + len(newMessages)
	updatedSummary.TokenCount = existingSummary.TokenCount + m.estimateTokens(newMessages)

	// 存储更新后的摘要
	m.mu.Lock()
	m.summaries[sessionID] = updatedSummary
	m.mu.Unlock()

	return updatedSummary, nil
}

// GetSummary 获取会话摘要