feat: checkpoint device agent runtime milestones
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from collections import deque
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
from core.models import Scene, utc_now
|
||||
from semantic.models import SemanticScene
|
||||
from world.config import DEFAULT_HISTORY_SIZE
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class WorldEvent:
|
||||
scene_summary: SemanticScene | Scene
|
||||
action: str
|
||||
success: bool
|
||||
timestamp: datetime = field(default_factory=utc_now)
|
||||
|
||||
def to_dict(self) -> dict[str, Any]:
|
||||
return {
|
||||
"scene_summary": self.scene_summary.to_dict(),
|
||||
"action": self.action,
|
||||
"success": self.success,
|
||||
"timestamp": self.timestamp.isoformat(),
|
||||
}
|
||||
|
||||
|
||||
@dataclass
|
||||
class WorldState:
|
||||
current_app: str | None = None
|
||||
current_page: str | None = None
|
||||
variables: dict[str, Any] = field(default_factory=dict)
|
||||
history: deque[WorldEvent] = field(
|
||||
default_factory=lambda: deque(maxlen=DEFAULT_HISTORY_SIZE)
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def with_history_bound(cls, history_size: int) -> "WorldState":
|
||||
maxlen = history_size if history_size > 0 else DEFAULT_HISTORY_SIZE
|
||||
return cls(history=deque(maxlen=maxlen))
|
||||
|
||||
def to_dict(self) -> dict[str, Any]:
|
||||
return {
|
||||
"current_app": self.current_app,
|
||||
"current_page": self.current_page,
|
||||
"variables": dict(self.variables),
|
||||
"history": [event.to_dict() for event in self.history],
|
||||
}
|
||||
Reference in New Issue
Block a user