Skip to content

Commit a75a9ae

Browse files
qiaodevcopybara-github
authored andcommitted
feat!: change GenerateImageConfig.NumberOfImages to value type. And add clearer error message and docstring to other APIs.
PiperOrigin-RevId: 734589810
1 parent 74d4647 commit a75a9ae

File tree

4 files changed

+9
-11
lines changed

4 files changed

+9
-11
lines changed

models.go

Lines changed: 3 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

samples/generate_images.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ func generateImage(ctx context.Context) {
5252
}
5353
// Pass in basic config
5454
var config *genai.GenerateImagesConfig = &genai.GenerateImagesConfig{
55-
NumberOfImages: genai.Ptr[int64](1),
55+
NumberOfImages: 1,
5656
OutputMIMEType: "image/jpeg",
5757
IncludeRAIReason: true,
5858
}

samples/imagen.go

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,6 @@ func imagen(ctx context.Context) {
6767
ctx, "imagen-3.0-generate-002",
6868
/*prompt=*/ "An umbrella in the foreground, and a rainy night sky in the background",
6969
&genai.GenerateImagesConfig{
70-
// TODO: Change this to value type.
71-
NumberOfImages: genai.Ptr[int64](1),
7270
IncludeRAIReason: true,
7371
OutputMIMEType: "image/jpeg",
7472
},
@@ -101,7 +99,7 @@ func imagen(ctx context.Context) {
10199
ReferenceID: 2,
102100
Config: &genai.MaskReferenceConfig{
103101
MaskMode: "MASK_MODE_BACKGROUND",
104-
MaskDilation: genai.Ptr(0.0),
102+
MaskDilation: genai.Ptr[float32](0.0),
105103
},
106104
}
107105
response3, err := client.Models.EditImage(

types.go

Lines changed: 4 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)