Method Compress

pkg/agent/prompt_optimizer.go:86–149

Compress 压缩 Prompt

(ctx context.Context, prompt string, opts *CompressOptions)

// Compress 压缩 Prompt
func (c *EnhancedPromptCompressor) Compress(ctx context.Context, prompt string, opts *CompressOptions) (*CompressResult, error) {
	if opts == nil {
		opts = &CompressOptions{
			Mode:  CompressionModeHybrid,
			Level: CompressionLevelModerate,
		}
	}

	// 计算原始 Token 数
	var originalTokens int
	var err error
	if c.tokenCounter != nil {
		originalTokens, err = c.tokenCounter.Count(ctx, prompt)
		if err != nil {
			originalTokens = len(prompt) / 4 // 降级估算
		}
	} else {
		originalTokens = len(prompt) / 4 // 无计数器时降级估算
	}

	var compressed string

	switch opts.Mode {
	case CompressionModeSimple:
		compressed = c.compressSimple(prompt, opts)
	case CompressionModeLLM:
		compressed, err = c.compressLLM(ctx, prompt, opts)
		if err != nil {
			// LLM 压缩失败，降级到简单压缩
			compressed = c.compressSimple(prompt, opts)
		}
	case CompressionModeHybrid:
		compressed, err = c.compressHybrid(ctx, prompt, opts)
		if err != nil {
			compressed = c.compressSimple(prompt, opts)
		}
	default:
		compressed = c.compressSimple(prompt, opts)
	}

	// 计算压缩后 Token 数
	var compressedTokens int
	if c.tokenCounter != nil {
		compressedTokens, _ = c.tokenCounter.Count(ctx, compressed)
	} else {
		compressedTokens = len(compressed) / 4
	}

	result := &CompressResult{
		Compressed:       compressed,
		OriginalLength:   len(prompt),
		CompressedLength: len(compressed),
		OriginalTokens:   originalTokens,
		CompressedTokens: compressedTokens,
		TokensSaved:      originalTokens - compressedTokens,
		Mode:             string(opts.Mode),
	}