Use standad userdata cells

quickstarts/Authentication.ipynb

@@ -131,6 +131,17 @@
         "You'll call `genai.configure` with your API key, but instead of pasting your key into the notebook, you'll read it from Colab Secrets."
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "562d43a97709"
+      },
+      "outputs": [],
+      "source": [
+        "import google.generativeai as genai"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -139,7 +150,6 @@
       },
       "outputs": [],
       "source": [
-        "import google.generativeai as genai\n",
         "from google.colab import userdata\n",
         "\n",
         "GOOGLE_API_KEY=userdata.get('GOOGLE_API_KEY')\n",

quickstarts/System_instructions.ipynb

@@ -84,6 +84,17 @@
         "To run the following cell, your API key must be stored it in a Colab Secret named `GOOGLE_API_KEY`. If you don't already have an API key, or you're not sure how to create a Colab Secret, see the [Authentication](https://github.com/google-gemini/cookbook/blob/main/quickstarts/Authentication.ipynb) quickstart for an example."
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "6872bcb201d3"
+      },
+      "outputs": [],
+      "source": [
+        "import google.generativeai as genai"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": 4,
@@ -93,8 +104,6 @@
       "outputs": [],
       "source": [
         "from google.colab import userdata\n",
-        "import google.generativeai as genai\n",
-        "\n",
         "genai.configure(api_key=userdata.get(\"GOOGLE_API_KEY\"))"
       ]
     },

quickstarts/rest/Safety_REST.ipynb

@@ -121,7 +121,7 @@
       },
       "outputs": [],
       "source": [
-        "os.environ['UNSAFE_PROMPT'] = "
+        "os.environ['UNSAFE_PROMPT'] = \"<UNSAFE_PROMPT>\""
       ]
     },
     {