diff --git a/integration/concurrency_test.go b/integration/concurrency_test.go index 8593285b..c59a1d9b 100644 --- a/integration/concurrency_test.go +++ b/integration/concurrency_test.go @@ -40,7 +40,7 @@ func TestMultiModelConcurrency(t *testing.T) { } resp = [2][]string{ []string{"sunlight"}, - []string{"england", "english", "massachusetts", "pilgrims", "british"}, + []string{"england", "english", "massachusetts", "pilgrims", "british", "festival"}, } ) var wg sync.WaitGroup diff --git a/integration/utils_test.go b/integration/utils_test.go index 7e1fcc10..cdf345ba 100644 --- a/integration/utils_test.go +++ b/integration/utils_test.go @@ -275,7 +275,7 @@ func DoGenerate(ctx context.Context, t *testing.T, client *api.Client, genReq ap break } } - require.True(t, atLeastOne, "none of %v found in %s", anyResp, response) + require.True(t, atLeastOne, "%s: none of %v found in %s", genReq.Model, anyResp, response) slog.Info("test pass", "model", genReq.Model, "prompt", genReq.Prompt, "contains", anyResp, "response", response) case <-ctx.Done(): t.Error("outer test context done while waiting for generate")