diff --git a/agent.py b/agent.py index ff9e198..23ba458 100644 --- a/agent.py +++ b/agent.py @@ -60,6 +60,12 @@ class ContractClockStudioV1Config(BaseModel): pass +class ContractClockState(BaseModel): + """Durable-state marker; records live in managed Postgres tables.""" + + durable_store: str = "managed_postgres" + + class Deadline(BaseModel): kind: Literal["renewal", "notice", "obligation"] date: str @@ -143,6 +149,7 @@ class ContractClockStudioV1(A2AAgent[ContractClockStudioV1Config, PlatformUserAu ) resources = Resources(cpu="500m", memory="512Mi", max_runtime_seconds=120) state = State.DURABLE + state_model = ContractClockState tools_used = ("postgres", "mcp") workspace_access = WorkspaceAccess.dynamic( max_files=4,