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

glaip_sdk/cli/transcript/capture.py

@@ -8,8 +8,13 @@ from __future__ import annotations
 
 import json
 from dataclasses import dataclass
+from io import StringIO
 from typing import Any
 
+from rich.console import Console
+
+from glaip_sdk.cli.auth import resolve_api_url_from_context
+from glaip_sdk.cli.context import get_ctx_value
 from glaip_sdk.cli.transcript.cache import (
     TranscriptPayload,
     TranscriptStoreResult,
@@ -18,7 +23,7 @@ from glaip_sdk.cli.transcript.cache import (
 from glaip_sdk.cli.transcript.cache import (
     build_payload as build_transcript_payload,
 )
-from glaip_sdk.utils.rendering.renderer.progress import format_tool_title
+from glaip_sdk.utils.rendering.layout.progress import format_tool_title
 
 
 @dataclass(slots=True)
@@ -65,11 +70,7 @@ def compute_finished_at(renderer: Any) -> float | None:
 
     if started_at is None:
         stream_processor = getattr(renderer, "stream_processor", None)
-        started_at = (
-            getattr(stream_processor, "streaming_started_at", None)
-            if stream_processor is not None
-            else None
-        )
+        started_at = getattr(stream_processor, "streaming_started_at", None) if stream_processor is not None else None
     if started_at is None or duration is None:
         return None
     try:
@@ -78,9 +79,7 @@ def compute_finished_at(renderer: Any) -> float | None:
         return None
 
 
-def extract_server_run_id(
-    meta: dict[str, Any], events: list[dict[str, Any]]
-) -> str | None:
+def extract_server_run_id(meta: dict[str, Any], events: list[dict[str, Any]]) -> str | None:
     """Derive a server-side run identifier from renderer metadata."""
     run_id = meta.get("run_id") or meta.get("id")
     if run_id:
@@ -107,9 +106,7 @@ def _coerce_meta(meta: Any) -> dict[str, Any]:
     return {"value": coerce_result_text(meta)}
 
 
-def register_last_transcript(
-    ctx: Any, payload: TranscriptPayload, store_result: TranscriptStoreResult
-) -> None:
+def register_last_transcript(ctx: Any, payload: TranscriptPayload, store_result: TranscriptStoreResult) -> None:
     """Persist last-run transcript references onto the Click context."""
     ctx_obj = getattr(ctx, "obj", None)
     if not isinstance(ctx_obj, dict):
@@ -119,6 +116,15 @@ def register_last_transcript(
     ctx_obj["_last_transcript_path"] = str(store_result.path)
 
 
+def _resolve_api_url(ctx: Any) -> str | None:
+    """Resolve API URL from context or account store (CLI/palette ignores env creds)."""
+    return resolve_api_url_from_context(
+        ctx,
+        get_api_url=lambda c: get_ctx_value(c, "api_url"),
+        get_account_name=lambda c: get_ctx_value(c, "account_name"),
+    )
+
+
 def _extract_step_summaries(renderer: Any) -> list[dict[str, Any]]:
     """Return lightweight step summaries for the transcript viewer."""
     steps = getattr(renderer, "steps", None)
@@ -172,6 +178,38 @@ def _format_step_display_name(name: str) -> str:
         return name
 
 
+def _extract_step_summary_lines(renderer: Any) -> list[str]:
+    """Render the live steps summary to plain text lines."""
+    if not hasattr(renderer, "_render_steps_text"):
+        return []
+
+    try:
+        renderable = renderer._render_steps_text()
+    except Exception:
+        return []
+
+    buffer = StringIO()
+    console = Console(file=buffer, record=True, force_terminal=False, width=120)
+    try:
+        console.print(renderable)
+    except Exception:
+        return []
+
+    text = console.export_text() or buffer.getvalue()
+    lines = [line.rstrip() for line in text.splitlines()]
+    half = len(lines) // 2
+    if half and lines[:half] == lines[half : half * 2]:
+        return lines[:half]
+    start = 0
+    prefixes = ("🤖", "🔧", "💭", "├", "└", "│", "•")
+    for idx, line in enumerate(lines):
+        if line.lstrip().startswith(prefixes):
+            start = idx
+            break
+    trimmed = lines[start:]
+    return [line for line in trimmed if line]
+
+
 def _collect_renderer_outputs(
     renderer: Any, final_result: Any
 ) -> tuple[
@@ -211,13 +249,23 @@ def _derive_transcript_meta(
     if step_summaries:
         meta["transcript_steps"] = step_summaries
 
+    step_lines = _extract_step_summary_lines(renderer)
+    if step_lines:
+        meta["transcript_step_lines"] = step_lines
+
     stream_processor = getattr(renderer, "stream_processor", None)
     stream_started_at = (
-        getattr(stream_processor, "streaming_started_at", None)
-        if stream_processor is not None
-        else None
+        getattr(stream_processor, "streaming_started_at", None) if stream_processor is not None else None
     )
     finished_at = compute_finished_at(renderer)
+    state = getattr(renderer, "state", None)
+    if state is not None:
+        duration_hint = getattr(state, "final_duration_seconds", None)
+        if duration_hint is not None:
+            try:
+                meta["final_duration_seconds"] = float(duration_hint)
+            except Exception:
+                pass
     model_name = meta.get("model") or model
     return meta, stream_started_at, finished_at, model_name
 
@@ -236,16 +284,19 @@ def store_transcript_for_session(
     if not hasattr(renderer, "get_transcript_events"):
         return None
 
-    events, aggregated_output, final_output = _collect_renderer_outputs(
-        renderer, final_result
-    )
+    events, aggregated_output, final_output = _collect_renderer_outputs(renderer, final_result)
 
     if not (events or aggregated_output or final_output):
         return None
 
-    meta, stream_started_at, finished_at, model_name = _derive_transcript_meta(
-        renderer, model
-    )
+    meta, stream_started_at, finished_at, model_name = _derive_transcript_meta(renderer, model)
+
+    try:
+        api_url = _resolve_api_url(ctx)
+    except Exception:
+        api_url = None
+    if api_url:
+        meta["api_url"] = api_url
 
     payload: TranscriptPayload = build_transcript_payload(
         events=events,