Function TestFallbackOrder

pkg/runtime/fallback_test.go:98–132

(t *testing.T)

}

func TestFallbackOrder(t *testing.T) {
	t.Parallel()

	synctest.Test(t, func(t *testing.T) {
		primary := &failingProvider{id: "primary/failing", err: errors.New("500 internal server error")}
		fallback1 := &failingProvider{id: "fallback1/failing", err: errors.New("503 service unavailable")}
		successStream := newStreamBuilder().
			AddContent("Success from fallback2").
			AddStopWithUsage(10, 5).
			Build()
		fallback2 := &mockProvider{id: "fallback2/success", stream: successStream}

		root := agent.New("root", "test",
			agent.WithModel(primary),
			agent.WithFallbackModel(fallback1),
			agent.WithFallbackModel(fallback2),
			agent.WithFallbackRetries(0),
		)

		tm := team.New(team.WithAgents(root))
		rt, err := NewLocalRuntime(t.Context(), tm, WithSessionCompaction(false), WithModelStore(mockModelStore{}))
		require.NoError(t, err)

		sess := session.New(session.WithUserMessage("test"))
		sess.Title = "Fallback Test"

		var gotContent bool
		for ev := range rt.RunStream(t.Context(), sess) {
			if choice, ok := ev.(*AgentChoiceEvent); ok && choice.Content == "Success from fallback2" {
				gotContent = true
			}
		}
		assert.True(t, gotContent, "should receive content from fallback2")
	})
}

func TestFallbackNoRetryOnNonRetryableError(t *testing.T) {
	t.Parallel()