Treat empty AG-UI state as missing

Author: MikeRyanDev

pydantic_ai_slim/pydantic_ai/ui/ag_ui/_adapter.py

@@ -2,11 +2,12 @@
 
 from __future__ import annotations
 
-from collections.abc import Sequence
+from collections.abc import Mapping, Sequence
 from functools import cached_property
 from typing import (
     TYPE_CHECKING,
     Any,
+    cast,
 )
 
 from ... import ExternalToolset, ToolDefinition
@@ -107,7 +108,14 @@ def toolset(self) -> AbstractToolset[AgentDepsT] | None:
     @cached_property
     def state(self) -> dict[str, Any] | None:
         """Frontend state from the AG-UI run input."""
-        return self.run_input.state
+        state = self.run_input.state
+        if state is None:
+            return None
+
+        if isinstance(state, Mapping) and not state:
+            return None
+
+        return cast('dict[str, Any]', state)
 
     @classmethod
     def load_messages(cls, messages: Sequence[Message]) -> list[ModelMessage]:

tests/test_ag_ui.py

@@ -1193,6 +1193,24 @@ async def test_request_with_state_without_handler() -> None:
             pass
 
 
+async def test_request_with_empty_state_without_handler() -> None:
+    agent = Agent(model=FunctionModel(stream_function=simple_stream))
+
+    run_input = create_input(
+        UserMessage(
+            id='msg_1',
+            content='Hello, how are you?',
+        ),
+        state={},
+    )
+
+    events = list[dict[str, Any]]()
+    async for event in run_ag_ui(agent, run_input):
+        events.append(json.loads(event.removeprefix('data: ')))
+
+    assert events == simple_result()
+
+
 async def test_request_with_state_with_custom_handler() -> None:
     @dataclass
     class CustomStateDeps: