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

glaip_sdk/cli/masking.py

@@ -6,9 +6,10 @@ Authors:
 
 from __future__ import annotations
 
-import os
 from typing import Any
 
+from glaip_sdk.cli.constants import MASK_SENSITIVE_FIELDS, MASKING_ENABLED
+
 __all__ = [
     "mask_payload",
     "mask_rows",
@@ -16,20 +17,9 @@ __all__ = [
     "_mask_any",
     "_maybe_mask_row",
     "_resolve_mask_fields",
+    "mask_api_key_display",
 ]
 
-_DEFAULT_MASK_FIELDS = {
-    "api_key",
-    "apikey",
-    "token",
-    "access_token",
-    "secret",
-    "client_secret",
-    "password",
-    "private_key",
-    "bearer",
-}
-
 
 def _mask_value(raw: Any) -> str:
     """Return a masked representation of the provided value.
@@ -90,22 +80,10 @@ def _maybe_mask_row(row: dict[str, Any], mask_fields: set[str]) -> dict[str, Any
 
 
 def _resolve_mask_fields() -> set[str]:
-    """Resolve the set of sensitive fields to mask based on environment.
-
-    Returns:
-        set[str]: Set of field names to mask. Empty set if masking is disabled
-                 via AIP_MASK_OFF environment variable, custom fields from
-                 AIP_MASK_FIELDS, or default fields if neither is set.
-    """
-    if os.getenv("AIP_MASK_OFF", "0") in {"1", "true", "on", "yes"}:
+    """Return the configured set of fields that should be masked."""
+    if not MASKING_ENABLED:
         return set()
-
-    env_fields = (os.getenv("AIP_MASK_FIELDS") or "").strip()
-    if env_fields:
-        parts = [part.strip().lower() for part in env_fields.split(",") if part.strip()]
-        return set(parts)
-
-    return set(_DEFAULT_MASK_FIELDS)
+    return set(MASK_SENSITIVE_FIELDS)
 
 
 def mask_payload(payload: Any) -> Any:
@@ -115,9 +93,7 @@ def mask_payload(payload: Any) -> Any:
         payload: Any data structure (dict, list, or primitive) to mask.
 
     Returns:
-        Any: The payload with sensitive fields masked based on environment
-             configuration. Returns original payload if masking is disabled
-             or if an error occurs during masking.
+        Any: The payload with sensitive fields masked based on configuration.
     """
     mask_fields = _resolve_mask_fields()
     if not mask_fields:
@@ -136,8 +112,8 @@ def mask_rows(rows: list[dict[str, Any]]) -> list[dict[str, Any]]:
 
     Returns:
         list[dict[str, Any]]: List of rows with sensitive fields masked based
-                              on environment configuration. Returns original
-                              rows if masking is disabled or if an error occurs.
+                              on configuration. Returns original rows if
+                              masking is disabled or if an error occurs.
     """
     mask_fields = _resolve_mask_fields()
     if not mask_fields:
@@ -146,3 +122,15 @@ def mask_rows(rows: list[dict[str, Any]]) -> list[dict[str, Any]]:
         return [_maybe_mask_row(row, mask_fields) for row in rows]
     except Exception:
         return rows
+
+
+def mask_api_key_display(value: str | None) -> str:
+    """Mask API keys for CLI display while preserving readability for short keys."""
+    if not value:
+        return ""
+    length = len(value)
+    if length <= 4:
+        return "***"
+    if length <= 8:
+        return value[:1] + "••••" + value[-1:]
+    return value[:4] + "••••" + value[-4:]

glaip_sdk/cli/pager.py

@@ -19,6 +19,8 @@ from typing import Any
 
 from rich.console import Console
 
+from glaip_sdk.cli.constants import PAGER_HEADER_ENABLED, PAGER_MODE, PAGER_WRAP_LINES
+
 __all__ = [
     "console",
     "_prepare_pager_env",
@@ -64,8 +66,7 @@ def _prepare_pager_env(clear_on_exit: bool = True) -> None:
       -R  : pass ANSI color escapes
       -S  : chop long lines (horizontal scroll with ←/→)
     (No -F, no -X) so we open a full-screen pager and clear on exit.
-    Toggle wrapping with AIP_PAGER_WRAP=1 to drop -S.
-    Power users can override via AIP_LESS_FLAGS.
+    Toggle wrapping via `PAGER_WRAP_LINES` (True drops -S).
 
     Args:
         clear_on_exit: Whether to clear the pager on exit (default: True)
@@ -75,10 +76,9 @@ def _prepare_pager_env(clear_on_exit: bool = True) -> None:
     """
     os.environ.pop("LESSSECURE", None)
     if os.getenv("LESS") is None:
-        want_wrap = os.getenv("AIP_PAGER_WRAP", "0") == "1"
-        base = "-R" if want_wrap else "-RS"
+        base = "-R" if PAGER_WRAP_LINES else "-RS"
         default_flags = base if clear_on_exit else (base + "FX")
-        os.environ["LESS"] = os.getenv("AIP_LESS_FLAGS", default_flags)
+        os.environ["LESS"] = default_flags
 
 
 def _render_ansi(renderable: Any) -> str:
@@ -111,8 +111,7 @@ def _pager_header() -> str:
     Returns:
         str: Header text containing navigation help, or empty string if disabled
     """
-    v = (os.getenv("AIP_PAGER_HEADER", "1") or "1").strip().lower()
-    if v in {"0", "false", "off"}:
+    if not PAGER_HEADER_ENABLED:
         return ""
     return "\n".join(
         [
@@ -254,10 +253,10 @@ def _should_page_output(row_count: int, is_tty: bool) -> bool:
         bool: True if output should be paginated, False otherwise
     """
     active_console = _get_console()
-    pager_env = (os.getenv("AIP_PAGER", "auto") or "auto").lower()
-    if pager_env in ("0", "off", "false"):
+    pager_mode = (PAGER_MODE or "auto").lower()
+    if pager_mode in ("0", "off", "false"):
         return False
-    if pager_env in ("1", "on", "true"):
+    if pager_mode in ("1", "on", "true"):
         return is_tty
     try:
         term_h = active_console.size.height or 24

glaip_sdk/cli/parsers/__init__.py

@@ -4,6 +4,4 @@ Authors:
     Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
 """
 
-from glaip_sdk.cli.parsers.json_input import parse_json_input
-
-__all__ = ["parse_json_input"]
+__all__: list[str] = []

glaip_sdk/cli/slash/__init__.py

@@ -6,19 +6,10 @@ Authors:
 
 from glaip_sdk.cli.commands.agents import get as agents_get_command
 from glaip_sdk.cli.commands.agents import run as agents_run_command
-from glaip_sdk.cli.commands.configure import configure_command, load_config
-from glaip_sdk.cli.slash.agent_session import AgentRunSession
-from glaip_sdk.cli.slash.prompt import _HAS_PROMPT_TOOLKIT
 from glaip_sdk.cli.slash.session import SlashSession
-from glaip_sdk.cli.utils import get_client
 
 __all__ = [
-    "AgentRunSession",
     "SlashSession",
-    "_HAS_PROMPT_TOOLKIT",
     "agents_get_command",
     "agents_run_command",
-    "configure_command",
-    "get_client",
-    "load_config",
 ]