diff --git a/integration/concurrency_test.go b/integration/concurrency_test.go index 42e9d074..74eb6361 100644 --- a/integration/concurrency_test.go +++ b/integration/concurrency_test.go @@ -42,7 +42,7 @@ func TestMultiModelConcurrency(t *testing.T) { } resp = [2][]string{ {"sunlight"}, - {"england", "english", "massachusetts", "pilgrims", "british"}, + {"england", "english", "massachusetts", "pilgrims", "british", "festival"}, } ) var wg sync.WaitGroup diff --git a/integration/utils_test.go b/integration/utils_test.go index a6010995..e76b63f2 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")