Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
"""LangSmith 式 trace：manifests/trace_log.jsonl"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import uuid
|
||
from contextlib import contextmanager
|
||
from datetime import datetime, timezone
|
||
from typing import Any, Iterator
|
||
|
||
from as_platform.config import TRACE_LOG, MANIFESTS
|
||
|
||
_current_trace: str | None = None
|
||
|
||
|
||
def _now() -> str:
|
||
return datetime.now(timezone.utc).isoformat()
|
||
|
||
|
||
def start_trace(name: str, **meta: Any) -> str:
|
||
global _current_trace
|
||
trace_id = f"trace-{uuid.uuid4().hex[:12]}"
|
||
_current_trace = trace_id
|
||
_append({"type": "trace_start", "trace_id": trace_id, "name": name, "ts": _now(), **meta})
|
||
return trace_id
|
||
|
||
|
||
def _append(entry: dict[str, Any]) -> None:
|
||
MANIFESTS.mkdir(parents=True, exist_ok=True)
|
||
if not TRACE_LOG.is_file():
|
||
TRACE_LOG.write_text("", encoding="utf-8")
|
||
with TRACE_LOG.open("a", encoding="utf-8") as f:
|
||
f.write(json.dumps(entry, ensure_ascii=False) + "\n")
|
||
|
||
|
||
@contextmanager
|
||
def trace_span(span_type: str, **fields: Any) -> Iterator[None]:
|
||
span_id = f"span-{uuid.uuid4().hex[:8]}"
|
||
_append({"type": span_type, "span_id": span_id, "trace_id": _current_trace, "ts": _now(), **fields})
|
||
try:
|
||
yield
|
||
finally:
|
||
_append({"type": f"{span_type}_end", "span_id": span_id, "trace_id": _current_trace, "ts": _now()})
|
||
|
||
|
||
def get_trace(trace_id: str) -> list[dict[str, Any]]:
|
||
if not TRACE_LOG.is_file():
|
||
return []
|
||
return [json.loads(l) for l in TRACE_LOG.read_text().strip().splitlines() if l.strip() and trace_id in l]
|
||
|
||
|
||
def list_traces(limit: int = 50) -> list[str]:
|
||
if not TRACE_LOG.is_file():
|
||
return []
|
||
ids = []
|
||
for line in TRACE_LOG.read_text().strip().splitlines():
|
||
try:
|
||
o = json.loads(line)
|
||
if o.get("type") == "trace_start" and o.get("trace_id"):
|
||
ids.append(o["trace_id"])
|
||
except json.JSONDecodeError:
|
||
pass
|
||
return list(reversed(ids[-limit:]))
|