diff --git a/genai/client.go b/genai/client.go
index 7be7254..931e3d2 100644
--- a/genai/client.go
+++ b/genai/client.go
@@ -15,7 +15,7 @@
 // To get the protoveneer tool:
 //    go install cloud.google.com/go/internal/protoveneer/cmd/protoveneer@latest
 
-//go:generate protoveneer -license license.txt config.yaml ../../../googleapis/google-cloud-go/ai/generativelanguage/apiv1/generativelanguagepb
+//go:generate protoveneer -license license.txt config.yaml ../../../googleapis/google-cloud-go/ai/generativelanguage/apiv1beta/generativelanguagepb
 
 package genai
 
@@ -27,8 +27,8 @@ import (
 	"reflect"
 	"strings"
 
-	gl "cloud.google.com/go/ai/generativelanguage/apiv1"
-	pb "cloud.google.com/go/ai/generativelanguage/apiv1/generativelanguagepb"
+	gl "cloud.google.com/go/ai/generativelanguage/apiv1beta"
+	pb "cloud.google.com/go/ai/generativelanguage/apiv1beta/generativelanguagepb"
 
 	"github.com/google/generative-ai-go/internal"
 	"github.com/google/generative-ai-go/internal/support"
diff --git a/genai/client_test.go b/genai/client_test.go
index 70fafe2..f66a631 100644
--- a/genai/client_test.go
+++ b/genai/client_test.go
@@ -121,12 +121,12 @@ func TestLive(t *testing.T) {
 		}
 
 		checkMatch(t,
-			send("Name puppy breeds.", false),
-			"Beagle", "Poodle")
+			send("Name the 5 most popular puppy breeds.", false),
+			"Retriever", "Poodle")
 
 		checkMatch(t,
 			send("Which is best?", true),
-			"best", "depends", "([Cc]onsider|research|compare)")
+			"best", "depends", "([Cc]onsider|research|compare|preferences)")
 	})
 
 	t.Run("image", func(t *testing.T) {
diff --git a/genai/config.yaml b/genai/config.yaml
index 65f0578..63a78e7 100644
--- a/genai/config.yaml
+++ b/genai/config.yaml
@@ -16,7 +16,7 @@
 
 package: genai
 
-protoImportPath: cloud.google.com/go/ai/generativelanguage/apiv1/generativelanguagepb
+protoImportPath: cloud.google.com/go/ai/generativelanguage/apiv1beta/generativelanguagepb
 supportImportPath: github.com/google/generative-ai-go/internal/support
 
 types:
@@ -83,6 +83,8 @@ types:
       fields:
         Index:
           type: int32
+        GroundingAttributions:
+          omit: true
 
     GenerateContentResponse_PromptFeedback:
       name: PromptFeedback
@@ -114,7 +116,6 @@ types:
           type: int32
 
 
-
 # Omit everything not explicitly configured.
 omitTypes:
   - '*'
diff --git a/genai/content.go b/genai/content.go
index c93d5f0..b6a97fe 100644
--- a/genai/content.go
+++ b/genai/content.go
@@ -17,7 +17,7 @@ package genai
 import (
 	"fmt"
 
-	pb "cloud.google.com/go/ai/generativelanguage/apiv1/generativelanguagepb"
+	pb "cloud.google.com/go/ai/generativelanguage/apiv1beta/generativelanguagepb"
 )
 
 const (
diff --git a/genai/embed.go b/genai/embed.go
index ade72a0..8af7624 100644
--- a/genai/embed.go
+++ b/genai/embed.go
@@ -17,7 +17,7 @@ package genai
 import (
 	"context"
 
-	pb "cloud.google.com/go/ai/generativelanguage/apiv1/generativelanguagepb"
+	pb "cloud.google.com/go/ai/generativelanguage/apiv1beta/generativelanguagepb"
 )
 
 // EmbeddingModel creates a new instance of the named embedding model.
diff --git a/genai/generativelanguagepb_veneer.gen.go b/genai/generativelanguagepb_veneer.gen.go
index 7b9cb93..aaa1874 100644
--- a/genai/generativelanguagepb_veneer.gen.go
+++ b/genai/generativelanguagepb_veneer.gen.go
@@ -19,7 +19,7 @@ package genai
 import (
 	"fmt"
 
-	pb "cloud.google.com/go/ai/generativelanguage/apiv1/generativelanguagepb"
+	pb "cloud.google.com/go/ai/generativelanguage/apiv1beta/generativelanguagepb"
 	"github.com/google/generative-ai-go/internal/support"
 )
 
diff --git a/genai/list_models.go b/genai/list_models.go
index 97b36d0..fb23299 100644
--- a/genai/list_models.go
+++ b/genai/list_models.go
@@ -17,8 +17,8 @@ package genai
 import (
 	"context"
 
-	gl "cloud.google.com/go/ai/generativelanguage/apiv1"
-	pb "cloud.google.com/go/ai/generativelanguage/apiv1/generativelanguagepb"
+	gl "cloud.google.com/go/ai/generativelanguage/apiv1beta"
+	pb "cloud.google.com/go/ai/generativelanguage/apiv1beta/generativelanguagepb"
 
 	"google.golang.org/api/iterator"
 )