Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Empty file.
23 changes: 23 additions & 0 deletions services/agent/observability/events.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
from dataclasses import dataclass, field
from datetime import datetime, UTC


@dataclass
class LearningEvent:
agent_id: str
event_type: str
payload: dict
created_at: datetime = field(
default_factory=lambda: datetime.now(UTC)
)


class LearningEventBus:
def __init__(self):
self.events = []

def publish(self, event: LearningEvent):
self.events.append(event)

def latest(self):
return self.events[-1] if self.events else None
23 changes: 23 additions & 0 deletions services/agent/observability/metrics.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
from dataclasses import dataclass, field
from datetime import datetime, UTC


@dataclass
class LearningMetric:
agent_id: str
metric: str
value: float
created_at: datetime = field(
default_factory=lambda: datetime.now(UTC)
)


class LearningMetricsStore:
def __init__(self):
self.metrics = []

def record(self, metric: LearningMetric):
self.metrics.append(metric)

def latest(self):
return self.metrics[-1] if self.metrics else None
23 changes: 23 additions & 0 deletions services/agent/observability/tracing.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
from dataclasses import dataclass, field
from datetime import datetime, UTC


@dataclass
class LearningTrace:
agent_id: str
action: str
outcome: str
created_at: datetime = field(
default_factory=lambda: datetime.now(UTC)
)


class LearningTracer:
def __init__(self):
self.traces = []

def trace(self, item: LearningTrace):
self.traces.append(item)

def last(self):
return self.traces[-1] if self.traces else None
54 changes: 54 additions & 0 deletions tests/agent/test_learning_observability.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
from services.agent.observability.metrics import (
LearningMetric,
LearningMetricsStore,
)
from services.agent.observability.events import (
LearningEvent,
LearningEventBus,
)
from services.agent.observability.tracing import (
LearningTrace,
LearningTracer,
)


def test_learning_metrics():
store = LearningMetricsStore()

store.record(
LearningMetric(
agent_id="AEON-001",
metric="accuracy",
value=0.98,
)
)

assert store.latest().value == 0.98


def test_learning_events():
bus = LearningEventBus()

bus.publish(
LearningEvent(
agent_id="AEON-001",
event_type="learning_update",
payload={"score": 0.95},
)
)

assert bus.latest().event_type == "learning_update"


def test_learning_trace():
tracer = LearningTracer()

tracer.trace(
LearningTrace(
agent_id="AEON-001",
action="policy_update",
outcome="approved",
)
)

assert tracer.last().outcome == "approved"
Loading