Story writing - Use standard userdata cell

examples/Story_Writing_with_Prompt_Chaining.ipynb

@@ -113,18 +113,37 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "sTedGxE7_eah"
+        "id": "ed07713014be"
       },
       "outputs": [],
       "source": [
         "import google.generativeai as genai\n",
         "from google.api_core import retry\n",
-        "from google.colab import userdata\n",
-        "from pprint import pprint\n",
         "\n",
+        "from pprint import pprint"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "14c8b88a75ef"
+      },
+      "outputs": [],
+      "source": [
+        "from google.colab import userdata\n",
         "GOOGLE_API_KEY=userdata.get('GOOGLE_API_KEY')\n",
-        "genai.configure(api_key=GOOGLE_API_KEY)\n",
-        "\n",
+        "genai.configure(api_key=GOOGLE_API_KEY)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "sTedGxE7_eah"
+      },
+      "outputs": [],
+      "source": [
         "model = genai.GenerativeModel('gemini-1.0-pro')\n",
         "\n",
         "# For convenience, a simple wrapper to let the SDK handle error retries\n",