feat: Add support for passing other data and agent data for custom code execution metric

PiperOrigin-RevId: 886400009

Author: vertex-sdk-bot

vertexai/_genai/_evals_metric_handlers.py

@@ -1296,7 +1296,7 @@ def __init__(
     def _build_request_payload(
         self, eval_case: types.EvalCase, response_index: int
     ) -> dict[str, Any]:
-        """Builds request payload for registered metric by assembling EvaluationInstance."""
+        """Builds the request parameters for evaluate instances request."""
         response_content = _get_response_from_eval_case(
             eval_case, response_index, self.metric_name
         )
@@ -1317,20 +1317,39 @@ def _build_request_payload(
             extracted_prompt
         )
 
+        other_data_map: dict[str, Any] = {}
+        if hasattr(eval_case, "context") and eval_case.context:
+            if isinstance(eval_case.context, str):
+                other_data_map["context"] = types.evals.InstanceData(
+                    text=eval_case.context
+                )
+            elif isinstance(eval_case.context, genai_types.Content):
+                other_data_map["context"] = (
+                    PredefinedMetricHandler._content_to_instance_data(eval_case.context)
+                )
+            else:
+                logger.warning(
+                    f"Unsupported type for context: {type(eval_case.context)}"
+                )
+
         instance_payload = types.EvaluationInstance(
             prompt=prompt_instance_data,
             response=PredefinedMetricHandler._content_to_instance_data(
                 response_content
             ),
             reference=reference_instance_data,
             rubric_groups=eval_case.rubric_groups,
+            other_data=(
+                types.MapInstance(map_instance=other_data_map)
+                if other_data_map
+                else None
+            ),
             agent_data=PredefinedMetricHandler._eval_case_to_agent_data(eval_case),
         )
 
-        request_payload = {
+        return {
             "instance": instance_payload,
         }
-        return request_payload
 
     @property
     def metric_name(self) -> str: