fix: GenAI Client(evals): Avoid required session_inputs column to eval dataset

PiperOrigin-RevId: 877610185

Author: vertex-sdk-bot

tests/unit/vertexai/genai/test_evals.py

@@ -4439,6 +4439,49 @@ async def test_run_adk_user_simulation_missing_columns(
         with pytest.raises(ValueError, match="User simulation requires"):
             await _evals_common._run_adk_user_simulation(row, mock_agent)
 
+    @mock.patch("vertexai._genai._evals_common.ADK_SessionInput")  # fmt: skip
+    @mock.patch("vertexai._genai._evals_common.EvaluationGenerator")  # fmt: skip
+    @mock.patch("vertexai._genai._evals_common.LlmBackedUserSimulator")  # fmt: skip
+    @mock.patch("vertexai._genai._evals_common.ConversationScenario")  # fmt: skip
+    @mock.patch("vertexai._genai._evals_common.LlmBackedUserSimulatorConfig")  # fmt: skip
+    @pytest.mark.asyncio
+    async def test_run_adk_user_simulation_missing_session_inputs(
+        self,
+        mock_config_cls,
+        mock_scenario_cls,
+        mock_simulator_cls,
+        mock_generator_cls,
+        mock_session_input_cls,
+    ):
+        row = pd.Series(
+            {
+                "starting_prompt": "start",
+                "conversation_plan": "plan",
+            }
+        )
+        mock_agent = mock.Mock()
+        mock_invocation = mock.Mock()
+        mock_invocation.user_content.model_dump.return_value = {"text": "user msg"}
+        mock_invocation.final_response.model_dump.return_value = {"text": "agent msg"}
+        mock_invocation.intermediate_data = None
+        mock_invocation.creation_timestamp = 12345
+        mock_invocation.invocation_id = "turn1"
+
+        mock_generator_cls._generate_inferences_from_root_agent = mock.AsyncMock(
+            return_value=[mock_invocation]
+        )
+
+        await _evals_common._run_adk_user_simulation(row, mock_agent)
+
+        mock_scenario_cls.assert_called_once_with(
+            starting_prompt="start", conversation_plan="plan"
+        )
+        mock_session_input_cls.assert_called_once_with(
+            app_name="user_simulation_app",
+            user_id="user_simulation_default_user",
+            state={},
+        )
+
 
 @pytest.mark.usefixtures("google_auth_mock")
 class TestLLMMetricHandlerPayload:

vertexai/_genai/_evals_common.py

@@ -856,16 +856,24 @@ async def _run_adk_user_simulation(
         conversation_scenario=scenario, config=user_simulator_config
     )
 
-    initial_session = _get_session_inputs(row)
+    try:
+        initial_session = _get_session_inputs(row)
+        app_name = initial_session.app_name or "user_simulation_app"
+        user_id = initial_session.user_id or "user_simulation_default_user"
+        state = initial_session.state or {}
+    except (KeyError, TypeError, ValueError):
+        app_name = "user_simulation_app"
+        user_id = "user_simulation_default_user"
+        state = {}
 
     invocations = await EvaluationGenerator._generate_inferences_from_root_agent(  # pylint: disable=protected-access
         root_agent=agent,
         user_simulator=user_simulator,
         reset_func=getattr(agent, "reset_data", None),
         initial_session=ADK_SessionInput(
-            app_name=initial_session.app_name or "user_simulation_app",
-            user_id=initial_session.user_id or "user_simulation_default_user",
-            state=initial_session.state or {},
+            app_name=app_name,
+            user_id=user_id,
+            state=state,
         ),
     )