genai: use different model for tests requiring vision (#135)

google · Jun 10, 2024 · 07538e5 · 07538e5
1 parent 9f3bd3b
commit 07538e5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/genai/client_test.go b/genai/client_test.go
@@ -35,8 +35,9 @@ import (
 )
 
 const (
- defaultModel = "gemini-1.0-pro"
- imageFile = "personWorkingOnComputer.jpg"
+ defaultModel = "gemini-1.0-pro"
+ modelForVision = "gemini-1.5-flash"
+ imageFile = "personWorkingOnComputer.jpg"
 )
 
 func TestLive(t *testing.T) {
@@ -131,7 +132,7 @@ func TestLive(t *testing.T) {
  })
 
  t.Run("image", func(t *testing.T) {
- vmodel := client.GenerativeModel(defaultModel + "-vision-latest")
+ vmodel := client.GenerativeModel(modelForVision)
  vmodel.Temperature = Ptr[float32](0)
 
  data, err := os.ReadFile(filepath.Join("testdata", imageFile))