From 28ce8f6d7a0c02fcc27fc9524c43fd47b6aab2cb Mon Sep 17 00:00:00 2001
From: Eric Dong <itseric@google.com>
Date: Tue, 24 Mar 2026 09:42:45 -0400
Subject: [PATCH] chore: Update readme and template (#4480)

---
 README.md                         |  2 +-
 docs/README.md                    |  7 +++++++
 notebooks/notebook_template.ipynb | 19 ++++++++++++++-----
 3 files changed, 22 insertions(+), 6 deletions(-)
 create mode 100644 docs/README.md

diff --git a/README.md b/README.md
index 4e2b02d8c..67849dcab 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # ![Google Cloud](https://avatars.githubusercontent.com/u/2810941?s=60&v=4) Google Cloud Vertex AI Samples
 
-This repository contains notebooks, code samples, sample apps, and other resources that demonstrate how to use, develop and manage machine learning and generative AI workflows using Google Cloud Vertex AI.
+This repository contains notebooks, code samples, sample apps, skills, and other resources that demonstrate how to use, develop and manage machine learning and generative AI workflows using Google Cloud Vertex AI.
 
 ## Overview
 
diff --git a/docs/README.md b/docs/README.md
new file mode 100644
index 000000000..e4bda1e3f
--- /dev/null
+++ b/docs/README.md
@@ -0,0 +1,7 @@
+# Vertex AI Documentation
+
+This directory contains deep-dive documentation, extended guides, and architectural references for Google Cloud Vertex AI infrastructure.
+
+
+## Contents
+- **`vertex-training-cluster/`**: Documentation and setup guides for configuring and managing Vertex AI training clusters.
diff --git a/notebooks/notebook_template.ipynb b/notebooks/notebook_template.ipynb
index 8357462d6..8f9549c29 100644
--- a/notebooks/notebook_template.ipynb
+++ b/notebooks/notebook_template.ipynb
@@ -8,7 +8,7 @@
       },
       "outputs": [],
       "source": [
-        "# Copyright 2026 Google LLC\n",
+        "# @title Copyright 2026 Google LLC\n",
         "#\n",
         "# Licensed under the Apache License, Version 2.0 (the \"License\");\n",
         "# you may not use this file except in compliance with the License.\n",
@@ -96,7 +96,7 @@
       },
       "outputs": [],
       "source": [
-        "! pip3 install --upgrade --quiet google-cloud-aiplatform"
+        "! pip3 install --upgrade --quiet google-genai google-cloud-aiplatform"
       ]
     },
     {
@@ -119,7 +119,12 @@
         "import os\n",
         "import sys\n",
         "\n",
-        "import vertexai"
+        "# Vertex AI SDK\n",
+        "# import vertexai\n",
+        "\n",
+        "# Gen AI SDK\n",
+        "# from google import genai\n",
+        "# from google.genai import types"
       ]
     },
     {
@@ -179,7 +184,7 @@
         "id": "d95621cbf42f"
       },
       "source": [
-        "### Initialize Vertex AI SDK"
+        "### Initialize Vertex AI SDK or Gen AI SDK"
       ]
     },
     {
@@ -190,7 +195,11 @@
       },
       "outputs": [],
       "source": [
-        "vertexai.init(project=PROJECT_ID, location=LOCATION)"
+        "# Vertex AI SDK\n",
+        "# vertexai.init(project=PROJECT_ID, location=LOCATION)\n",
+        "\n",
+        "# Gen AI SDK\n",
+        "# client = genai.Client(vertexai=True, project=PROJECT_ID, location=LOCATION)"
       ]
     },
     {