agent-tracegraph 0.2.0__py3-none-any.whl

tracegraph/adapters/__init__.py

@@ -0,0 +1,13 @@
+"""Ingestion adapters: external trace source -> RawTrace."""
+
+from tracegraph.adapters.base import TraceAdapter
+from tracegraph.adapters.langgraph_checkpoint import LangGraphCheckpointAdapter
+from tracegraph.adapters.otlp_spans import OTLPSpanAdapter
+from tracegraph.adapters.phoenix_export import PhoenixExportAdapter
+
+__all__ = [
+    "TraceAdapter",
+    "LangGraphCheckpointAdapter",
+    "OTLPSpanAdapter",
+    "PhoenixExportAdapter",
+]

tracegraph/adapters/base.py

@@ -0,0 +1,22 @@
+"""The adapter interface: turn an external trace source into a :class:`RawTrace`.
+
+An adapter never computes the derived tree — it only reconstructs the raw causal graph
+(steps + ``CAUSED_BY`` edges). ``normalize`` does the rest.
+"""
+
+from __future__ import annotations
+
+from typing import Protocol, runtime_checkable
+
+from tracegraph.model import RawTrace
+
+
+@runtime_checkable
+class TraceAdapter(Protocol):
+    def discover(self) -> list[str]:
+        """Best-effort list of available trace ids (may be unsupported by a source)."""
+        ...
+
+    def ingest(self, trace_id: str) -> RawTrace:
+        """Reconstruct one trace's raw causal graph. Raises ``KeyError`` if not found."""
+        ...