feat: GenAI Client(evals) - Add pd.DataFrame as input for dataset in create_evaluation_runin Vertex AI GenAI SDK evals

PiperOrigin-RevId: 825658094

Author: vertex-sdk-bot

vertexai/_genai/evals.py

@@ -1332,7 +1332,7 @@ def get_evaluation_run(
     def create_evaluation_run(
         self,
         *,
-        dataset: Union[types.EvaluationRunDataSource, types.EvaluationDataset],
+        dataset: Union[types.EvaluationRunDataSource, types.EvaluationDatasetOrDict],
         dest: str,
         name: Optional[str] = None,
         display_name: Optional[str] = None,
@@ -1346,7 +1346,7 @@ def create_evaluation_run(
         """Creates an EvaluationRun.
 
         Args:
-          dataset: The dataset to evaluate. Either an EvaluationRunDataSource or an EvaluationDataset.
+          dataset: The dataset to evaluate. Either an EvaluationRunDataSource or an EvaluationDataset or dict.
           dest: The GCS URI prefix to write the evaluation results to.
           name: The name of the evaluation run.
           display_name: The display name of the evaluation run.
@@ -1358,6 +1358,8 @@ def create_evaluation_run(
         Returns:
             The created evaluation run.
         """
+        if isinstance(dataset, dict):
+            dataset = types.EvaluationDataset.model_validate(dataset)
         if type(dataset).__name__ == "EvaluationDataset":
             logger.warning(
                 "EvaluationDataset input is experimental and may change in future versions."
@@ -2185,7 +2187,7 @@ async def get_evaluation_run(
     async def create_evaluation_run(
         self,
         *,
-        dataset: Union[types.EvaluationRunDataSource, types.EvaluationDataset],
+        dataset: Union[types.EvaluationRunDataSource, types.EvaluationDatasetOrDict],
         dest: str,
         name: Optional[str] = None,
         display_name: Optional[str] = None,
@@ -2199,7 +2201,7 @@ async def create_evaluation_run(
         """Creates an EvaluationRun.
 
         Args:
-          dataset: The dataset to evaluate. Either an EvaluationRunDataSource or an EvaluationDataset.
+          dataset: The dataset to evaluate. Either an EvaluationRunDataSource, EvaluationDataset, or a dataframe.
           dest: The GCS URI prefix to write the evaluation results to.
           name: The name of the evaluation run.
           display_name: The display name of the evaluation run.
@@ -2211,6 +2213,8 @@ async def create_evaluation_run(
         Returns:
             The created evaluation run.
         """
+        if isinstance(dataset, dict):
+            dataset = types.EvaluationDataset.model_validate(dataset)
         if type(dataset).__name__ == "EvaluationDataset":
             logger.warning(
                 "EvaluationDataset input is experimental and may change in future versions."