diff --git a/genai/content_cache/content_cache_test.go b/genai/content_cache/content_cache_test.go index 1c9253efa1..da8746a0b0 100644 --- a/genai/content_cache/content_cache_test.go +++ b/genai/content_cache/content_cache_test.go @@ -27,7 +27,8 @@ func TestContentCaching(t *testing.T) { tc := testutil.SystemTest(t) t.Setenv("GOOGLE_GENAI_USE_VERTEXAI", "1") - t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + // t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + t.Setenv("GOOGLE_CLOUD_LOCATION", "global") t.Setenv("GOOGLE_CLOUD_PROJECT", tc.ProjectID) buf := new(bytes.Buffer) diff --git a/genai/controlled_generation/controlled_generation_examples_test.go b/genai/controlled_generation/controlled_generation_examples_test.go index 311e86129f..17b9dbe54a 100644 --- a/genai/controlled_generation/controlled_generation_examples_test.go +++ b/genai/controlled_generation/controlled_generation_examples_test.go @@ -25,7 +25,8 @@ func TestTextGeneration(t *testing.T) { tc := testutil.SystemTest(t) t.Setenv("GOOGLE_GENAI_USE_VERTEXAI", "1") - t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + // t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + t.Setenv("GOOGLE_CLOUD_LOCATION", "global") t.Setenv("GOOGLE_CLOUD_PROJECT", tc.ProjectID) buf := new(bytes.Buffer) diff --git a/genai/count_tokens/count_tokens_examples_test.go b/genai/count_tokens/count_tokens_examples_test.go index 22da59959a..64dea2d4b8 100644 --- a/genai/count_tokens/count_tokens_examples_test.go +++ b/genai/count_tokens/count_tokens_examples_test.go @@ -25,7 +25,8 @@ func TestTextGeneration(t *testing.T) { tc := testutil.SystemTest(t) t.Setenv("GOOGLE_GENAI_USE_VERTEXAI", "1") - t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + // t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + t.Setenv("GOOGLE_CLOUD_LOCATION", "global") t.Setenv("GOOGLE_CLOUD_PROJECT", tc.ProjectID) buf := new(bytes.Buffer) diff --git a/genai/text_generation/text_generation_examples_test.go b/genai/text_generation/text_generation_examples_test.go index 453a57463b..c87022490c 100644 --- a/genai/text_generation/text_generation_examples_test.go +++ b/genai/text_generation/text_generation_examples_test.go @@ -25,7 +25,8 @@ func TestTextGeneration(t *testing.T) { tc := testutil.SystemTest(t) t.Setenv("GOOGLE_GENAI_USE_VERTEXAI", "1") - t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + // t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + t.Setenv("GOOGLE_CLOUD_LOCATION", "global") t.Setenv("GOOGLE_CLOUD_PROJECT", tc.ProjectID) buf := new(bytes.Buffer) diff --git a/genai/tools/tools_examples_test.go b/genai/tools/tools_examples_test.go index 867e868ae1..fb5534b874 100644 --- a/genai/tools/tools_examples_test.go +++ b/genai/tools/tools_examples_test.go @@ -25,7 +25,8 @@ func TestTextGeneration(t *testing.T) { tc := testutil.SystemTest(t) t.Setenv("GOOGLE_GENAI_USE_VERTEXAI", "1") - t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + // t.Setenv("GOOGLE_CLOUD_LOCATION", "us-central1") + t.Setenv("GOOGLE_CLOUD_LOCATION", "global") t.Setenv("GOOGLE_CLOUD_PROJECT", tc.ProjectID) buf := new(bytes.Buffer)