diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/FirebaseVertexAI.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/FirebaseVertexAI.kt
index ec79fd3614a..9e78e8811ba 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/FirebaseVertexAI.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/FirebaseVertexAI.kt
@@ -81,6 +81,15 @@ internal constructor(
     )
   }
 
+  /**
+   * Instantiates a new [ImagenModel] given the provided parameters.
+   *
+   * @param modelName The name of the model to use, for example `"imagen-3.0-generate-001"`.
+   * @param generationConfig The configuration parameters to use for image generation.
+   * @param safetySettings The safety bounds the model will abide by during image generation.
+   * @param requestOptions Configuration options for sending requests to the backend.
+   * @return The initialized [ImagenModel] instance.
+   */
   @JvmOverloads
   public fun imagenModel(
     modelName: String,
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/ImagenModel.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/ImagenModel.kt
index 7930a86eb60..7b68a2d8ef6 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/ImagenModel.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/ImagenModel.kt
@@ -1,3 +1,19 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai
 
 import com.google.firebase.appcheck.interop.InteropAppCheckTokenProvider
@@ -20,6 +36,10 @@ import com.google.firebase.vertexai.type.ImagenInlineImage
 import com.google.firebase.vertexai.type.ImagenSafetySettings
 import com.google.firebase.vertexai.type.RequestOptions
 
+/**
+ * Represents a generative model (like Imagen), capable of generating images based on various input
+ * types.
+ */
 public class ImagenModel
 internal constructor(
   private val modelName: String,
@@ -27,6 +47,7 @@ internal constructor(
   private val safetySettings: ImagenSafetySettings? = null,
   private val controller: APIController,
 ) {
+  @JvmOverloads
   internal constructor(
     modelName: String,
     apiKey: String,
@@ -48,8 +69,14 @@ internal constructor(
     ),
   )
 
+  /**
+   * Generates an image, returning the result directly to the caller.
+   *
+   * @param prompt The input(s) given to the model as a prompt.
+   */
   public suspend fun generateImages(prompt: String): ImagenGenerationResponse<ImagenInlineImage> =
     try {
+      controller.generateImage(constructRequest(prompt, null, generationConfig)).toPublicInline()
       controller
         .generateImage(constructRequest(prompt, null, generationConfig))
         .validate()
@@ -58,6 +85,13 @@ internal constructor(
       throw FirebaseVertexAIException.from(e)
     }
 
+  /**
+   * Generates an image, storing the result in Google Cloud Storage and returning a URL
+   *
+   * @param prompt The input(s) given to the model as a prompt.
+   * @param gcsUri Specifies where in Google Cloud Storage to store the image (for example, a
+   * specific bucket or folder).
+   */
   public suspend fun generateImages(
     prompt: String,
     gcsUri: String,
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageRequest.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageRequest.kt
index 95535be5b62..d826b11e87f 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageRequest.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageRequest.kt
@@ -1,3 +1,19 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.internal
 
 import com.google.firebase.vertexai.common.Request
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageResponse.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageResponse.kt
index a8b97785456..fa6dff21fcd 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageResponse.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/GenerateImageResponse.kt
@@ -1,3 +1,19 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.internal
 
 import kotlinx.serialization.Serializable
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/AppCheckHeaderProvider.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/AppCheckHeaderProvider.kt
index 67140156070..2df31dcf660 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/AppCheckHeaderProvider.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/AppCheckHeaderProvider.kt
@@ -1,3 +1,19 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.internal.util
 
 import android.util.Log
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenAspectRatio.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenAspectRatio.kt
index db3602c6ede..8bcdaf70307 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenAspectRatio.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenAspectRatio.kt
@@ -1,11 +1,33 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/** Represents the aspect ratio that the generated image should conform to. */
 public class ImagenAspectRatio private constructor(internal val internalVal: String) {
   public companion object {
+    /** A square image, useful for icons, profile pictures, etc. */
     @JvmField public val SQUARE_1x1: ImagenAspectRatio = ImagenAspectRatio("1:1")
+    /** A portrait image in 3:4, the aspect ratio of older TVs. */
     @JvmField public val PORTRAIT_3x4: ImagenAspectRatio = ImagenAspectRatio("3:4")
+    /** A landscape image in 4:3, the aspect ratio of older TVs. */
     @JvmField public val LANDSCAPE_4x3: ImagenAspectRatio = ImagenAspectRatio("4:3")
+    /** A portrait image in 9:16, the aspect ratio of modern monitors and phone screens. */
     @JvmField public val PORTRAIT_9x16: ImagenAspectRatio = ImagenAspectRatio("9:16")
+    /** A landscape image in 16:9, the aspect ratio of modern monitors and phone screens. */
     @JvmField public val LANDSCAPE_16x9: ImagenAspectRatio = ImagenAspectRatio("16:9")
   }
 }
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGCSImage.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGCSImage.kt
index b6af1542f1a..765e27b3583 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGCSImage.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGCSImage.kt
@@ -1,4 +1,26 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/**
+ * Represents an Imagen-generated image that is contained in Google Cloud Storage.
+ *
+ * @param gcsUri Contains the `gs://` URI for the image.
+ * @param mimeType Contains the MIME type of the image (for example, `"image/png"`).
+ */
 public class ImagenGCSImage
 internal constructor(public val gcsUri: String, public val mimeType: String) {}
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationConfig.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationConfig.kt
index 6f794e42434..a90bb2c4288 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationConfig.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationConfig.kt
@@ -1,5 +1,31 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/**
+ * Contains extra settings to configure image generation.
+ *
+ * @param negativePrompt This string contains things that should be explicitly excluded from
+ * generated images.
+ * @param numberOfImages How many images should be generated.
+ * @param aspectRatio The aspect ratio of the generated images.
+ * @param imageFormat The file format/compression of the generated images.
+ * @param addWatermark Adds an invisible watermark to mark the image as AI generated.
+ */
 public class ImagenGenerationConfig(
   public val negativePrompt: String? = null,
   public val numberOfImages: Int? = 1,
@@ -7,6 +33,19 @@ public class ImagenGenerationConfig(
   public val imageFormat: ImagenImageFormat? = null,
   public val addWatermark: Boolean? = null,
 ) {
+  /**
+   * Builder for creating a [ImagenGenerationConfig].
+   *
+   * This is mainly intended for Java interop. For Kotlin, use [imagenGenerationConfig] for a
+   * more idiomatic experience.
+   *
+   * @property negativePrompt See [ImagenGenerationConfig.negativePrompt].
+   * @property numberOfImages See [ImagenGenerationConfig.numberOfImages].
+   * @property aspectRatio See [ImagenGenerationConfig.aspectRatio].
+   * @property imageFormat See [ImagenGenerationConfig.imageFormat]
+   * @property addWatermark See [ImagenGenerationConfig.addWatermark]
+   * @see [imagenGenerationConfig]
+   */
   public class Builder {
     @JvmField public var negativePrompt: String? = null
     @JvmField public var numberOfImages: Int? = 1
@@ -14,6 +53,12 @@ public class ImagenGenerationConfig(
     @JvmField public var imageFormat: ImagenImageFormat? = null
     @JvmField public var addWatermark: Boolean? = null
 
+    /**
+     * Alternative casing for [ImagenGenerationConfig.Builder]:
+     * ```
+     * val config = GenerationConfig.builder()
+     * ```
+     */
     public fun build(): ImagenGenerationConfig =
       ImagenGenerationConfig(
         negativePrompt = negativePrompt,
@@ -29,6 +74,20 @@ public class ImagenGenerationConfig(
   }
 }
 
+/**
+ * Helper method to construct a [ImagenGenerationConfig] in a DSL-like manner.
+ *
+ * Example Usage:
+ * ```
+ * imagenGenerationConfig {
+ *   negativePrompt = "People, black and white, painting"
+ *   numberOfImages = 1
+ *   aspectRatio = ImagenAspecRatio.SQUARE_1x1
+ *   imageFormat = ImagenImageFormat.png()
+ *   addWatermark = false
+ * }
+ * ```
+ */
 public fun imagenGenerationConfig(
   init: ImagenGenerationConfig.Builder.() -> Unit
 ): ImagenGenerationConfig {
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationResponse.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationResponse.kt
index e4fea054187..1873ce83258 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationResponse.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationResponse.kt
@@ -1,4 +1,27 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/**
+ * Represents a response from a call to [ImagenModel#generateImages]
+ *
+ * @param images contains the generated images
+ * @param filteredReason if fewer images were generated than were requested, this field will contain
+ * the reason they were filtered out.
+ */
 public class ImagenGenerationResponse<T>
 internal constructor(public val images: List<T>, public val filteredReason: String?) {}
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenImageFormat.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenImageFormat.kt
index 32638724224..feaec329374 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenImageFormat.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenImageFormat.kt
@@ -1,13 +1,43 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/**
+ * Represents the format an image should be returned in.
+ * @param mimeType A string (like `"image/jpeg"`) specifying the encoding MIME type of the image.
+ * @param compressionQuality an int (1-100) representing the quality of the image; a lower number
+ * means the image is permitted to be lower quality to reduce size. This parameter is not relevant
+ * for every MIME type.
+ */
 public class ImagenImageFormat
 private constructor(public val mimeType: String, public val compressionQuality: Int?) {
 
   public companion object {
+    /**
+     * An [ImagenImageFormat] representing a JPEG image.
+     *
+     * @param compressionQuality an int (1-100) representing the quality of the image; a lower
+     * number means the image is permitted to be lower quality to reduce size.
+     */
     public fun jpeg(compressionQuality: Int? = null): ImagenImageFormat {
       return ImagenImageFormat("image/jpeg", compressionQuality)
     }
 
+    /** An [ImagenImageFormat] representing a PNG image */
     public fun png(): ImagenImageFormat {
       return ImagenImageFormat("image/png", null)
     }
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenInlineImage.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenInlineImage.kt
index 35332fd8ab6..5d83cea7d3b 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenInlineImage.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenInlineImage.kt
@@ -1,12 +1,37 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
 import android.graphics.Bitmap
 import android.graphics.BitmapFactory
 import android.util.Base64
 
+/**
+ * Represents an Imagen-generated image that is contained inline
+ *
+ * @param data Contains the raw bytes of the image
+ * @param mimeType Contains the MIME type of the image (for example, `"image/png"`)
+ */
 public class ImagenInlineImage
 internal constructor(public val data: ByteArray, public val mimeType: String) {
 
+  /**
+   * Returns the image as an Android OS native [Bitmap] so that it can be saved or sent to the UI.
+   */
   public fun asBitmap(): Bitmap {
     val data = Base64.decode(data, Base64.NO_WRAP)
     return BitmapFactory.decodeByteArray(data, 0, data.size)
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenPersonFilterLevel.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenPersonFilterLevel.kt
index ef82b88a1dc..22d4bc13cc3 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenPersonFilterLevel.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenPersonFilterLevel.kt
@@ -1,10 +1,30 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/** A filter used to prevent images from containing depictions of children or people. */
 public class ImagenPersonFilterLevel private constructor(internal val internalVal: String) {
   public companion object {
+    /** No filters applied. */
     @JvmField public val ALLOW_ALL: ImagenPersonFilterLevel = ImagenPersonFilterLevel("allow_all")
+    /** Filters out any images containing depictions of children. */
     @JvmField
     public val ALLOW_ADULT: ImagenPersonFilterLevel = ImagenPersonFilterLevel("allow_adult")
+    /** Filters out any images containing depictions of people. */
     @JvmField public val BLOCK_ALL: ImagenPersonFilterLevel = ImagenPersonFilterLevel("dont_allow")
   }
 }
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetyFilterLevel.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetyFilterLevel.kt
index 013d83bc9fc..788fd0760b1 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetyFilterLevel.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetyFilterLevel.kt
@@ -1,15 +1,39 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/** Used for safety filtering. */
 public class ImagenSafetyFilterLevel private constructor(internal val internalVal: String) {
   public companion object {
+    /** Strongest filtering level, most strict blocking. */
     @JvmField
     public val BLOCK_LOW_AND_ABOVE: ImagenSafetyFilterLevel =
       ImagenSafetyFilterLevel("block_low_and_above")
+    /** Block some problematic prompts and responses. */
     @JvmField
     public val BLOCK_MEDIUM_AND_ABOVE: ImagenSafetyFilterLevel =
       ImagenSafetyFilterLevel("block_medium_and_above")
+    /**
+     * Reduces the number of requests blocked due to safety filters. May increase objectionable
+     * content generated by the Imagen model.
+     */
     @JvmField
     public val BLOCK_ONLY_HIGH: ImagenSafetyFilterLevel = ImagenSafetyFilterLevel("block_only_high")
+    /** Turns off all optional safety filters. */
     @JvmField public val BLOCK_NONE: ImagenSafetyFilterLevel = ImagenSafetyFilterLevel("block_none")
   }
 }
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetySettings.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetySettings.kt
index df6ea11beab..47fc8144870 100644
--- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetySettings.kt
+++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenSafetySettings.kt
@@ -1,5 +1,27 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package com.google.firebase.vertexai.type
 
+/**
+ * A configuration for filtering unsafe content or images containing people.
+ *
+ * @param safetyFilterLevel Used to filter unsafe content.
+ * @param personFilterLevel Used to filter images containing people.
+ */
 public class ImagenSafetySettings(
   internal val safetyFilterLevel: ImagenSafetyFilterLevel,
   internal val personFilterLevel: ImagenPersonFilterLevel,