patch colab credentials

Change-Id: I5a3cb3168448a565eb3cdc8a0063ae041c41a260

google/generativeai/client.py

@@ -30,6 +30,18 @@
 GENAI_API_DISCOVERY_URL = "https://generativelanguage.googleapis.com/$discovery/rest"
 
 
+@contextmanager
+def patch_colab_gce_credentials():
+    get_gce = google.auth._default._get_gce_credentials
+    if "COLAB_RELEASE_TAG" in os.environ:
+        google.auth._default._get_gce_credentials = (lambda *args, **kwargs: None, None)
+
+    try:
+        yield
+    finally:
+        google.auth._default._get_gce_credentials = get_gce
+
+
 class FileServiceClient(glm.FileServiceClient):
     def __init__(self, *args, **kwargs):
         self._discovery_api = None
@@ -184,12 +196,15 @@ def make_client(self, name):
             configure()
 
         try:
-            client = cls(**self.client_config)
+            with patch_colab_gce_credentials():
+                client = cls(**self.client_config)
         except ga_exceptions.DefaultCredentialsError as e:
-            e.args = ("\n  No API_KEY or ADC found. Please either:\n"
-                      "    - Set the `GOOGLE_API_KEY` environment variable.\n"
-                      "    - Manually pass the key with `genai.configure(api_key=my_api_key)`.\n"
-                      "    - Or set up Application Default Credentials, see https://ai.google.dev/gemini-api/docs/oauth for more information.",)
+            e.args = (
+                "\n  No API_KEY or ADC found. Please either:\n"
+                "    - Set the `GOOGLE_API_KEY` environment variable.\n"
+                "    - Manually pass the key with `genai.configure(api_key=my_api_key)`.\n"
+                "    - Or set up Application Default Credentials, see https://ai.google.dev/gemini-api/docs/oauth for more information.",
+            )
             raise e
 
         if not self.default_metadata: