This is an automated email from the ASF dual-hosted git repository.
shahar pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 62010b752f5 Change region in gen_ai generative_model system test
(#61545)
62010b752f5 is described below
commit 62010b752f5b71dcea06368921fea635e318e791
Author: Nitochkin <[email protected]>
AuthorDate: Fri Feb 6 21:08:54 2026 +0100
Change region in gen_ai generative_model system test (#61545)
---
.../google/cloud/gen_ai/example_gen_ai_generative_model.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git
a/providers/google/tests/system/google/cloud/gen_ai/example_gen_ai_generative_model.py
b/providers/google/tests/system/google/cloud/gen_ai/example_gen_ai_generative_model.py
index ad902c8d4a5..516adab5178 100644
---
a/providers/google/tests/system/google/cloud/gen_ai/example_gen_ai_generative_model.py
+++
b/providers/google/tests/system/google/cloud/gen_ai/example_gen_ai_generative_model.py
@@ -165,6 +165,7 @@ EVAL_DATASET = {
"Baking a decadent chocolate cake requires creaming butter and sugar,
beating in eggs and alternating dry ingredients with buttermilk before baking
until done.",
],
}
+REGION_GLOBAL = "global"
def _get_metrics():
@@ -256,7 +257,7 @@ with DAG(
task_id="count_tokens_task",
project_id=PROJECT_ID,
contents=CONTENTS,
- location=REGION,
+ location=REGION_GLOBAL,
model=MULTIMODAL_MODEL,
)
# [END how_to_cloud_gen_ai_count_tokens_operator]
@@ -266,7 +267,7 @@ with DAG(
task_id="generate_content_task",
project_id=PROJECT_ID,
contents=CONTENTS,
- location=REGION,
+ location=REGION_GLOBAL,
generation_config=GENERATION_CONFIG_CREATE_CONTENT,
model=MULTIMODAL_MODEL,
)
@@ -312,7 +313,7 @@ with DAG(
create_cached_content_task = GenAICreateCachedContentOperator(
task_id="create_cached_content_task",
project_id=PROJECT_ID,
- location=REGION,
+ location=REGION_GLOBAL,
model=CACHED_MODEL,
cached_content_config=CACHED_CONTENT_CONFIG,
)
@@ -322,7 +323,7 @@ with DAG(
generate_from_cached_content_task = GenAIGenerateContentOperator(
task_id="generate_from_cached_content_task",
project_id=PROJECT_ID,
- location=REGION,
+ location=REGION_GLOBAL,
contents=["What are the papers about?"],
generation_config={
"cached_content": create_cached_content_task.output,