flock-core 0.5.7__py3-none-any.whl → 0.5.9__py3-none-any.whl

flock/dashboard/collector.py

@@ -11,7 +11,7 @@ import json
 import traceback
 from collections import defaultdict, deque
 from dataclasses import dataclass, field
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from typing import TYPE_CHECKING, Any, Optional
 
 from pydantic import PrivateAttr
@@ -159,7 +159,7 @@ class DashboardEventCollector(AgentComponent):
             Unmodified inputs (pass-through)
         """
         # Record start time for duration calculation
-        self._run_start_times[ctx.task_id] = datetime.now(timezone.utc).timestamp()
+        self._run_start_times[ctx.task_id] = datetime.now(UTC).timestamp()
 
         # Extract consumed types and artifact IDs
         consumed_types = list({artifact.type for artifact in inputs})
@@ -280,7 +280,7 @@ class DashboardEventCollector(AgentComponent):
         # Calculate duration
         start_time = self._run_start_times.get(ctx.task_id)
         if start_time:
-            duration_ms = (datetime.now(timezone.utc).timestamp() - start_time) * 1000
+            duration_ms = (datetime.now(UTC).timestamp() - start_time) * 1000
             del self._run_start_times[ctx.task_id]
         else:
             duration_ms = 0.0
@@ -319,7 +319,7 @@ class DashboardEventCollector(AgentComponent):
             run.status = "completed"
             run.duration_ms = duration_ms
             run.metrics = dict(metrics)
-            run.completed_at = datetime.now(timezone.utc)
+            run.completed_at = datetime.now(UTC)
             for artifact_id in artifacts_produced:
                 if artifact_id not in run.produced_artifacts:
                     run.produced_artifacts.append(artifact_id)
@@ -345,7 +345,7 @@ class DashboardEventCollector(AgentComponent):
         error_traceback = "".join(
             traceback.format_exception(type(error), error, error.__traceback__)
         )
-        failed_at = datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        failed_at = datetime.now(UTC).isoformat().replace("+00:00", "Z")
 
         # Clean up start time tracking
         if ctx.task_id in self._run_start_times:
@@ -374,7 +374,7 @@ class DashboardEventCollector(AgentComponent):
             )
             run.status = "error"
             run.error_message = error_message
-            run.completed_at = datetime.now(timezone.utc)
+            run.completed_at = datetime.now(UTC)
             self._agent_status[agent.name] = "error"
             await self._update_agent_snapshot_locked(agent)
 
@@ -442,7 +442,7 @@ class DashboardEventCollector(AgentComponent):
                 run_id=run_id,
                 agent_name=agent_name,
                 correlation_id=correlation_id,
-                started_at=datetime.now(timezone.utc) if ensure_started else None,
+                started_at=datetime.now(UTC) if ensure_started else None,
             )
             self._run_registry[run_id] = run
         else:
@@ -450,11 +450,11 @@ class DashboardEventCollector(AgentComponent):
             if correlation_id:
                 run.correlation_id = correlation_id
             if ensure_started and run.started_at is None:
-                run.started_at = datetime.now(timezone.utc)
+                run.started_at = datetime.now(UTC)
         return run
 
     async def _update_agent_snapshot_locked(self, agent: "Agent") -> None:
-        now = datetime.now(timezone.utc)
+        now = datetime.now(UTC)
         description = agent.description or ""
         subscriptions = sorted(
             {

flock/dashboard/events.py

@@ -4,7 +4,7 @@ Defines 5 core event types that capture agent execution lifecycle.
 All schemas match DATA_MODEL.md specification exactly.
 """
 
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -40,7 +40,7 @@ class AgentActivatedEvent(BaseModel):
     # Event metadata
     correlation_id: str
     timestamp: str = Field(
-        default_factory=lambda: datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        default_factory=lambda: datetime.now(UTC).isoformat().replace("+00:00", "Z")
     )
 
     # Agent identification
@@ -74,7 +74,7 @@ class MessagePublishedEvent(BaseModel):
     # Event metadata
     correlation_id: str
     timestamp: str = Field(
-        default_factory=lambda: datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        default_factory=lambda: datetime.now(UTC).isoformat().replace("+00:00", "Z")
     )
 
     # Artifact identification
@@ -109,7 +109,7 @@ class StreamingOutputEvent(BaseModel):
     # Event metadata
     correlation_id: str
     timestamp: str = Field(
-        default_factory=lambda: datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        default_factory=lambda: datetime.now(UTC).isoformat().replace("+00:00", "Z")
     )
 
     # Agent identification
@@ -137,7 +137,7 @@ class AgentCompletedEvent(BaseModel):
     # Event metadata
     correlation_id: str
     timestamp: str = Field(
-        default_factory=lambda: datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        default_factory=lambda: datetime.now(UTC).isoformat().replace("+00:00", "Z")
     )
 
     # Agent identification
@@ -163,7 +163,7 @@ class AgentErrorEvent(BaseModel):
     # Event metadata
     correlation_id: str
     timestamp: str = Field(
-        default_factory=lambda: datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        default_factory=lambda: datetime.now(UTC).isoformat().replace("+00:00", "Z")
     )
 
     # Agent identification
@@ -187,7 +187,7 @@ class CorrelationGroupUpdatedEvent(BaseModel):
 
     # Event metadata
     timestamp: str = Field(
-        default_factory=lambda: datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        default_factory=lambda: datetime.now(UTC).isoformat().replace("+00:00", "Z")
     )
 
     # Agent identification
@@ -224,7 +224,7 @@ class BatchItemAddedEvent(BaseModel):
 
     # Event metadata
     timestamp: str = Field(
-        default_factory=lambda: datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+        default_factory=lambda: datetime.now(UTC).isoformat().replace("+00:00", "Z")
     )
 
     # Agent identification

flock/dashboard/service.py

@@ -8,7 +8,7 @@ Provides real-time dashboard capabilities by:
 """
 
 import os
-from datetime import datetime, timedelta, timezone
+from datetime import UTC, datetime, timedelta
 from importlib.metadata import PackageNotFoundError, version
 from pathlib import Path
 from typing import Any
@@ -716,10 +716,10 @@ class DashboardHTTPService(BlackboardHTTPService):
                     if time_range and time_range[0]:
                         # Convert nanoseconds to datetime
                         oldest_trace = datetime.fromtimestamp(
-                            time_range[0] / 1_000_000_000, tz=timezone.utc
+                            time_range[0] / 1_000_000_000, tz=UTC
                         ).isoformat()
                         newest_trace = datetime.fromtimestamp(
-                            time_range[1] / 1_000_000_000, tz=timezone.utc
+                            time_range[1] / 1_000_000_000, tz=UTC
                         ).isoformat()
 
                 # Get file size
@@ -1051,7 +1051,7 @@ def _get_correlation_groups(
     if not groups:
         return []
 
-    now = datetime.now(timezone.utc)
+    now = datetime.now(UTC)
     result = []
 
     for corr_key, group in groups.items():
@@ -1059,7 +1059,7 @@ def _get_correlation_groups(
         if group.created_at_time:
             created_at_time = group.created_at_time
             if created_at_time.tzinfo is None:
-                created_at_time = created_at_time.replace(tzinfo=timezone.utc)
+                created_at_time = created_at_time.replace(tzinfo=UTC)
             elapsed = (now - created_at_time).total_seconds()
         else:
             elapsed = 0
@@ -1146,11 +1146,11 @@ def _get_batch_state(
     if not accumulator or not accumulator.artifacts:
         return None
 
-    now = datetime.now(timezone.utc)
+    now = datetime.now(UTC)
     # Ensure accumulator.created_at is timezone-aware
     created_at = accumulator.created_at
     if created_at.tzinfo is None:
-        created_at = created_at.replace(tzinfo=timezone.utc)
+        created_at = created_at.replace(tzinfo=UTC)
     elapsed = (now - created_at).total_seconds()
 
     # Calculate items collected (needed for all batch types)