chore: Add state_delta param to Runner.run

PiperOrigin-RevId: 929454639

Author: google-genai-bot

src/google/adk/runners.py

@@ -887,6 +887,7 @@ def run(
       user_id: str,
       session_id: str,
       new_message: types.Content,
+      state_delta: Optional[dict[str, Any]] = None,
       run_config: Optional[RunConfig] = None,
   ) -> Generator[Event, None, None]:
     """Runs the agent.
@@ -904,6 +905,7 @@ def run(
       user_id: The user ID of the session.
       session_id: The session ID of the session.
       new_message: A new message to append to the session.
+      state_delta: Optional state changes to apply to the session.
       run_config: The run config for the agent.
 
     Yields:
@@ -919,6 +921,7 @@ async def _invoke_run_async():
                 user_id=user_id,
                 session_id=session_id,
                 new_message=new_message,
+                state_delta=state_delta,
                 run_config=run_config,
             )
         ) as agen:

tests/unittests/test_runners.py

@@ -724,6 +724,45 @@ async def test_run_live_auto_create_session():
   assert session is not None
 
 
+def test_run_passes_state_delta():
+  """run should forward state_delta down to run_async."""
+  import asyncio
+
+  session_service = InMemorySessionService()
+  runner = Runner(
+      app_name=TEST_APP_ID,
+      agent=MockAgent("test_agent"),
+      session_service=session_service,
+      artifact_service=InMemoryArtifactService(),
+      auto_create_session=True,
+  )
+
+  state_delta = {"test_key": "test_value"}
+
+  events = list(
+      runner.run(
+          user_id=TEST_USER_ID,
+          session_id=TEST_SESSION_ID,
+          new_message=types.Content(
+              role="user", parts=[types.Part(text="hello")]
+          ),
+          state_delta=state_delta,
+      )
+  )
+
+  assert len(events) >= 1
+
+  session = asyncio.run(
+      session_service.get_session(
+          app_name=TEST_APP_ID, user_id=TEST_USER_ID, session_id=TEST_SESSION_ID
+      )
+  )
+  session_events = session.events
+
+  user_event = next(e for e in session_events if e.author == "user")
+  assert user_event.actions.state_delta == state_delta
+
+
 @pytest.mark.asyncio
 async def test_run_live_persists_event_callback_modifications():
   """run_live should persist the same event it streams after callback changes."""