Function BenchmarkPhase4Stack

pkg/middleware/phase4_bench_test.go:176–220

BenchmarkPhase4Stack 基准测试: Phase 4 完整中间件栈

(b *testing.B)

// BenchmarkPhase4Stack 基准测试: Phase 4 完整中间件栈
func BenchmarkPhase4Stack(b *testing.B) {
	ctx := context.Background()

	// 创建后端
	backend := newMockBackend()
	backend.files["/agent.md"] = "You are a helpful assistant"

	// 创建中间件
	memoryMW, _ := NewAgentMemoryMiddleware(&AgentMemoryMiddlewareConfig{
		Backend:    backend,
		MemoryPath: "/memories/",
	})

	summarizationMW, _ := NewSummarizationMiddleware(&SummarizationMiddlewareConfig{
		Summarizer:             mockSummarizer("Summary", false),
		MaxTokensBeforeSummary: 100000, // 不触发总结
		MessagesToKeep:         3,
	})

	req := &ModelRequest{
		SystemPrompt: "Test prompt",
		Messages: []types.Message{
			{Role: types.MessageRoleUser, ContentBlocks: []types.ContentBlock{&types.TextBlock{Text: "Hello"}}},
		},
	}

	// 构建中间件链
	handler := func(ctx context.Context, req *ModelRequest) (*ModelResponse, error) {
		return &ModelResponse{
			Message: types.Message{
				Role:          types.MessageRoleAssistant,
				ContentBlocks: []types.ContentBlock{&types.TextBlock{Text: "Response"}},
			},
		}, nil
	}

	// Memory -> Summarization -> Handler
	summarizationHandler := func(ctx context.Context, req *ModelRequest) (*ModelResponse, error) {
		return summarizationMW.WrapModelCall(ctx, req, handler)
	}

	for b.Loop() {
		_, _ = memoryMW.WrapModelCall(ctx, req, summarizationHandler)
	}
}

// BenchmarkExtractMessageContent 基准测试: 消息内容提取
func BenchmarkExtractMessageContent(b *testing.B) {