Function TestFallbackNoRetryOnNonRetryableError

pkg/runtime/fallback_test.go:134–168

(t *testing.T)

}

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

	synctest.Test(t, func(t *testing.T) {
		primary := &failingProvider{id: "primary/auth-fail", err: errors.New("401 unauthorized")}
		successStream := newStreamBuilder().
			AddContent("Should not see this").
			AddStopWithUsage(10, 5).
			Build()
		fallback := &mockProvider{id: "fallback/success", stream: successStream}

		root := agent.New("root", "test",
			agent.WithModel(primary),
			agent.WithFallbackModel(fallback),
		)

		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 = "Non-Retryable Test"

		var gotError, gotFallbackContent bool
		for ev := range rt.RunStream(t.Context(), sess) {
			if _, ok := ev.(*ErrorEvent); ok {
				gotError = true
			}
			if choice, ok := ev.(*AgentChoiceEvent); ok && choice.Content == "Should not see this" {
				gotFallbackContent = true
			}
		}
		assert.True(t, gotFallbackContent || gotError, "should either get fallback content or error")
	})
}

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