Change type of examples_dataframe to PandasDataFrame

silverdev · web-flow · commit 7b998df86d65 · 2026-01-06T10:40:10.000-06:00
Updated reference to fix issue where imports to fail when pandas is not installed.
diff --git 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.""",
     )

Original file line number	Diff line number	Diff line change
`@@ -3796,7 +3796,7 @@ class OptimizeConfig(_common.BaseModel):`
`3796`	`3796`	`default=None,`
`3797`	`3797`	`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.""",`
`3798`	`3798`	`)`
`3799`		`- examples_dataframe: Optional[pd.DataFrame] = Field(`
	`3799`	`+ examples_dataframe: Optional[PandasDataFrame] = Field(`
`3800`	`3800`	`default=None,`
`3801`	`3801`	`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.""",`
`3802`	`3802`	`)`