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

glaip_sdk/utils/client_utils.py

@@ -90,9 +90,7 @@ def extract_ids(items: list[str | Any] | None) -> list[str] | None:
     return result if result else None
 
 
-def create_model_instances(
-    data: list[dict] | None, model_class: type, client: Any
-) -> list[Any]:
+def create_model_instances(data: list[dict] | None, model_class: type, client: Any) -> list[Any]:
     """Create model instances from API data with client association.
 
     This is a common pattern used across different clients (agents, tools, mcps)
@@ -112,9 +110,7 @@ def create_model_instances(
     return [model_class(**item_data)._set_client(client) for item_data in data]
 
 
-def find_by_name(
-    items: list[Any], name: str, case_sensitive: bool = False
-) -> list[Any]:
+def find_by_name(items: list[Any], name: str, case_sensitive: bool = False) -> list[Any]:
     """Filter items by name with optional case sensitivity.
 
     This is a common pattern used across different clients for client-side
@@ -234,9 +230,7 @@ def _handle_streaming_error(
     if isinstance(e, httpx.ReadTimeout):
         logger.error(f"Read timeout during streaming: {e}")
         logger.error("This usually indicates the backend is taking too long to respond")
-        logger.error(
-            "Consider increasing the timeout value or checking backend performance"
-        )
+        logger.error("Consider increasing the timeout value or checking backend performance")
         raise AgentTimeoutError(timeout_seconds or 30.0, agent_name)
 
     elif isinstance(e, httpx.TimeoutException):
@@ -275,9 +269,7 @@ def _yield_event_data(event_data: dict[str, Any] | None) -> Iterator[dict[str, A
         yield event_data
 
 
-def _flush_remaining_buffer(
-    buf: list[str], event_type: str | None, event_id: str | None
-) -> Iterator[dict[str, Any]]:
+def _flush_remaining_buffer(buf: list[str], event_type: str | None, event_id: str | None) -> Iterator[dict[str, Any]]:
     """Flush any remaining data in buffer."""
     if buf:
         yield {
@@ -317,9 +309,7 @@ def iter_sse_events(
             if line is None:
                 continue
 
-            buf, event_type, event_id, event_data, completed = _process_sse_line(
-                line, buf, event_type, event_id
-            )
+            buf, event_type, event_id, event_data, completed = _process_sse_line(line, buf, event_type, event_id)
 
             yield from _yield_event_data(event_data)
             if completed:
@@ -385,9 +375,7 @@ def _create_form_data(message: str) -> dict[str, Any]:
     return {"input": message, "message": message, "stream": True}
 
 
-def _prepare_file_entry(
-    item: str | BinaryIO, stack: ExitStack
-) -> tuple[str, tuple[str, BinaryIO, str]]:
+def _prepare_file_entry(item: str | BinaryIO, stack: ExitStack) -> tuple[str, tuple[str, BinaryIO, str]]:
     """Prepare a single file entry for multipart data."""
     if isinstance(item, str):
         return _prepare_path_entry(item, stack)
@@ -395,9 +383,7 @@ def _prepare_file_entry(
         return _prepare_stream_entry(item)
 
 
-def _prepare_path_entry(
-    path_str: str, stack: ExitStack
-) -> tuple[str, tuple[str, BinaryIO, str]]:
+def _prepare_path_entry(path_str: str, stack: ExitStack) -> tuple[str, tuple[str, BinaryIO, str]]:
     """Prepare a file path entry."""
     file_path = Path(path_str)
     if not file_path.exists():

glaip_sdk/utils/display.py

@@ -109,9 +109,7 @@ def print_agent_updated(agent: Any) -> None:
     """
     if RICH_AVAILABLE:
         console = _create_console()
-        console.print(
-            f"[{SUCCESS_STYLE}]✅ Agent '{agent.name}' updated successfully[/]"
-        )
+        console.print(f"[{SUCCESS_STYLE}]✅ Agent '{agent.name}' updated successfully[/]")
     else:
         print(f"✅ Agent '{agent.name}' updated successfully")
 
@@ -124,8 +122,6 @@ def print_agent_deleted(agent_id: str) -> None:
     """
     if RICH_AVAILABLE:
         console = _create_console()
-        console.print(
-            f"[{SUCCESS_STYLE}]✅ Agent deleted successfully (ID: {agent_id})[/]"
-        )
+        console.print(f"[{SUCCESS_STYLE}]✅ Agent deleted successfully (ID: {agent_id})[/]")
     else:
         print(f"✅ Agent deleted successfully (ID: {agent_id})")

glaip_sdk/utils/general.py

@@ -76,9 +76,7 @@ def format_datetime(dt: datetime | str | None) -> str:
     return str(dt)
 
 
-def progress_bar(
-    iterable: Iterable[Any], description: str = "Processing"
-) -> Iterator[Any]:
+def progress_bar(iterable: Iterable[Any], description: str = "Processing") -> Iterator[Any]:
     """Simple progress bar using click.
 
     Args:

glaip_sdk/utils/import_export.py

@@ -71,14 +71,10 @@ def _get_default_array_fields() -> list[str]:
 
 def _should_use_cli_value(cli_value: Any) -> bool:
     """Check if CLI value should be used."""
-    return cli_value is not None and (
-        not isinstance(cli_value, list | tuple) or len(cli_value) > 0
-    )
+    return cli_value is not None and (not isinstance(cli_value, (list, tuple)) or len(cli_value) > 0)
 
 
-def _handle_array_field_merge(
-    key: str, cli_value: Any, import_data: dict[str, Any]
-) -> Any:
+def _handle_array_field_merge(key: str, cli_value: Any, import_data: dict[str, Any]) -> Any:
     """Handle merging of array fields."""
     import_value = import_data[key]
     if isinstance(import_value, list):
@@ -107,9 +103,7 @@ def _merge_cli_values_with_import(
             merged[key] = import_data[key]
 
 
-def _add_import_only_fields(
-    merged: dict[str, Any], import_data: dict[str, Any]
-) -> None:
+def _add_import_only_fields(merged: dict[str, Any], import_data: dict[str, Any]) -> None:
     """Add fields that exist only in import data."""
     for key, import_value in import_data.items():
         if key not in merged:

glaip_sdk/utils/rendering/formatting.py

@@ -12,7 +12,14 @@ import time
 from collections.abc import Callable
 from typing import Any
 
-from glaip_sdk.icons import ICON_AGENT_STEP, ICON_DELEGATE, ICON_TOOL_STEP
+from glaip_sdk.icons import (
+    ICON_AGENT_STEP,
+    ICON_DELEGATE,
+    ICON_STATUS_FAILED,
+    ICON_STATUS_SUCCESS,
+    ICON_STATUS_WARNING,
+    ICON_TOOL_STEP,
+)
 
 # Constants for argument formatting
 DEFAULT_ARGS_MAX_LEN = 100
@@ -37,9 +44,20 @@ SECRET_VALUE_PATTERNS = [
     re.compile(r"eyJ[a-zA-Z0-9_-]+\.[a-zA-Z0-9_-]+\.[a-zA-Z0-9_-]+"),  # JWT tokens
 ]
 SENSITIVE_PATTERNS = re.compile(
-    r"password\s*[:=]\s*[^\s,}]+|secret\s*[:=]\s*[^\s,}]+|token\s*[:=]\s*[^\s,}]+|key\s*[:=]\s*[^\s,}]+|api_key\s*[:=]\s*[^\s,}]+|^password$|^secret$|^token$|^key$|^api_key$",
+    r"(?:password|secret|token|key|api_key)(?:\s*[:=]\s*[^\s,}]+)?",
     re.IGNORECASE,
 )
+CONNECTOR_VERTICAL = "│   "
+CONNECTOR_EMPTY = "    "
+CONNECTOR_BRANCH = "├─ "
+CONNECTOR_LAST = "└─ "
+ROOT_MARKER = ""
+SECRET_MASK = "••••••"
+STATUS_GLYPHS = {
+    "success": ICON_STATUS_SUCCESS,
+    "failed": ICON_STATUS_FAILED,
+    "warning": ICON_STATUS_WARNING,
+}
 
 
 def _truncate_string(s: str, max_len: int) -> str:
@@ -53,7 +71,7 @@ def mask_secrets_in_string(text: str) -> str:
     """Mask sensitive information in a string."""
     result = text
     for pattern in SECRET_VALUE_PATTERNS:
-        result = re.sub(pattern, "••••••", result)
+        result = re.sub(pattern, SECRET_MASK, result)
     return result
 
 
@@ -74,7 +92,7 @@ def _redact_dict_values(text: dict) -> dict:
     result = {}
     for key, value in text.items():
         if _is_sensitive_key(key):
-            result[key] = "••••••"
+            result[key] = SECRET_MASK
         elif _should_recurse_redaction(value):
             result[key] = redact_sensitive(value)
         else:
@@ -92,11 +110,11 @@ def _redact_string_content(text: str) -> str:
     result = text
     # First mask secrets
     for pattern in SECRET_VALUE_PATTERNS:
-        result = re.sub(pattern, "••••••", result)
+        result = re.sub(pattern, SECRET_MASK, result)
     # Then redact sensitive patterns
     result = re.sub(
         SENSITIVE_PATTERNS,
-        lambda m: m.group(0).split("=")[0] + "=••••••",
+        lambda m: m.group(0).split("=")[0] + "=" + SECRET_MASK,
         result,
     )
     return result
@@ -105,15 +123,37 @@ def _redact_string_content(text: str) -> str:
 def _is_sensitive_key(key: str) -> bool:
     """Check if a key contains sensitive information."""
     key_lower = key.lower()
-    return any(
-        sensitive in key_lower
-        for sensitive in ["password", "secret", "token", "key", "api_key"]
-    )
+    return any(sensitive in key_lower for sensitive in ["password", "secret", "token", "key", "api_key"])
 
 
 def _should_recurse_redaction(value: Any) -> bool:
     """Check if a value should be recursively processed."""
-    return isinstance(value, dict | list) or isinstance(value, str)
+    return isinstance(value, (dict, list)) or isinstance(value, str)
+
+
+def glyph_for_status(icon_key: str | None) -> str | None:
+    """Return glyph representing a step status icon key."""
+    if not icon_key:
+        return None
+    return STATUS_GLYPHS.get(icon_key)
+
+
+def normalise_display_label(label: str | None) -> str:
+    """Return a user facing label or the Unknown fallback."""
+    label = (label or "").strip()
+    return label or "Unknown step detail"
+
+
+def build_connector_prefix(branch_state: tuple[bool, ...]) -> str:
+    """Build connector prefix for a tree line based on ancestry state."""
+    if not branch_state:
+        return ROOT_MARKER
+
+    parts: list[str] = []
+    for ancestor_is_last in branch_state[:-1]:
+        parts.append(CONNECTOR_EMPTY if ancestor_is_last else CONNECTOR_VERTICAL)
+    parts.append(CONNECTOR_LAST if branch_state[-1] else CONNECTOR_BRANCH)
+    return "".join(parts)
 
 
 def pretty_args(args: dict | None, max_len: int = DEFAULT_ARGS_MAX_LEN) -> str:
@@ -132,7 +172,7 @@ def pretty_args(args: dict | None, max_len: int = DEFAULT_ARGS_MAX_LEN) -> str:
     try:
         args_str = json.dumps(masked_args, ensure_ascii=False, separators=(",", ":"))
         return _truncate_string(args_str, max_len)
-    except (TypeError, ValueError, Exception):
+    except Exception:
         # Fallback to string representation if JSON serialization fails
         args_str = str(masked_args)
         return _truncate_string(args_str, max_len)

glaip_sdk/utils/rendering/models.py

@@ -30,19 +30,32 @@ class Step:
     context_id: str | None = None
     started_at: float = field(default_factory=monotonic)
     duration_ms: int | None = None
-
-    def finish(self, duration_raw: float | None) -> None:
+    duration_source: str | None = None
+    display_label: str | None = None
+    status_icon: str | None = None
+    failure_reason: str | None = None
+    branch_failed: bool = False
+    is_parallel: bool = False
+    server_started_at: float | None = None
+    server_finished_at: float | None = None
+    duration_unknown: bool = False
+
+    def finish(self, duration_raw: float | None, *, source: str | None = None) -> None:
         """Mark the step as finished and calculate duration.
 
         Args:
             duration_raw: Raw duration in seconds, or None to calculate from started_at
+            source: Optional duration source tag
         """
-        if isinstance(duration_raw, int | float) and duration_raw > 0:
+        self.duration_unknown = False
+        if isinstance(duration_raw, (int, float)) and duration_raw > 0:
             # Use provided duration if it's a positive number (even if very small)
             self.duration_ms = round(float(duration_raw) * 1000)
+            self.duration_source = source or self.duration_source or "provided"
         else:
             # Calculate from started_at if duration_raw is None, negative, or zero
             self.duration_ms = int((monotonic() - self.started_at) * 1000)
+            self.duration_source = source or self.duration_source or "monotonic"
         self.status = "finished"
 
 
@@ -68,8 +81,4 @@ class RunStats:
         Returns:
             Duration in seconds if run is finished, None otherwise
         """
-        return (
-            None
-            if self.finished_at is None
-            else round(self.finished_at - self.started_at, 2)
-        )
+        return None if self.finished_at is None else round(self.finished_at - self.started_at, 2)

glaip_sdk/utils/rendering/renderer/__init__.py

@@ -35,12 +35,9 @@ def make_silent_renderer() -> RichStreamRenderer:
     cfg = RendererConfig(
         live=False,
         persist_live=False,
-        show_delegate_tool_panels=False,
         render_thinking=False,
     )
-    return RichStreamRenderer(
-        console=Console(file=io.StringIO(), force_terminal=False), cfg=cfg
-    )
+    return RichStreamRenderer(console=Console(file=io.StringIO(), force_terminal=False), cfg=cfg)
 
 
 def make_minimal_renderer() -> RichStreamRenderer:
@@ -51,7 +48,6 @@ def make_minimal_renderer() -> RichStreamRenderer:
     cfg = RendererConfig(
         live=False,
         persist_live=False,
-        show_delegate_tool_panels=False,
         render_thinking=False,
     )
     return RichStreamRenderer(console=Console(), cfg=cfg)