Integration tests conditionally pull

If images aren't present, pull them.
Also fixes the expected responses
This commit is contained in:
Daniel Hiltgen
2024-03-24 16:22:38 -07:00
parent acfa2b9422
commit 7b6cbc10ec
4 changed files with 70 additions and 10 deletions

View File

@@ -30,7 +30,7 @@ func TestIntegrationMultimodal(t *testing.T) {
}
resp := "the ollamas"
ctx, cancel := context.WithTimeout(context.Background(), time.Second*60)
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Minute)
defer cancel()
GenerateTestHelper(ctx, t, &http.Client{}, req, []string{resp})
}