chore: Update Gen AI types

sasha-gitg · copybara-github · commit b7968d8fb303 · 2025-05-29T16:14:38.000-07:00
PiperOrigin-RevId: 764924433
diff --git a/vertexai/_genai/evals.py b/vertexai/_genai/evals.py
@@ -690,6 +690,7 @@ def _EvaluationDataset_to_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
     to_object: dict[str, Any] = {}
+
     if getv(from_object, ["bigquery_source"]) is not None:
         setv(
             to_object,
@@ -966,6 +967,7 @@ def _EvaluationDataset_from_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
     to_object: dict[str, Any] = {}
+
     if getv(from_object, ["bigquerySource"]) is not None:
         setv(
             to_object,
diff --git a/vertexai/_genai/types.py b/vertexai/_genai/types.py
@@ -1804,6 +1804,25 @@ class EvaluateInstancesResponseDict(TypedDict, total=False):
 ]
 
 
+class EvalCase(_common.BaseModel):
+    """A comprehensive representation of a GenAI interaction for evaluation."""
+
+    prompt: Optional[genai_types.Content] = Field(
+        default=None,
+        description="""The most recent user message (current input).""",
+    )
+
+
+class EvalCaseDict(TypedDict, total=False):
+    """A comprehensive representation of a GenAI interaction for evaluation."""
+
+    prompt: Optional[genai_types.Content]
+    """The most recent user message (current input)."""
+
+
+EvalCaseOrDict = Union[EvalCase, EvalCaseDict]
+
+
 class BigQuerySource(_common.BaseModel):
     """The BigQuery location for the input content."""
 
@@ -1845,6 +1864,9 @@ class GcsSourceDict(TypedDict, total=False):
 class EvaluationDataset(_common.BaseModel):
     """The dataset used for evaluation."""
 
+    eval_cases: Optional[list[EvalCase]] = Field(
+        default=None, description="""The evaluation cases to be evaluated."""
+    )
     bigquery_source: Optional[BigQuerySource] = Field(
         default=None, description="""BigQuery source holds the dataset."""
     )
@@ -1857,6 +1879,9 @@ class EvaluationDataset(_common.BaseModel):
 class EvaluationDatasetDict(TypedDict, total=False):
     """The dataset used for evaluation."""
 
+    eval_cases: Optional[list[EvalCaseDict]]
+    """The evaluation cases to be evaluated."""
+
     bigquery_source: Optional[BigQuerySourceDict]
     """BigQuery source holds the dataset."""
 
