adding importing from env

STRIDES · Nov 20, 2024 · c19fe7e · c19fe7e
1 parent 0b5c513
commit c19fe7e
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/notebooks/GenAI/Gemini_Intro.ipynb b/notebooks/GenAI/Gemini_Intro.ipynb
@@ -95,9 +95,22 @@
     "import vertexai\n",
     "from vertexai.generative_models import GenerativeModel, Image, GenerativeModel, ChatSession, Part, GenerationConfig\n",
     "\n",
-    "# TODO(developer): Update and un-comment below lines\n",
-    "project_id = \"<PROJECT_ID>\"\n",
-    "location = \"<LOCATION>\" #(e.g., us-central1)\n",
+    "import json\n",
+    "# Load env.json\n",
+    "try:\n",
+    "    with open(\"env.json\") as f:\n",
+    "        config = json.load(f)\n",
+    "except FileNotFoundError:\n",
+    "    config = {}\n",
+    "\n",
+    "# Assign parameters from Env.\n",
+    "project_id = config.get(\"NOTEBOOK_GCP_PROJECT_ID\")\n",
+    "location = config.get(\"NOTEBOOK_GCP_LOCATION\")\n",
+    "\n",
+    "# TODO( FOR developer): If not defined in ENV earlier, uncomment and add it below\n",
+    "#project_id = \"<PROJECT_ID>\"\n",
+    "#location = \"<LOCATION>\" #(e.g., us-central1)\n",
+    "\n",
     "vertexai.init(project=project_id, location=location)"
    ]
   },