hermes-agent/tests/telemetry/test_insights_integration.py
emozilla ccfa079252 feat(telemetry): local-first telemetry & observability
Add a built-in telemetry system that records what the agent does — workflows,
model calls, tool calls, errors — to the local machine, powers `/insights`, and
can export to an operator-chosen destination. Default-on locally; nothing leaves
the machine unless the user exports it or opts into the aggregate plane.

Three planes with a hard wall between them:
  - local: full-fidelity observability (real model/provider/tool names), on by
    default, never leaves the machine.
  - aggregate: opt-in metadata, default off. No uploader ships — consent is
    recorded via telemetry.consent_state, and `preview` shows what would be
    produced, computed locally.
  - trajectories: full message content, opt-in, exported only to the operator's
    own destination.

Mechanism:
  - Bundled `telemetry` plugin registers observational lifecycle hooks
    (on_session_start / post_api_request / post_tool_call / on_session_finalize).
    No core call sites are edited; hooks already carry the data.
  - Fire-and-forget emitter: emit() returns in microseconds, never blocks or
    raises into a model/tool call. A daemon thread writes events to an
    append-only JSONL log and the tel_* tables in state.db (its own sqlite
    connection, separate from SessionDB).
  - tel_runs / tel_model_calls / tel_tool_calls live in the declarative
    SCHEMA_SQL and are reconciled automatically; SCHEMA_VERSION 16 -> 17.
  - metrics derives rollups for /usage and /insights; rollup builds per-run
    summaries for `hermes telemetry preview`.

Consent is config, not a parallel command surface. The config file is the root
of trust: set telemetry.consent_state with `hermes config set`, or pin any
telemetry.* key (including allow_aggregate) via managed scope, which overrides
the user's value per key. `hermes telemetry` exposes only what config cannot:
status (report), preview (query), and export.

Export:
  - exporter_bulk writes telemetry (and, when the trajectories plane is enabled,
    session content) to ndjson/json.
  - otlp_exporter streams spans to a configured OpenTelemetry Collector over
    OTLP/HTTP. The SDK is an optional extra (hermes-agent[otlp]), lazily
    installed via tools.lazy_deps on first use.
  - Secrets are always redacted on every export path
    (redact_sensitive_text(force=True)); content export is gated by the
    trajectories plane, and PII scrubbing follows telemetry.content_redaction.
    OTLP auth headers reference environment variable names, never inline values.

No outbound emission to Nous. The aggregate uploader is intentionally not built.
2026-06-24 02:14:02 -04:00

82 lines
3.1 KiB
Python

"""Insights ↔ telemetry integration: the observability section in /insights output."""
from __future__ import annotations
import time
import pytest
from hermes_state import SessionDB
from agent.insights import InsightsEngine
from agent.telemetry.emitter import TelemetryEmitter
from agent.telemetry.events import ModelCallEvent, RunEvent, ToolCallEvent
@pytest.fixture()
def db(tmp_path):
session_db = SessionDB(db_path=tmp_path / "ins_tel.db")
yield session_db
session_db.close()
def _seed_telemetry(db_path):
em = TelemetryEmitter(events_path=db_path.parent / "tel" / "events.jsonl", db_path=db_path)
now = time.time_ns()
em.emit(RunEvent(run_id="r1", trace_id="t1", entrypoint="gateway",
platform="telegram", end_reason="completed",
start_ns=now - 90_000_000, end_ns=now))
em.emit(RunEvent(run_id="r2", trace_id="t2", entrypoint="cli",
end_reason="failed", start_ns=now - 11_000_000, end_ns=now))
em.emit(ModelCallEvent(span_id="m1", run_id="r1", provider="anthropic",
model="claude-opus-4", input_tokens=5000, output_tokens=800,
cache_read_tokens=1000, latency_ms=2200))
em.emit(ToolCallEvent(span_id="tc1", run_id="r1", tool_name="web_search", result_class="ok"))
em.emit(ToolCallEvent(span_id="tc2", run_id="r1", tool_name="browser_navigate", result_class="error"))
em.flush()
em.close()
def test_report_includes_telemetry_when_present(db):
# A session so generate() isn't the empty branch
db.create_session(session_id="s1", source="cli", model="anthropic/claude-sonnet-4")
_seed_telemetry(db.db_path)
engine = InsightsEngine(db)
report = engine.generate(days=30)
tel = report.get("telemetry")
assert tel, "telemetry section missing"
assert tel["workflows"]["total_runs"] == 2
assert tel["workflows"]["success_rate"] == 0.5
assert tel["tool_calls"]["total"] == 2
assert tel["tool_calls"]["failure_rate"] == 0.5
assert tel["model_calls"]["by_provider"]["anthropic"] == 1
def test_terminal_output_renders_observability_section(db):
db.create_session(session_id="s1", source="cli", model="anthropic/claude-sonnet-4")
_seed_telemetry(db.db_path)
engine = InsightsEngine(db)
out = engine.format_terminal(engine.generate(days=30))
assert "Observability" in out
assert "Workflows:" in out
assert "Failure rate:" in out
assert "Providers:" in out
def test_telemetry_section_absent_when_no_tel_rows(db):
# Session present, but no telemetry events seeded.
db.create_session(session_id="s1", source="cli", model="anthropic/claude-sonnet-4")
engine = InsightsEngine(db)
report = engine.generate(days=30)
assert report.get("telemetry") == {}
out = engine.format_terminal(report)
assert "Observability" not in out
def test_empty_report_has_telemetry_key(db):
# No sessions at all -> empty branch still carries the key (renderer-safe).
engine = InsightsEngine(db)
report = engine.generate(days=30)
assert report.get("empty") is True
assert report.get("telemetry") == {}