chore: import upstream snapshot with attribution

surfaces/interactive_shell/utils/telemetry/sinks/__init__.py

@@ -0,0 +1,8 @@
+"""Prompt logging sinks."""
+
+from surfaces.interactive_shell.utils.telemetry.sinks.local_jsonl import (
+    append_prompt_log_record,
+)
+from surfaces.interactive_shell.utils.telemetry.sinks.posthog_ai import capture_ai_generation
+
+__all__ = ["append_prompt_log_record", "capture_ai_generation"]

surfaces/interactive_shell/utils/telemetry/sinks/local_jsonl.py

@@ -0,0 +1,32 @@
+"""Local JSONL sink for prompt/response records."""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+from typing import Any
+
+_DEFAULT_MAX_BYTES = 50 * 1024 * 1024
+
+
+def append_prompt_log_record(
+    *,
+    path: Path,
+    record: dict[str, Any],
+    max_bytes: int = _DEFAULT_MAX_BYTES,
+) -> None:
+    path.parent.mkdir(parents=True, exist_ok=True)
+    _rotate_if_needed(path, max_bytes=max_bytes)
+    with path.open("a", encoding="utf-8") as fh:
+        fh.write(json.dumps(record, ensure_ascii=False) + "\n")
+
+
+def _rotate_if_needed(path: Path, *, max_bytes: int) -> None:
+    if max_bytes <= 0 or not path.exists():
+        return
+    if path.stat().st_size <= max_bytes:
+        return
+    backup = path.with_name(path.name + ".1")
+    if backup.exists():
+        backup.unlink()
+    path.rename(backup)

surfaces/interactive_shell/utils/telemetry/sinks/posthog_ai.py

@@ -0,0 +1,10 @@
+"""PostHog sink for `$ai_generation` events."""
+
+from __future__ import annotations
+
+from platform.analytics.events import Event
+from platform.analytics.provider import JsonValue, get_analytics
+
+
+def capture_ai_generation(properties: dict[str, JsonValue]) -> None:
+    get_analytics().capture(Event.AI_GENERATION, properties)