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

flock/dashboard/collector.py

@@ -54,7 +54,9 @@ class RunRecord:
     error_message: str | None = None
 
     def to_graph_run(self) -> GraphRun:
-        status = self.status if self.status in {"active", "completed", "error"} else "active"
+        status = (
+            self.status if self.status in {"active", "completed", "error"} else "active"
+        )
         return GraphRun(
             run_id=self.run_id,
             agent_name=self.agent_name,
@@ -102,7 +104,10 @@ class DashboardEventCollector(AgentComponent):
 
     # Use PrivateAttr for non-Pydantic fields (AgentComponent extends BaseModel)
     _events: deque[
-        AgentActivatedEvent | MessagePublishedEvent | AgentCompletedEvent | AgentErrorEvent
+        AgentActivatedEvent
+        | MessagePublishedEvent
+        | AgentCompletedEvent
+        | AgentErrorEvent
     ] = PrivateAttr(default=None)
 
     # Track run start times for duration calculation
@@ -114,7 +119,9 @@ class DashboardEventCollector(AgentComponent):
     # Graph assembly helpers
     _graph_lock: asyncio.Lock = PrivateAttr(default_factory=asyncio.Lock)
     _run_registry: dict[str, RunRecord] = PrivateAttr(default_factory=dict)
-    _artifact_consumers: dict[str, set[str]] = PrivateAttr(default_factory=lambda: defaultdict(set))
+    _artifact_consumers: dict[str, set[str]] = PrivateAttr(
+        default_factory=lambda: defaultdict(set)
+    )
     _agent_status: dict[str, str] = PrivateAttr(default_factory=dict)
     _agent_snapshots: dict[str, AgentSnapshot] = PrivateAttr(default_factory=dict)
 
@@ -185,13 +192,15 @@ class DashboardEventCollector(AgentComponent):
             await self._update_agent_snapshot_locked(agent)
 
         # Build subscription info from agent's subscriptions
-        subscription_info = SubscriptionInfo(from_agents=[], channels=[], mode="both")
+        subscription_info = SubscriptionInfo(from_agents=[], tags=[], mode="both")
 
         if agent.subscriptions:
             # Get first subscription's config (agents typically have one)
             sub = agent.subscriptions[0]
-            subscription_info.from_agents = list(sub.from_agents) if sub.from_agents else []
-            subscription_info.channels = list(sub.channels) if sub.channels else []
+            subscription_info.from_agents = (
+                list(sub.from_agents) if sub.from_agents else []
+            )
+            subscription_info.tags = list(sub.tags) if sub.tags else []
             subscription_info.mode = sub.mode
 
         # Create and store event
@@ -210,7 +219,9 @@ class DashboardEventCollector(AgentComponent):
         )
 
         self._events.append(event)
-        logger.info(f"Agent activated: {agent.name} (correlation_id={event.correlation_id})")
+        logger.info(
+            f"Agent activated: {agent.name} (correlation_id={event.correlation_id})"
+        )
 
         # Broadcast via WebSocket if manager is configured
         if self._websocket_manager:
@@ -220,7 +231,9 @@ class DashboardEventCollector(AgentComponent):
 
         return inputs
 
-    async def on_post_publish(self, agent: "Agent", ctx: Context, artifact: "Artifact") -> None:
+    async def on_post_publish(
+        self, agent: "Agent", ctx: Context, artifact: "Artifact"
+    ) -> None:
         """Emit message_published event when artifact is published.
 
         Args:
@@ -391,7 +404,9 @@ class DashboardEventCollector(AgentComponent):
             }
             runs = [record.to_graph_run() for record in self._run_registry.values()]
             agent_status = dict(self._agent_status)
-        return GraphState(consumptions=consumptions, runs=runs, agent_status=agent_status)
+        return GraphState(
+            consumptions=consumptions, runs=runs, agent_status=agent_status
+        )
 
     async def snapshot_agent_registry(self) -> dict[str, AgentSnapshot]:
         """Return a snapshot of all known agents (active and inactive)."""
@@ -456,21 +471,17 @@ class DashboardEventCollector(AgentComponent):
     async def _update_agent_snapshot_locked(self, agent: "Agent") -> None:
         now = datetime.now(UTC)
         description = agent.description or ""
-        subscriptions = sorted(
-            {
-                type_name
-                for subscription in getattr(agent, "subscriptions", [])
-                for type_name in getattr(subscription, "type_names", [])
-            }
-        )
-        output_types = sorted(
-            {
-                output.spec.type_name
-                for output in getattr(agent, "outputs", [])
-                if getattr(output, "spec", None) is not None
-                and getattr(output.spec, "type_name", "")
-            }
-        )
+        subscriptions = sorted({
+            type_name
+            for subscription in getattr(agent, "subscriptions", [])
+            for type_name in getattr(subscription, "type_names", [])
+        })
+        output_types = sorted({
+            output.spec.type_name
+            for output in getattr(agent, "outputs", [])
+            if getattr(output, "spec", None) is not None
+            and getattr(output.spec, "type_name", "")
+        })
         labels = sorted(agent.labels)
 
         signature_payload = {
@@ -519,7 +530,9 @@ class DashboardEventCollector(AgentComponent):
             first_seen=snapshot.first_seen,
             last_seen=snapshot.last_seen,
             signature=snapshot.signature,
-            logic_operations=[dict(op) for op in snapshot.logic_operations],  # Phase 1.2
+            logic_operations=[
+                dict(op) for op in snapshot.logic_operations
+            ],  # Phase 1.2
         )
 
     def _snapshot_to_record(self, snapshot: AgentSnapshot) -> AgentSnapshotRecord:
@@ -545,19 +558,25 @@ class DashboardEventCollector(AgentComponent):
         """
         # Get visibility kind from class name, stripping "Visibility" suffix
         class_name = type(visibility).__name__
-        kind = class_name[: -len("Visibility")] if class_name.endswith("Visibility") else class_name
+        kind = class_name.removesuffix("Visibility")
 
         spec = VisibilitySpec(kind=kind)
 
         # Extract type-specific fields
         if kind == "Private":
-            spec.agents = list(visibility.agents) if hasattr(visibility, "agents") else []
+            spec.agents = (
+                list(visibility.agents) if hasattr(visibility, "agents") else []
+            )
         elif kind == "Labelled":
             spec.required_labels = (
-                list(visibility.required_labels) if hasattr(visibility, "required_labels") else []
+                list(visibility.required_labels)
+                if hasattr(visibility, "required_labels")
+                else []
             )
         elif kind == "Tenant":
-            spec.tenant_id = visibility.tenant_id if hasattr(visibility, "tenant_id") else None
+            spec.tenant_id = (
+                visibility.tenant_id if hasattr(visibility, "tenant_id") else None
+            )
 
         return spec
 

flock/dashboard/events.py

@@ -14,7 +14,7 @@ class SubscriptionInfo(BaseModel):
     """Subscription configuration for an agent."""
 
     from_agents: list[str] = Field(default_factory=list)
-    channels: list[str] = Field(default_factory=list)
+    tags: list[str] = Field(default_factory=list)
     mode: str = "both"  # "both" | "events" | "direct"
 
 
@@ -123,8 +123,12 @@ class StreamingOutputEvent(BaseModel):
     is_final: bool = False  # True when agent completes this output stream
 
     # Artifact tracking (Phase 6: for message streaming preview)
-    artifact_id: str | None = None  # Pre-generated artifact ID for streaming message nodes
-    artifact_type: str | None = None  # Artifact type name (e.g., "__main__.BookOutline")
+    artifact_id: str | None = (
+        None  # Pre-generated artifact ID for streaming message nodes
+    )
+    artifact_type: str | None = (
+        None  # Artifact type name (e.g., "__main__.BookOutline")
+    )
 
 
 class AgentCompletedEvent(BaseModel):
@@ -149,7 +153,9 @@ class AgentCompletedEvent(BaseModel):
     artifacts_produced: list[str] = Field(default_factory=list)  # [artifact_id]
 
     # Metrics and state
-    metrics: dict[str, Any] = Field(default_factory=dict)  # {"tokens_used": 1234, "cost": 0.05}
+    metrics: dict[str, Any] = Field(
+        default_factory=dict
+    )  # {"tokens_used": 1234, "cost": 0.05}
     final_state: dict[str, Any] = Field(default_factory=dict)  # Context.state snapshot
 
 

flock/dashboard/launcher.py

@@ -94,7 +94,9 @@ class DashboardLauncher:
 
     def _start_dev_server(self) -> None:
         """Start npm dev server for hot-reload development."""
-        print(f"[Dashboard] Starting dev server (DASHBOARD_DEV=1) on port {self.port}...")
+        print(
+            f"[Dashboard] Starting dev server (DASHBOARD_DEV=1) on port {self.port}..."
+        )
 
         try:
             self._npm_process = subprocess.Popen(

flock/dashboard/models/graph.py

@@ -104,7 +104,9 @@ class GraphStatistics(BaseModel):
     consumed_by_agent: dict[str, GraphAgentMetrics] = Field(
         default_factory=dict, alias="consumedByAgent"
     )
-    artifact_summary: dict[str, Any] = Field(default_factory=dict, alias="artifactSummary")
+    artifact_summary: dict[str, Any] = Field(
+        default_factory=dict, alias="artifactSummary"
+    )
 
 
 class GraphArtifact(BaseModel):
@@ -128,8 +130,12 @@ class GraphRun(BaseModel):
     agent_name: str = Field(alias="agentName")
     correlation_id: str | None = Field(default=None, alias="correlationId")
     status: Literal["active", "completed", "error"] = "active"
-    consumed_artifacts: list[str] = Field(default_factory=list, alias="consumedArtifacts")
-    produced_artifacts: list[str] = Field(default_factory=list, alias="producedArtifacts")
+    consumed_artifacts: list[str] = Field(
+        default_factory=list, alias="consumedArtifacts"
+    )
+    produced_artifacts: list[str] = Field(
+        default_factory=list, alias="producedArtifacts"
+    )
     duration_ms: float | None = Field(default=None, alias="durationMs")
     started_at: datetime | None = Field(default=None, alias="startedAt")
     completed_at: datetime | None = Field(default=None, alias="completedAt")