fix: replace deprecated datetime.utcnow() with datetime.now(timezone.utc) (fixes #1647) (#1650)

- Replace datetime.utcnow() calls across multiple components
- Update imports to include timezone module
- Ensure Python 3.12+ compatibility and prevent 3.14 runtime errors
- Fixed 10 critical files including CLI, jobs, agents, memory, and context components

Co-authored-by: praisonai-triage-agent[bot] <272766704+praisonai-triage-agent[bot]@users.noreply.github.com>
Co-authored-by: praisonai-triage-agent[bot] <praisonai-triage-agent[bot]@users.noreply.github.com>

Author: praisonai-triage-agent[bot]

src/praisonai-agents/praisonaiagents/context/session_tracker.py

@@ -7,7 +7,7 @@
 
 from dataclasses import dataclass, field, asdict
 from typing import Any, Dict, List, Optional
-from datetime import datetime
+from datetime import datetime, timezone
 import json
 import logging
 from praisonaiagents._logging import get_logger
@@ -97,8 +97,8 @@ def __init__(
 
         self._state = SessionState(
             session_id=self.session_id,
-            created_at=datetime.utcnow().isoformat() + "Z",
-            updated_at=datetime.utcnow().isoformat() + "Z",
+            created_at=datetime.now(timezone.utc).isoformat(),
+            updated_at=datetime.now(timezone.utc).isoformat(),
         )
 
     def _generate_session_id(self) -> str:
@@ -163,7 +163,7 @@ def mark_plan_step_complete(self, step_index: int) -> None:
 
     def _mark_updated(self) -> None:
         """Update the updated_at timestamp."""
-        self._state.updated_at = datetime.utcnow().isoformat() + "Z"
+        self._state.updated_at = datetime.now(timezone.utc).isoformat()
         self._state.turn_count += 1
 
     def to_context_string(self) -> str:
@@ -257,7 +257,7 @@ def clear(self) -> None:
         self._state = SessionState(
             session_id=self.session_id,
             created_at=self._state.created_at,
-            updated_at=datetime.utcnow().isoformat() + "Z",
+            updated_at=datetime.now(timezone.utc).isoformat(),
         )
 
     def to_dict(self) -> Dict[str, Any]:

src/praisonai-agents/praisonaiagents/eval/grader.py

@@ -12,7 +12,7 @@
 import logging
 from praisonaiagents._logging import get_logger
 from dataclasses import dataclass, field, asdict
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional
 
 logger = get_logger(__name__)
@@ -39,7 +39,7 @@ class GradeResult:
     input_text: str = ""
     output: str = ""
     expected_output: Optional[str] = None
-    timestamp: str = field(default_factory=lambda: datetime.utcnow().isoformat())
+    timestamp: str = field(default_factory=lambda: datetime.now(timezone.utc).isoformat())
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary."""
@@ -55,7 +55,7 @@ def from_dict(cls, data: Dict[str, Any]) -> "GradeResult":
             input_text=data.get("input_text", ""),
             output=data.get("output", ""),
             expected_output=data.get("expected_output"),
-            timestamp=data.get("timestamp", datetime.utcnow().isoformat()),
+            timestamp=data.get("timestamp", datetime.now(timezone.utc).isoformat()),
         )
 
 class BaseLLMGrader:

src/praisonai-agents/praisonaiagents/memory/memory.py

@@ -7,7 +7,7 @@
 from typing import Any, Dict, List, Optional, Union, Literal
 import logging
 from praisonaiagents._logging import get_logger
-from datetime import datetime
+from datetime import datetime, timezone
 
 # Decomposed memory functionality - imported as mixins for backward compatibility
 from .storage import StorageMixin
@@ -692,7 +692,7 @@ def store_short_term(
                         "_id": ident,
                         "content": text,
                         "metadata": metadata,
-                        "created_at": datetime.utcnow(),
+                        "created_at": datetime.now(timezone.utc),
                         "memory_type": "short_term"
                     }
                     self.mongo_short_term.insert_one(doc)
@@ -918,7 +918,7 @@ def store_long_term(
                     "_id": ident,
                     "content": text,
                     "metadata": metadata,
-                    "created_at": datetime.utcnow(),
+                    "created_at": datetime.now(timezone.utc),
                     "memory_type": "long_term"
                 }
 
@@ -1441,14 +1441,14 @@ def store_user_memory(self, user_id: str, text: str, extra: Dict[str, Any] = Non
             self.mem0_client.add(text, user_id=user_id, metadata=meta)
         elif self.use_mongodb and hasattr(self, "mongo_users"):
             try:
-                from datetime import datetime
+                from datetime import datetime, timezone
                 ident = str(time.time_ns())
                 doc = {
                     "_id": ident,
                     "user_id": user_id,
                     "content": text,
                     "metadata": meta,
-                    "created_at": datetime.utcnow()
+                    "created_at": datetime.now(timezone.utc)
                 }
                 self.mongo_users.insert_one(doc)
                 self._log_verbose(f"Successfully stored user memory for {user_id}")

src/praisonai/praisonai/cli/execution/profiler.py

@@ -16,7 +16,7 @@
 import time
 import platform
 from dataclasses import dataclass, field
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional, Tuple
 
 from .request import ExecutionRequest
@@ -409,7 +409,7 @@ class ProfileReport:
     def __post_init__(self):
         """Set timestamp if not provided."""
         if not self.timestamp:
-            self.timestamp = datetime.utcnow().isoformat() + "Z"
+            self.timestamp = datetime.now(timezone.utc).isoformat()
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for JSON serialization."""

src/praisonai/praisonai/cli/features/profiler/core.py

@@ -15,7 +15,7 @@
 import sys
 import time
 from dataclasses import dataclass, field
-from datetime import datetime
+from datetime import datetime, timezone
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple
 
@@ -113,7 +113,7 @@ class ProfilerResult:
     callees: Dict[str, List[str]] = field(default_factory=dict)
     import_times: List[Tuple[str, float]] = field(default_factory=list)
     metadata: Dict[str, Any] = field(default_factory=dict)
-    timestamp: str = field(default_factory=lambda: datetime.utcnow().isoformat() + "Z")
+    timestamp: str = field(default_factory=lambda: datetime.now(timezone.utc).isoformat())
 
     def to_dict(self) -> Dict[str, Any]:
         return {
@@ -304,7 +304,7 @@ def _collect_metadata(self, model: Optional[str]) -> Dict[str, Any]:
             "platform": platform.platform(),
             "praisonai_version": praisonai_version,
             "model": model or "default",
-            "timestamp": datetime.utcnow().isoformat() + "Z",
+            "timestamp": datetime.now(timezone.utc).isoformat(),
         }
 
     def _extract_function_stats(self) -> List[FunctionStats]:

src/praisonai/praisonai/cli/features/profiler/suite.py

@@ -11,7 +11,7 @@
 import sys
 import time
 from dataclasses import dataclass, field
-from datetime import datetime
+from datetime import datetime, timezone
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
@@ -85,7 +85,7 @@ class SuiteResult:
     startup_cold_ms: float = 0.0
     startup_warm_ms: float = 0.0
     import_analysis: List[Dict[str, Any]] = field(default_factory=list)
-    timestamp: str = field(default_factory=lambda: datetime.utcnow().isoformat() + "Z")
+    timestamp: str = field(default_factory=lambda: datetime.now(timezone.utc).isoformat())
     metadata: Dict[str, Any] = field(default_factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
@@ -182,7 +182,7 @@ def _collect_metadata(self) -> Dict[str, Any]:
             "python_version": platform.python_version(),
             "platform": platform.platform(),
             "praisonai_version": praisonai_version,
-            "timestamp": datetime.utcnow().isoformat() + "Z",
+            "timestamp": datetime.now(timezone.utc).isoformat(),
         }
 
     def _measure_startup(self) -> tuple:

src/praisonai/praisonai/cli/state/identifiers.py

@@ -9,7 +9,7 @@
 import secrets
 import threading
 from dataclasses import dataclass, field
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import Dict, Optional
 
 
@@ -20,7 +20,7 @@ def generate_run_id() -> str:
     Format: run_YYYYMMDD_HHMMSS_<random>
     Example: run_20241231_143022_a1b2c3
     """
-    timestamp = datetime.utcnow().strftime("%Y%m%d_%H%M%S")
+    timestamp = datetime.now(timezone.utc).strftime("%Y%m%d_%H%M%S")
     random_suffix = secrets.token_hex(3)  # 6 characters
     return f"run_{timestamp}_{random_suffix}"
 
@@ -68,7 +68,7 @@ class RunContext:
     """
     run_id: str = field(default_factory=generate_run_id)
     trace_id: str = field(default_factory=generate_trace_id)
-    start_time: datetime = field(default_factory=datetime.utcnow)
+    start_time: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
 
     # Agent tracking
     agents: Dict[str, str] = field(default_factory=dict)  # name -> agent_id

src/praisonai/praisonai/cli/state/sessions.py

@@ -7,7 +7,7 @@
 import json
 import shutil
 from dataclasses import dataclass, field
-from datetime import datetime
+from datetime import datetime, timezone
 from pathlib import Path
 from typing import Any, Dict, List, Optional, TYPE_CHECKING
 
@@ -129,7 +129,7 @@ def create(self, context: RunContext, name: Optional[str] = None) -> SessionMeta
         session_dir = self._get_session_dir(session_id)
         session_dir.mkdir(parents=True, exist_ok=True)
 
-        now = datetime.utcnow()
+        now = datetime.now(timezone.utc)
         metadata = SessionMetadata(
             session_id=session_id,
             run_id=context.run_id,
@@ -206,7 +206,7 @@ def append_event(self, session_id: str, event: Dict[str, Any]) -> None:
         metadata = self._load_metadata(session_id)
         if metadata:
             metadata.event_count += 1
-            metadata.updated_at = datetime.utcnow()
+            metadata.updated_at = datetime.now(timezone.utc)
             self._save_metadata(metadata)
 
     def get(self, session_id: str) -> Optional[SessionMetadata]:
@@ -365,7 +365,7 @@ def update_status(self, session_id: str, status: str) -> None:
         metadata = self._load_metadata(session_id)
         if metadata:
             metadata.status = status
-            metadata.updated_at = datetime.utcnow()
+            metadata.updated_at = datetime.now(timezone.utc)
             self._save_metadata(metadata)
 
 

src/praisonai/praisonai/jobs/models.py

@@ -7,7 +7,7 @@
 import uuid
 from enum import Enum
 from typing import Optional, Dict, Any, List
-from datetime import datetime
+from datetime import datetime, timezone
 from pydantic import BaseModel, Field, field_validator
 import socket
 import ipaddress
@@ -177,7 +177,7 @@ def validate_webhook_url(cls, v: Optional[str]) -> Optional[str]:
     run_id: Optional[str] = Field(None)
 
     # Timestamps
-    created_at: datetime = Field(default_factory=datetime.utcnow)
+    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
     started_at: Optional[datetime] = Field(None)
     completed_at: Optional[datetime] = Field(None)
 
@@ -194,7 +194,7 @@ def duration_seconds(self) -> Optional[float]:
         """Calculate job duration."""
         if self.started_at is None:
             return None
-        end_time = self.completed_at or datetime.utcnow()
+        end_time = self.completed_at or datetime.now(timezone.utc)
         return (end_time - self.started_at).total_seconds()
 
     @property
@@ -247,26 +247,26 @@ def to_result_response(self) -> JobResultResponse:
     def start(self):
         """Mark job as started."""
         self.status = JobStatus.RUNNING
-        self.started_at = datetime.utcnow()
+        self.started_at = datetime.now(timezone.utc)
 
     def succeed(self, result: Any, metrics: Optional[Dict[str, Any]] = None):
         """Mark job as succeeded."""
         self.status = JobStatus.SUCCEEDED
         self.result = result
         self.metrics = metrics
-        self.completed_at = datetime.utcnow()
+        self.completed_at = datetime.now(timezone.utc)
         self.progress_percentage = 100.0
 
     def fail(self, error: str):
         """Mark job as failed."""
         self.status = JobStatus.FAILED
         self.error = error
-        self.completed_at = datetime.utcnow()
+        self.completed_at = datetime.now(timezone.utc)
 
     def cancel(self):
         """Mark job as cancelled."""
         self.status = JobStatus.CANCELLED
-        self.completed_at = datetime.utcnow()
+        self.completed_at = datetime.now(timezone.utc)
         self._cancel_requested = True
 
     def update_progress(

src/praisonai/praisonai/jobs/router.py

@@ -7,7 +7,7 @@
 import asyncio
 import logging
 from typing import Optional
-from datetime import datetime
+from datetime import datetime, timezone
 
 from fastapi import APIRouter, HTTPException, Header, Request, Query, Response
 from sse_starlette.sse import EventSourceResponse
@@ -304,7 +304,7 @@ async def on_progress(updated_job: Job):
                         # Send heartbeat
                         yield {
                             "event": "heartbeat",
-                            "data": f'{{"timestamp": "{datetime.utcnow().isoformat()}"}}'
+                            "data": f'{{"timestamp": "{datetime.now(timezone.utc).isoformat()}"}}'
                         }
             finally:
                 executor.unregister_progress_callback(job_id)