Skip to content

Commit e383c94

Browse files
authored
fix: Upgrade gemin model used in data_warehouse (#397)
1 parent 2f7acb8 commit e383c94

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

examples/data_warehouse/main.tf

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ module "data_warehouse" {
1919
version = "~> 9.0"
2020

2121
project_id = var.project_id
22-
region = "asia-southeast1"
22+
region = "us-central1"
2323
deletion_protection = false
2424
force_destroy = true
2525
}

modules/data_warehouse/src/sql/sp_bigqueryml_generate_create.sql

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,4 @@
1414

1515
CREATE OR REPLACE MODEL `${project_id}.${dataset_id}.${model_name}`
1616
REMOTE WITH CONNECTION `${connection_id}`
17-
OPTIONS (ENDPOINT = 'gemini-1.5-flash');
17+
OPTIONS (ENDPOINT = 'gemini-2.0-flash');

test/integration/data_warehouse/data_warehouse_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,11 @@ func TestDataWarehouse(t *testing.T) {
4646
projectID := dwh.GetTFSetupStringOutput("project_id")
4747
bucket := dwh.GetStringOutput("raw_bucket")
4848
workflow := "initial-workflow"
49-
location := "asia-southeast1"
49+
location := "us-central1"
5050

5151
// Assert that the bucket is in location defined above
5252
bucketOP := gcloud.Runf(t, "storage buckets describe gs://%s --project %s", bucket, projectID)
53-
assert.Equal("ASIA-SOUTHEAST1", bucketOP.Get("location").String(), "Bucket should be in %[1]s \n", strings.ToUpper(location))
53+
assert.Equal("US-CENTRAL1", bucketOP.Get("location").String(), "Bucket should be in %[1]s \n", strings.ToUpper(location))
5454

5555
// Assert that Workflow ran successfully
5656
verifyWorkflows := func() (bool, error) {

0 commit comments

Comments
 (0)