Rename exception

Author: emilypgoogle

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

@@ -131,7 +131,7 @@ internal constructor(
         .body<GenerateContentResponse>()
         .validate()
     } catch (e: Throwable) {
-      throw GoogleGenerativeAIException.from(e)
+      throw FirebaseCommonAIException.from(e)
     }
 
   fun generateContentStream(request: GenerateContentRequest): Flow<GenerateContentResponse> =
@@ -142,7 +142,7 @@ internal constructor(
         applyCommonConfiguration(request)
       }
       .map { it.validate() }
-      .catch { throw GoogleGenerativeAIException.from(it) }
+      .catch { throw FirebaseCommonAIException.from(it) }
 
   suspend fun countTokens(request: CountTokensRequest): CountTokensResponse =
     try {
@@ -154,7 +154,7 @@ internal constructor(
         .also { validateResponse(it) }
         .body()
     } catch (e: Throwable) {
-      throw GoogleGenerativeAIException.from(e)
+      throw FirebaseCommonAIException.from(e)
     }
 
   private fun HttpRequestBuilder.applyCommonConfiguration(request: Request) {

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

@@ -20,19 +20,19 @@ import io.ktor.serialization.JsonConvertException
 import kotlinx.coroutines.TimeoutCancellationException
 
 /** Parent class for any errors that occur. */
-internal sealed class GoogleGenerativeAIException(message: String, cause: Throwable? = null) :
+internal sealed class FirebaseCommonAIException(message: String, cause: Throwable? = null) :
   RuntimeException(message, cause) {
   companion object {
 
     /**
-     * Converts a [Throwable] to a [GoogleGenerativeAIException].
+     * Converts a [Throwable] to a [FirebaseCommonAIException].
      *
      * Will populate default messages as expected, and propagate the provided [cause] through the
      * resulting exception.
      */
-    fun from(cause: Throwable): GoogleGenerativeAIException =
+    fun from(cause: Throwable): FirebaseCommonAIException =
       when (cause) {
-        is GoogleGenerativeAIException -> cause
+        is FirebaseCommonAIException -> cause
         is JsonConvertException,
         is kotlinx.serialization.SerializationException ->
           SerializationException(
@@ -48,15 +48,15 @@ internal sealed class GoogleGenerativeAIException(message: String, cause: Throwa
 
 /** Something went wrong while trying to deserialize a response from the server. */
 internal class SerializationException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)
 
 /** The server responded with a non 200 response code. */
 internal class ServerException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)
 
 /** The server responded that the API Key is no valid. */
 internal class InvalidAPIKeyException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)
 
 /**
  * A request was blocked for some reason.
@@ -69,7 +69,7 @@ internal class PromptBlockedException(
   val response: GenerateContentResponse,
   cause: Throwable? = null
 ) :
-  GoogleGenerativeAIException(
+  FirebaseCommonAIException(
     "Prompt was blocked: ${response.promptFeedback?.blockReason?.name}",
     cause,
   )
@@ -82,15 +82,15 @@ internal class PromptBlockedException(
  * territories) where the API is available.
  */
 internal class UnsupportedUserLocationException(cause: Throwable? = null) :
-  GoogleGenerativeAIException("User location is not supported for the API use.", cause)
+  FirebaseCommonAIException("User location is not supported for the API use.", cause)
 
 /**
  * Some form of state occurred that shouldn't have.
  *
  * Usually indicative of consumer error.
  */
 internal class InvalidStateException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)
 
 /**
  * A request was stopped during generation for some reason.
@@ -101,7 +101,7 @@ internal class ResponseStoppedException(
   val response: GenerateContentResponse,
   cause: Throwable? = null
 ) :
-  GoogleGenerativeAIException(
+  FirebaseCommonAIException(
     "Content generation stopped. Reason: ${response.candidates?.first()?.finishReason?.name}",
     cause,
   )
@@ -112,16 +112,16 @@ internal class ResponseStoppedException(
  * Usually occurs due to a user specified [timeout][RequestOptions.timeout].
  */
 internal class RequestTimeoutException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)
 
 /** The quota for this API key is depleted, retry this request at a later time. */
 internal class QuotaExceededException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)
 
 /** The service is not enabled for this project. Visit the Firebase Console to enable it. */
 internal class ServiceDisabledException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)
 
 /** Catch all case for exceptions not explicitly expected. */
 internal class UnknownException(message: String, cause: Throwable? = null) :
-  GoogleGenerativeAIException(message, cause)
+  FirebaseCommonAIException(message, cause)

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

@@ -17,7 +17,7 @@
 package com.google.firebase.vertexai.type
 
 import com.google.firebase.vertexai.FirebaseVertexAI
-import com.google.firebase.vertexai.common.GoogleGenerativeAIException
+import com.google.firebase.vertexai.common.FirebaseCommonAIException
 import com.google.firebase.vertexai.internal.util.toPublic
 import kotlinx.coroutines.TimeoutCancellationException
 
@@ -36,7 +36,7 @@ sealed class FirebaseVertexAIException(message: String, cause: Throwable? = null
     fun from(cause: Throwable): FirebaseVertexAIException =
       when (cause) {
         is FirebaseVertexAIException -> cause
-        is GoogleGenerativeAIException ->
+        is FirebaseCommonAIException ->
           when (cause) {
             is com.google.firebase.vertexai.common.SerializationException ->
               SerializationException(cause.message ?: "", cause.cause)