add the annotation more widely

Author: David Motsonashvili

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/FirebaseVertexAI.kt

@@ -91,8 +91,8 @@ internal constructor(
    * @param requestOptions Configuration options for sending requests to the backend.
    * @return The initialized [ImagenModel] instance.
    */
-  @OptIn(PublicPreviewAPI::class)
   @JvmOverloads
+  @PublicPreviewAPI
   public fun imagenModel(
     modelName: String,
     generationConfig: ImagenGenerationConfig? = null,

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/APIController.kt

@@ -82,6 +82,7 @@ internal val JSON = Json {
  * @property apiClient The value to pass in the `x-goog-api-client` header.
  * @property headerProvider A provider that generates extra headers to include in all HTTP requests.
  */
+@OptIn(PublicPreviewAPI::class)
 internal class APIController
 internal constructor(
   private val key: String,

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/Request.kt

@@ -74,6 +74,7 @@ internal data class GenerateImageRequest(
 ) : Request {
   @Serializable internal data class ImagenPrompt(val prompt: String)
 
+  @OptIn(PublicPreviewAPI::class)
   @Serializable
   internal data class ImagenParameters(
     val sampleCount: Int = 1,

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/java/ImagenModelFutures.kt

@@ -29,7 +29,7 @@ import com.google.firebase.vertexai.type.ImagenInlineImage
  *
  * @see [ImagenModel]
  */
-@OptIn(PublicPreviewAPI::class)
+@PublicPreviewAPI
 public abstract class ImagenModelFutures internal constructor() {
   /**
    * Generates an image, returning the result directly to the caller.

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenAspectRatio.kt

@@ -16,7 +16,10 @@
 
 package com.google.firebase.vertexai.type
 
+import com.google.firebase.vertexai.common.PublicPreviewAPI
+
 /** Represents the aspect ratio that the generated image should conform to. */
+@PublicPreviewAPI
 public class ImagenAspectRatio private constructor(internal val internalVal: String) {
   public companion object {
     /** A square image, useful for icons, profile pictures, etc. */

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGCSImage.kt

@@ -16,11 +16,14 @@
 
 package com.google.firebase.vertexai.type
 
+import com.google.firebase.vertexai.common.PublicPreviewAPI
+
 /**
  * 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"`).
  */
+@PublicPreviewAPI
 public class ImagenGCSImage
 internal constructor(public val gcsUri: String, public val mimeType: String) {}

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationConfig.kt

@@ -16,6 +16,8 @@
 
 package com.google.firebase.vertexai.type
 
+import com.google.firebase.vertexai.common.PublicPreviewAPI
+
 /**
  * Contains extra settings to configure image generation.
  *
@@ -26,6 +28,7 @@ package com.google.firebase.vertexai.type
  * @param imageFormat The file format/compression of the generated images.
  * @param addWatermark Adds an invisible watermark to mark the image as AI generated.
  */
+@PublicPreviewAPI
 public class ImagenGenerationConfig(
   public val negativePrompt: String? = null,
   public val numberOfImages: Int? = 1,
@@ -88,6 +91,7 @@ public class ImagenGenerationConfig(
  * }
  * ```
  */
+@PublicPreviewAPI
 public fun imagenGenerationConfig(
   init: ImagenGenerationConfig.Builder.() -> Unit
 ): ImagenGenerationConfig {

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenGenerationResponse.kt

@@ -16,6 +16,7 @@
 
 package com.google.firebase.vertexai.type
 
+import com.google.firebase.vertexai.common.PublicPreviewAPI
 import kotlinx.serialization.Serializable
 
 /**
@@ -25,6 +26,7 @@ import kotlinx.serialization.Serializable
  * @param filteredReason if fewer images were generated than were requested, this field will contain
  * the reason they were filtered out.
  */
+@PublicPreviewAPI
 public class ImagenGenerationResponse<T>
 internal constructor(public val images: List<T>, public val filteredReason: String?) {
 

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenImageFormat.kt

@@ -16,6 +16,7 @@
 
 package com.google.firebase.vertexai.type
 
+import com.google.firebase.vertexai.common.PublicPreviewAPI
 import kotlinx.serialization.Serializable
 
 /**
@@ -26,6 +27,7 @@ import kotlinx.serialization.Serializable
  * means the image is permitted to be lower quality to reduce size. This parameter is not relevant
  * for every MIME type.
  */
+@PublicPreviewAPI
 public class ImagenImageFormat
 private constructor(public val mimeType: String, public val compressionQuality: Int?) {
 

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ImagenInlineImage.kt

@@ -19,13 +19,15 @@ package com.google.firebase.vertexai.type
 import android.graphics.Bitmap
 import android.graphics.BitmapFactory
 import android.util.Base64
+import com.google.firebase.vertexai.common.PublicPreviewAPI
 
 /**
  * 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"`)
  */
+@PublicPreviewAPI
 public class ImagenInlineImage
 internal constructor(public val data: ByteArray, public val mimeType: String) {