Function TestModelFallbackManager_Stream

pkg/agent/model_fallback_test.go:396–470

(t *testing.T)

}

func TestModelFallbackManager_Stream(t *testing.T) {
	factory := NewMockProviderFactory()

	// 第一个模型失败
	factory.SetProvider("openai/gpt-4", &MockProvider{
		name:       "openai/gpt-4",
		shouldFail: true,
		failCount:  999,
	})

	// 第二个模型成功
	factory.SetProvider("anthropic/claude-3", &MockProvider{
		name: "anthropic/claude-3",
	})

	deps := &Dependencies{
		ProviderFactory: factory,
	}

	fallbacks := []*ModelFallback{
		{
			Config: &types.ModelConfig{
				Provider: "openai",
				Model:    "gpt-4",
			},
			MaxRetries: 0,
			Enabled:    true,
			Priority:   1,
		},
		{
			Config: &types.ModelConfig{
				Provider: "anthropic",
				Model:    "claude-3",
			},
			MaxRetries: 0,
			Enabled:    true,
			Priority:   2,
		},
	}

	manager, err := NewModelFallbackManager(fallbacks, deps)
	if err != nil {
		t.Fatalf("Failed to create manager: %v", err)
	}

	ctx := context.Background()
	messages := []types.Message{
		{Role: "user", Content: "Hello"},
	}

	stream, err := manager.Stream(ctx, messages, nil)
	if err != nil {
		t.Fatalf("Stream failed: %v", err)
	}

	// 读取流
	var content string
	var contentSb453 strings.Builder