ktfmt

Author: David Motsonashvili

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

@@ -20,9 +20,9 @@ import android.graphics.Bitmap
 import com.google.firebase.appcheck.interop.InteropAppCheckTokenProvider
 import com.google.firebase.auth.internal.InternalAuthProvider
 import com.google.firebase.vertexai.common.APIController
+import com.google.firebase.vertexai.common.AppCheckHeaderProvider
 import com.google.firebase.vertexai.common.CountTokensRequest
 import com.google.firebase.vertexai.common.GenerateContentRequest
-import com.google.firebase.vertexai.common.AppCheckHeaderProvider
 import com.google.firebase.vertexai.type.Content
 import com.google.firebase.vertexai.type.CountTokensResponse
 import com.google.firebase.vertexai.type.FinishReason

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

@@ -19,8 +19,8 @@ package com.google.firebase.vertexai
 import com.google.firebase.appcheck.interop.InteropAppCheckTokenProvider
 import com.google.firebase.auth.internal.InternalAuthProvider
 import com.google.firebase.vertexai.common.APIController
-import com.google.firebase.vertexai.common.ContentBlockedException
 import com.google.firebase.vertexai.common.AppCheckHeaderProvider
+import com.google.firebase.vertexai.common.ContentBlockedException
 import com.google.firebase.vertexai.common.GenerateImageRequest
 import com.google.firebase.vertexai.common.ImagenParameters
 import com.google.firebase.vertexai.common.ImagenPromptInstance

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

@@ -58,7 +58,6 @@ import kotlinx.coroutines.flow.catch
 import kotlinx.coroutines.flow.channelFlow
 import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.launch
-import kotlinx.coroutines.tasks.await
 import kotlinx.coroutines.withTimeout
 import kotlinx.serialization.ExperimentalSerializationApi
 import kotlinx.serialization.json.Json
@@ -318,4 +317,4 @@ private fun GenerateContentResponse.Internal.validate() = apply {
     ?.mapNotNull { it.finishReason }
     ?.firstOrNull { it != FinishReason.Internal.STOP }
     ?.let { throw ResponseStoppedException(this) }
-}
+}

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

@@ -19,46 +19,46 @@ package com.google.firebase.vertexai.common
 import android.util.Log
 import com.google.firebase.appcheck.interop.InteropAppCheckTokenProvider
 import com.google.firebase.auth.internal.InternalAuthProvider
-import kotlinx.coroutines.tasks.await
 import kotlin.time.Duration
 import kotlin.time.Duration.Companion.seconds
+import kotlinx.coroutines.tasks.await
 
 internal class AppCheckHeaderProvider(
-    private val logTag: String,
-    private val appCheckTokenProvider: InteropAppCheckTokenProvider? = null,
-    private val internalAuthProvider: InternalAuthProvider? = null,
+  private val logTag: String,
+  private val appCheckTokenProvider: InteropAppCheckTokenProvider? = null,
+  private val internalAuthProvider: InternalAuthProvider? = null,
 ) : HeaderProvider {
-    override val timeout: Duration
-        get() = 10.seconds
-
-    override suspend fun generateHeaders(): Map<String, String> {
-        val headers = mutableMapOf<String, String>()
-        if (appCheckTokenProvider == null) {
-            Log.w(logTag, "AppCheck not registered, skipping")
-        } else {
-            val token = appCheckTokenProvider.getToken(false).await()
+  override val timeout: Duration
+    get() = 10.seconds
 
-            if (token.error != null) {
-                Log.w(logTag, "Error obtaining AppCheck token", token.error)
-            }
-            // The Firebase App Check backend can differentiate between apps without App Check, and
-            // wrongly configured apps by verifying the value of the token, so it always needs to be
-            // included.
-            headers["X-Firebase-AppCheck"] = token.token
-        }
+  override suspend fun generateHeaders(): Map<String, String> {
+    val headers = mutableMapOf<String, String>()
+    if (appCheckTokenProvider == null) {
+      Log.w(logTag, "AppCheck not registered, skipping")
+    } else {
+      val token = appCheckTokenProvider.getToken(false).await()
 
-        if (internalAuthProvider == null) {
-            Log.w(logTag, "Auth not registered, skipping")
-        } else {
-            try {
-                val token = internalAuthProvider.getAccessToken(false).await()
+      if (token.error != null) {
+        Log.w(logTag, "Error obtaining AppCheck token", token.error)
+      }
+      // The Firebase App Check backend can differentiate between apps without App Check, and
+      // wrongly configured apps by verifying the value of the token, so it always needs to be
+      // included.
+      headers["X-Firebase-AppCheck"] = token.token
+    }
 
-                headers["Authorization"] = "Firebase ${token.token!!}"
-            } catch (e: Exception) {
-                Log.w(logTag, "Error getting Auth token ", e)
-            }
-        }
+    if (internalAuthProvider == null) {
+      Log.w(logTag, "Auth not registered, skipping")
+    } else {
+      try {
+        val token = internalAuthProvider.getAccessToken(false).await()
 
-        return headers
+        headers["Authorization"] = "Firebase ${token.token!!}"
+      } catch (e: Exception) {
+        Log.w(logTag, "Error getting Auth token ", e)
+      }
     }
-}
+
+    return headers
+  }
+}

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

@@ -23,7 +23,7 @@ import kotlinx.serialization.Serializable
  *
  * @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.
+ * the reason they were filtered out.
  */
 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

@@ -23,8 +23,8 @@ import kotlinx.serialization.Serializable
  *
  * @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.
+ * 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?) {
@@ -38,7 +38,7 @@ private constructor(public val mimeType: String, public val compressionQuality:
      * 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.
+     * 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)