add regional bucket and fix deployed index access

GoogleCloudPlatform · Jun 6, 2024 · d1a02dc · d1a02dc
1 parent e0a50f2
commit d1a02dc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/examples/genai-rag-multimodal/multimodal_rag_langchain.ipynb b/examples/genai-rag-multimodal/multimodal_rag_langchain.ipynb
@@ -699,7 +699,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "SERVICE_ATTACHMENT=index_endpoint.deployed_indexes.private_endpoints.service_attachment\n",
+    "SERVICE_ATTACHMENT=index_endpoint.deployed_indexes[0].private_endpoints.service_attachment\n",
     "\n",
     "SERVICE_ATTACHMENT"
    ]

diff --git a/examples/genai-rag-multimodal/multimodal_rag_langchain_infra.tf b/examples/genai-rag-multimodal/multimodal_rag_langchain_infra.tf
@@ -75,7 +75,8 @@ resource "random_string" "suffix" {
 
 resource "google_storage_bucket" "vector_search_bucket" {
   name                        = "vector-search-${random_string.suffix.result}"
-  location                    = "US"
+  location                    = var.vector_search_bucket_location
+  storage_class               = "REGIONAL"
   project                     = var.machine_learning_project
   uniform_bucket_level_access = true
 }

diff --git a/examples/genai-rag-multimodal/variables.tf b/examples/genai-rag-multimodal/variables.tf
@@ -4,6 +4,12 @@ variable "service_account_name" {
   default     = "rag-notebook-runner"
 }
 
+variable "vector_search_bucket_location" {
+  description = "Bucket Region"
+  type        = string
+  default     = "US-CENTRAL1"
+}
+
 variable "machine_learning_project" {
   description = "Machine Learning Project ID"
   type        = string