Spaces:
Sleeping
Sleeping
File size: 1,198 Bytes
116524e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | """ObservabilityStep — logs pipeline metrics to Opik."""
from __future__ import annotations
import logging
from ..core.context import ACEStepContext
logger = logging.getLogger(__name__)
class ObservabilityStep:
"""Log pipeline metrics to the observability backend.
Optional side-effect step — only requires ``skillbook`` (always present).
Reads other context fields optionally so the same step works in both
ACE and TraceAnalyser pipelines.
"""
requires: frozenset[str] = frozenset({"skillbook"})
provides: frozenset[str] = frozenset()
def __call__(self, ctx: ACEStepContext) -> ACEStepContext:
metrics: dict = {"skill_count": len(ctx.skillbook) if ctx.skillbook else 0}
if ctx.reflections:
metrics["key_insight"] = ctx.reflections[-1].key_insight
metrics["reflections_count"] = len(ctx.reflections)
if ctx.skill_manager_output:
metrics["operations_count"] = len(ctx.skill_manager_output.operations)
if ctx.trace:
metrics["trace_type"] = type(ctx.trace).__name__
logger.info("ObservabilityStep: %s", metrics)
return ctx
|