From a2564cc3ea5c4860ee732f14cea9db2c10b52420 Mon Sep 17 00:00:00 2001 From: Yeesian Ng Date: Tue, 6 Jan 2026 14:38:22 -0800 Subject: [PATCH] fix: Update `examples_dataframe` type to `PandasDataFrame` in Prompt Optimizer. PiperOrigin-RevId: 852940223 --- vertexai/_genai/types/common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vertexai/_genai/types/common.py b/vertexai/_genai/types/common.py index bbe24d4095..3f657d0a38 100644 --- a/vertexai/_genai/types/common.py +++ b/vertexai/_genai/types/common.py @@ -3796,7 +3796,7 @@ class OptimizeConfig(_common.BaseModel): default=None, description="""The optimization target for the prompt optimizer. It must be one of the OptimizeTarget enum values: OPTIMIZATION_TARGET_GEMINI_NANO for the prompts from Android core API, OPTIMIZATION_TARGET_FEW_SHOT_RUBRICS for the few-shot prompt optimizer with rubrics, OPTIMIZATION_TARGET_FEW_SHOT_TARGET_RESPONSE for the few-shot prompt optimizer with target responses.""", ) - examples_dataframe: Optional[pd.DataFrame] = Field( + examples_dataframe: Optional[PandasDataFrame] = Field( default=None, description="""The examples dataframe for the few-shot prompt optimizer. It must contain "prompt" and "model_response" columns. Depending on which optimization target is used, it also needs to contain "rubrics" and "rubrics_evaluations" or "target_response" columns.""", ) @@ -3811,7 +3811,7 @@ class OptimizeConfigDict(TypedDict, total=False): optimization_target: Optional[OptimizeTarget] """The optimization target for the prompt optimizer. It must be one of the OptimizeTarget enum values: OPTIMIZATION_TARGET_GEMINI_NANO for the prompts from Android core API, OPTIMIZATION_TARGET_FEW_SHOT_RUBRICS for the few-shot prompt optimizer with rubrics, OPTIMIZATION_TARGET_FEW_SHOT_TARGET_RESPONSE for the few-shot prompt optimizer with target responses.""" - examples_dataframe: Optional[pd.DataFrame] + examples_dataframe: Optional[PandasDataFrame] """The examples dataframe for the few-shot prompt optimizer. It must contain "prompt" and "model_response" columns. Depending on which optimization target is used, it also needs to contain "rubrics" and "rubrics_evaluations" or "target_response" columns."""