diff --git a/document-parser/persistence/analysis_repo.py b/document-parser/persistence/analysis_repo.py index 0652e5b..9d6a7a7 100644 --- a/document-parser/persistence/analysis_repo.py +++ b/document-parser/persistence/analysis_repo.py @@ -2,7 +2,7 @@ from __future__ import annotations -from datetime import datetime +from datetime import UTC, datetime from domain.models import AnalysisJob, AnalysisStatus from persistence.database import get_connection @@ -29,7 +29,7 @@ def _row_to_job(row) -> AnalysisJob: error_message=row["error_message"], started_at=_parse_dt(row["started_at"]), completed_at=_parse_dt(row["completed_at"]), - created_at=_parse_dt(row["created_at"]) or datetime.now(), + created_at=_parse_dt(row["created_at"]) or datetime.now(UTC), document_filename=row["filename"] if "filename" in keys else None, ) diff --git a/document-parser/persistence/document_repo.py b/document-parser/persistence/document_repo.py index 623e36a..04ac473 100644 --- a/document-parser/persistence/document_repo.py +++ b/document-parser/persistence/document_repo.py @@ -2,7 +2,7 @@ from __future__ import annotations -from datetime import datetime +from datetime import UTC, datetime from domain.models import Document from persistence.database import get_connection @@ -12,6 +12,8 @@ def _row_to_document(row) -> Document: created = row["created_at"] if isinstance(created, str): created = datetime.fromisoformat(created) + if created.tzinfo is None: + created = created.replace(tzinfo=UTC) return Document( id=row["id"], filename=row["filename"],