fix(fastapi): pass state_delta to runner in /run endpoint

Author: pguerra-ce

src/google/adk/cli/adk_web_server.py

@@ -993,6 +993,7 @@ async def run_agent(req: RunAgentRequest) -> list[Event]:
               user_id=req.user_id,
               session_id=req.session_id,
               new_message=req.new_message,
+              state_delta=req.state_delta,
           )
       ) as agen:
         events = [event async for event in agen]

tests/unittests/cli/test_fast_api.py

@@ -94,6 +94,19 @@ def _event_3():
   )
 
 
+def _event_state_delta(state_delta: dict[str, Any]):
+  return Event(
+      author="dummy agent",
+      invocation_id="invocation_id",
+      content=types.Content(
+          role="model",
+          parts=[
+              types.Part(text=json.dumps(state_delta, sort_keys=True))
+          ],
+      ),
+  )
+
+
 # Define mocked async generator functions for the Runner
 async def dummy_run_live(self, session, live_request_queue):
   yield _event_1()
@@ -110,6 +123,7 @@ async def dummy_run_async(
     user_id,
     session_id,
     new_message,
+    state_delta=None,
     run_config: RunConfig = RunConfig(),
 ):
   yield _event_1()
@@ -119,6 +133,10 @@ async def dummy_run_async(
   await asyncio.sleep(0)
 
   yield _event_3()
+  await asyncio.sleep(0)
+
+  if state_delta is not None:
+    yield _event_state_delta(state_delta)
 
 
 # Define a local mock for EvalCaseResult specific to fast_api tests
@@ -743,6 +761,26 @@ def test_agent_run(test_app, create_test_session):
 
   logger.info("Agent run test completed successfully")
 
+def test_agent_run_passes_state_delta(test_app, create_test_session):
+  """Test /run forwards state_delta and surfaces it in events."""
+  info = create_test_session
+  payload = {
+      "app_name": info["app_name"],
+      "user_id": info["user_id"],
+      "session_id": info["session_id"],
+      "new_message": {"role": "user", "parts": [{"text": "Hello"}]},
+      "streaming": False,
+      "state_delta": {"k": "v", "count": 1},
+  }
+
+  response = test_app.post("/run", json=payload)
+  assert response.status_code == 200
+  data = response.json()
+  assert isinstance(data, list)
+  assert len(data) == 4
+
+  text = data[3]["content"]["parts"][0]["text"]
+  assert json.loads(text) == payload["state_delta"]
 
 def test_list_artifact_names(test_app, create_test_session):
   """Test listing artifact names for a session."""