From 9e80e1d9abd79ef11b0537e8ea36e131bc5a3332 Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Thu, 16 Jul 2026 10:12:20 -0700 Subject: [PATCH] chore: Fix type error in `_get_default_prompt_template` for optional `.values` PiperOrigin-RevId: 949036874 --- agentplatform/_genai/_evals_common.py | 8 ++++---- vertexai/_genai/_evals_common.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/agentplatform/_genai/_evals_common.py b/agentplatform/_genai/_evals_common.py index f33ceb7497..88c61441cd 100644 --- a/agentplatform/_genai/_evals_common.py +++ b/agentplatform/_genai/_evals_common.py @@ -484,10 +484,10 @@ def _get_default_prompt_template( and eval_item.evaluation_request.prompt and eval_item.evaluation_request.prompt.prompt_template_data ): - if ( - "prompt" - in eval_item.evaluation_request.prompt.prompt_template_data.values - ): + template_values = ( + eval_item.evaluation_request.prompt.prompt_template_data.values + ) + if template_values and "prompt" in template_values: return "{prompt}" except Exception as e: logger.warning("Failed to get prompt template from evaluation set: %s", e) diff --git a/vertexai/_genai/_evals_common.py b/vertexai/_genai/_evals_common.py index 4936a60a96..4d406b7b10 100644 --- a/vertexai/_genai/_evals_common.py +++ b/vertexai/_genai/_evals_common.py @@ -469,10 +469,10 @@ def _get_default_prompt_template( and eval_item.evaluation_request.prompt and eval_item.evaluation_request.prompt.prompt_template_data ): - if ( - "prompt" - in eval_item.evaluation_request.prompt.prompt_template_data.values - ): + template_values = ( + eval_item.evaluation_request.prompt.prompt_template_data.values + ) + if template_values and "prompt" in template_values: return "{prompt}" except Exception as e: logger.warning("Failed to get prompt template from evaluation set: %s", e)