[DB] Fix: align datetime writes with tz-naive columns (#3634)

Co-authored-by: Suchintan <suchintan@users.noreply.github.com>
This commit is contained in:
Celal Zamanoglu
2025-10-12 12:57:38 +03:00
committed by GitHub
parent a30c877408
commit 27b925a906

View File

@@ -1,5 +1,5 @@
import json import json
from datetime import datetime, timedelta, timezone from datetime import datetime, timedelta
from typing import Any, List, Literal, Sequence, overload from typing import Any, List, Literal, Sequence, overload
import structlog import structlog
@@ -577,7 +577,7 @@ class AgentDB:
step.status = status step.status = status
if status.is_terminal() and step.finished_at is None: if status.is_terminal() and step.finished_at is None:
step.finished_at = datetime.now(timezone.utc) step.finished_at = datetime.utcnow()
if output is not None: if output is not None:
step.output = output.model_dump(exclude_none=True) step.output = output.model_dump(exclude_none=True)
if is_last is not None: if is_last is not None:
@@ -4502,7 +4502,7 @@ class AgentDB:
WorkflowScriptModel.deleted_at.is_(None), WorkflowScriptModel.deleted_at.is_(None),
) )
) )
.values(deleted_at=datetime.now(timezone.utc)) .values(deleted_at=datetime.utcnow())
) )
result = await session.execute(stmt) result = await session.execute(stmt)
@@ -4539,7 +4539,7 @@ class AgentDB:
WorkflowScriptModel.deleted_at.is_(None), WorkflowScriptModel.deleted_at.is_(None),
) )
) )
.values(deleted_at=datetime.now(timezone.utc)) .values(deleted_at=datetime.utcnow())
) )
if statuses: if statuses: