Add Agent Event Bus observability bridge

2026-07-09 10:16:23 +08:00
parent 665c33f69a
commit 6fc538bf50
1 changed files with 37 additions and 0 deletions
@@ -0,0 +1,37 @@
+"""Bridge Agent Event Bus events into Agent Observability metrics."""
+
+from .agent_observability_hooks import AgentObservabilityHooks
+
+
+class AgentEventObservabilityBridge:
+    def __init__(self, hooks: AgentObservabilityHooks | None = None):
+        self.hooks = hooks or AgentObservabilityHooks()
+
+    def consume(self, event: dict):
+        kind = event.get("kind")
+        payload = event.get("payload", {})
+        if kind == "agent_started":
+            return self.hooks.start(
+                user_id=payload.get("user_id", "unknown"),
+                visible_scope=payload.get("scope", "self"),
+                entry_agent=payload.get("agent", "unknown"),
+                execution_id=event.get("id"),
+            )
+        if kind == "agent_delegated":
+            return self.hooks.delegate(event["execution_id"], payload.get("agent", "unknown"))
+        if kind == "token_consumed":
+            return self.hooks.tokens(
+                event["execution_id"],
+                payload.get("input_tokens", 0),
+                payload.get("output_tokens", 0),
+            )
+        if kind in {"agent_completed", "agent_failed"}:
+            return self.hooks.finish(
+                event["execution_id"],
+                payload.get("closed_by"),
+                "failed" if kind == "agent_failed" else "completed",
+            )
+        return {"status": "ignored", "kind": kind}
+
+
+__all__ = ["AgentEventObservabilityBridge"]