Function TestMultiModelTokenCounter

pkg/context/token_counter_test.go:251–313

(t *testing.T)

}

func TestMultiModelTokenCounter(t *testing.T) {
	multi := NewMultiModelTokenCounter()

	// 注册计数器
	multi.RegisterCounter("gpt-4", NewGPT4Counter())
	multi.RegisterCounter("claude", NewClaudeCounter())

	tests := []struct {
		name      string
		modelName string
		text      string
		wantModel string
	}{
		{
			name:      "exact match gpt-4",
			modelName: "gpt-4",
			text:      "Hello",
			wantModel: "gpt-4",
		},
		{
			name:      "exact match claude",
			modelName: "claude",
			text:      "Hello",
			wantModel: "claude-sonnet-4-5",
		},
		{
			name:      "fuzzy match gpt-4-turbo",
			modelName: "gpt-4-turbo",
			text:      "Hello",
			wantModel: "gpt-4",
		},
		{
			name:      "fuzzy match claude-opus",
			modelName: "claude-3-opus",
			text:      "Hello",
			wantModel: "claude-sonnet-4-5",
		},
		{
			name:      "unknown model uses default",
			modelName: "unknown-model",
			text:      "Hello",
			wantModel: "default",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			counter := multi.GetCounter(tt.modelName)
			if counter.ModelName() != tt.wantModel {
				t.Errorf("GetCounter(%v).ModelName() = %v, want %v", tt.modelName, counter.ModelName(), tt.wantModel)
			}

			// 测试 CountForModel
			count, err := multi.CountForModel(context.Background(), tt.modelName, tt.text)
			if err != nil {
				t.Errorf("CountForModel() error = %v", err)
			}
			if count <= 0 {