Method compressSimple

pkg/agent/prompt_optimizer.go:152–201

compressSimple 基于规则的简单压缩

(prompt string, opts *CompressOptions)

// compressSimple 基于规则的简单压缩
func (c *EnhancedPromptCompressor) compressSimple(prompt string, opts *CompressOptions) string {
	// 分析段落
	sections := c.analyzeSections(prompt)

	// 评分和排序
	scoredSections := c.scoreSections(sections, opts.PreserveSections)

	// 根据目标长度选择段落
	targetLen := opts.TargetLength
	if targetLen == 0 {
		// 根据压缩级别计算目标长度
		switch opts.Level {
		case CompressionLevelLight:
			targetLen = int(float64(len(prompt)) * 0.65)
		case CompressionLevelModerate:
			targetLen = int(float64(len(prompt)) * 0.45)
		case CompressionLevelAggressive:
			targetLen = int(float64(len(prompt)) * 0.25)
		default:
			targetLen = int(float64(len(prompt)) * 0.50)
		}
	}

	// 选择段落
	var result []string
	currentLen := 0

	// 首先添加必须保留的段落
	for _, ss := range scoredSections {
		if ss.MustKeep {
			result = append(result, ss.Content)
			currentLen += len(ss.Content)
		}
	}

	// 然后按评分添加其他段落
	for _, ss := range scoredSections {
		if !ss.MustKeep && currentLen+len(ss.Content) <= targetLen {
			result = append(result, ss.Content)
			currentLen += len(ss.Content)
		}
	}

	// 应用基本优化
	compressed := strings.Join(result, "\n\n")
	compressed = c.removeDuplicateEmptyLines(compressed)
	compressed = c.compactFormat(compressed)

	return compressed
}

// compressLLM LLM 驱动的压缩
func (c *EnhancedPromptCompressor) compressLLM(ctx context.Context, prompt string, opts *CompressOptions) (string, error) {