glaip-sdk 0.0.19__py3-none-any.whl → 0.0.20__py3-none-any.whl

glaip_sdk/cli/transcript/export.py

@@ -0,0 +1,38 @@
+"""Shared helpers for transcript export workflows.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Any
+
+from glaip_sdk.cli.transcript.cache import (
+    latest_manifest_entry,
+    resolve_manifest_entry,
+)
+
+
+def resolve_manifest_for_export(ctx: Any, run_id: str | None) -> dict[str, Any] | None:
+    """Resolve a manifest entry for export based on run id or recent context."""
+    if run_id:
+        return resolve_manifest_entry(run_id)
+
+    ctx_obj = ctx if isinstance(ctx, dict) else getattr(ctx, "obj", None)
+    if isinstance(ctx_obj, dict):
+        candidate = ctx_obj.get("_last_transcript_manifest")
+        if isinstance(candidate, dict):
+            return candidate
+
+    return latest_manifest_entry()
+
+
+def normalise_export_destination(path: Path) -> Path:
+    """Return an absolute path for the export destination."""
+    expanded = path.expanduser()
+    return expanded if expanded.is_absolute() else Path.cwd() / expanded
+
+
+__all__ = ["resolve_manifest_for_export", "normalise_export_destination"]

glaip_sdk/cli/transcript/launcher.py

@@ -0,0 +1,79 @@
+"""Utilities for launching the post-run transcript viewer.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+import sys
+from pathlib import Path
+from typing import Any
+
+from rich.console import Console
+
+from glaip_sdk.cli.context import get_ctx_value
+from glaip_sdk.cli.transcript.cache import (
+    export_transcript as export_cached_transcript,
+)
+from glaip_sdk.cli.transcript.capture import StoredTranscriptContext
+from glaip_sdk.cli.transcript.viewer import ViewerContext, run_viewer_session
+
+
+def should_launch_post_run_viewer(
+    ctx: Any, console: Console, *, slash_mode: bool
+) -> bool:
+    """Return True if the viewer should open automatically."""
+    if slash_mode:
+        return False
+    ctx_obj = getattr(ctx, "obj", None)
+    if isinstance(ctx_obj, dict) and ctx_obj.get("_slash_session"):
+        return False
+    if get_ctx_value(ctx, "view", "rich") != "rich":
+        return False
+    if not bool(get_ctx_value(ctx, "tty", True)):
+        return False
+    if not console.is_terminal:
+        return False
+    try:
+        if not sys.stdin.isatty():
+            return False
+    except Exception:
+        return False
+    return True
+
+
+def maybe_launch_post_run_viewer(
+    ctx: Any,
+    transcript_context: StoredTranscriptContext | None,
+    *,
+    console: Console,
+    slash_mode: bool,
+) -> None:
+    """Launch the post-run viewer when context and settings allow it."""
+    if transcript_context is None:
+        return
+    if not should_launch_post_run_viewer(ctx, console, slash_mode=slash_mode):
+        return
+
+    manifest_entry = transcript_context.store_result.manifest_entry
+    run_id = manifest_entry.get("run_id")
+    if not run_id:
+        return
+
+    viewer_ctx = ViewerContext(
+        manifest_entry=manifest_entry,
+        events=transcript_context.payload.events,
+        default_output=transcript_context.payload.default_output,
+        final_output=transcript_context.payload.final_output,
+        stream_started_at=None,
+        meta=transcript_context.payload.meta,
+    )
+
+    def _export(destination: Path) -> Path:
+        return export_cached_transcript(destination=destination, run_id=run_id)
+
+    run_viewer_session(console, viewer_ctx, _export)
+
+
+__all__ = ["should_launch_post_run_viewer", "maybe_launch_post_run_viewer"]