AgentStateT -> StateT

Author: wdoppenberg

python-sdk/ag_ui/core/events.py

@@ -7,7 +7,7 @@
 
 from pydantic import Field
 
-from .types import ConfiguredBaseModel, Message, AgentStateT, JSONValue
+from .types import ConfiguredBaseModel, Message, StateT, JSONValue
 
 
 class EventType(str, Enum):
@@ -193,13 +193,13 @@ class ThinkingEndEvent(BaseEvent):
     type: Literal[EventType.THINKING_END] = EventType.THINKING_END  # pyright: ignore[reportIncompatibleVariableOverride]
 
 
-class StateSnapshotEvent(BaseEvent, Generic[AgentStateT]):
+class StateSnapshotEvent(BaseEvent, Generic[StateT]):
     """
     Event containing a snapshot of the state.
     """
 
     type: Literal[EventType.STATE_SNAPSHOT] = EventType.STATE_SNAPSHOT  # pyright: ignore[reportIncompatibleVariableOverride]
-    snapshot: AgentStateT
+    snapshot: StateT
 
 
 class StateDeltaEvent(BaseEvent):

python-sdk/ag_ui/core/types.py

@@ -9,7 +9,7 @@
 from pydantic.alias_generators import to_camel
 
 JSONValue = Union[str, int, float, bool, None, dict[str, Any], list[Any]]
-AgentStateT = TypeVar("AgentStateT", default=JSONValue, contravariant=True)
+StateT = TypeVar("StateT", default=JSONValue, contravariant=True)
 FwdPropsT = TypeVar("FwdPropsT", default=JSONValue, contravariant=True)
 
 
@@ -127,14 +127,14 @@ class Tool(ConfiguredBaseModel):
     parameters: Any  # JSON Schema for the tool parameters
 
 
-class RunAgentInput(ConfiguredBaseModel, Generic[AgentStateT, FwdPropsT]):
+class RunAgentInput(ConfiguredBaseModel, Generic[StateT, FwdPropsT]):
     """
     Input for running an agent.
     """
 
     thread_id: str
     run_id: str
-    state: AgentStateT
+    state: StateT
     messages: List[Message]
     tools: List[Tool]
     context: List[Context]