add new error for disabled service, with test (google-gemini#148)

Co-authored-by: David Motsonashvili <[email protected]>

common/src/commonMain/kotlin/dev/shreyaspatil/ai/client/generativeai/common/Exceptions.kt

@@ -112,6 +112,10 @@ class RequestTimeoutException(message: String, cause: Throwable? = null) :
 class QuotaExceededException(message: String, cause: Throwable? = null) :
     GoogleGenerativeAIException(message, cause)
 
+/** The service is not enabled for this project. Visit the Firebase Console to enable it. */
+class ServiceDisabledException(message: String, cause: Throwable? = null) :
+  GoogleGenerativeAIException(message, cause)
+
 /** Catch all case for exceptions not explicitly expected. */
 class UnknownException(message: String, cause: Throwable? = null) :
     GoogleGenerativeAIException(message, cause)

common/src/main/kotlin/com/google/ai/client/generativeai/common/APIController.kt

@@ -223,12 +223,13 @@ private fun fullModelName(name: String): String = name.takeIf { it.contains("/")
 private suspend fun validateResponse(response: HttpResponse) {
   if (response.status == HttpStatusCode.OK) return
   val text = response.bodyAsText()
-  val message =
+  val error =
     try {
-      JSON.decodeFromString<GRpcErrorResponse>(text).error.message
+      JSON.decodeFromString<GRpcErrorResponse>(text).error
     } catch (e: Throwable) {
-      "Unexpected Response:\n$text"
+      throw ServerException("Unexpected Response:\n$text $e")
     }
+  val message = error.message
   if (message.contains("API key not valid")) {
     throw InvalidAPIKeyException(message)
   }
@@ -239,6 +240,9 @@ private suspend fun validateResponse(response: HttpResponse) {
   if (message.contains("quota")) {
     throw QuotaExceededException(message)
   }
+  if (error.details.any { "SERVICE_DISABLED" == it.reason }) {
+    throw ServiceDisabledException(message)
+  }
   throw ServerException(message)
 }
 

common/src/main/kotlin/com/google/ai/client/generativeai/common/server/Types.kt

@@ -141,4 +141,7 @@ enum class FinishReason {
 data class GRpcError(
   val code: Int,
   val message: String,
+  val details: List<GRpcErrorDetails>,
 )
+
+@Serializable data class GRpcErrorDetails(val reason: String? = null)

common/src/test/java/com/google/ai/client/generativeai/common/UnarySnapshotTests.kt

@@ -291,4 +291,14 @@ internal class UnarySnapshotTests {
         }
       }
     }
+
+  @Test
+  fun `service disabled`() =
+    goldenUnaryFile("failure-service-disabled.json", HttpStatusCode.Forbidden) {
+      withTimeout(testTimeout) {
+        shouldThrow<ServiceDisabledException> {
+          apiController.generateContent(textGenerateContentRequest("prompt"))
+        }
+      }
+    }
 }

common/src/test/resources/golden-files/unary/failure-service-disabled.json

@@ -0,0 +1,27 @@
+{
+  "error": {
+    "code": 403,
+    "message": "Firebase ML API has not been used in project 12345 before or it is disabled. Enable it by visiting https://console.developers.google.com/apis/api/firebaseml.googleapis.com/overview?project=12345 then retry. If you enabled this API recently, wait a few minutes for the action to propagate to our systems and retry.",
+    "status": "PERMISSION_DENIED",
+    "details": [
+      {
+        "@type": "type.googleapis.com/google.rpc.Help",
+        "links": [
+          {
+            "description": "Google developers console API activation",
+            "url": "https://console.developers.google.com/apis/api/firebaseml.googleapis.com/overview?project=12345"
+          }
+        ]
+      },
+      {
+        "@type": "type.googleapis.com/google.rpc.ErrorInfo",
+        "reason": "SERVICE_DISABLED",
+        "domain": "googleapis.com",
+        "metadata": {
+          "service": "firebaseml.googleapis.com",
+          "consumer": "projects/12345"
+        }
+      }
+    ]
+  }
+}