glaip-sdk 0.1.0__py3-none-any.whl → 0.6.10__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/mcp_validators.py

@@ -79,9 +79,7 @@ def validate_mcp_config_structure(
         requirement = "Missing required 'url' field with a non-empty string value."
         if transport:
             requirement += f" Required for transport '{transport}'."
-        raise click.ClickException(
-            format_validation_error(f"Invalid {source} value", requirement)
-        )
+        raise click.ClickException(format_validation_error(f"Invalid {source} value", requirement))
 
     parsed_url = urlparse(url_value)
     if parsed_url.scheme not in {"http", "https"} or not parsed_url.netloc:
@@ -95,9 +93,7 @@ def validate_mcp_config_structure(
     return config
 
 
-def _validate_headers_mapping(
-    headers: Any, *, source: str, context: str
-) -> dict[str, str]:
+def _validate_headers_mapping(headers: Any, *, source: str, context: str) -> dict[str, str]:
     """Validate headers mapping for authentication.
 
     Args:
@@ -156,9 +152,7 @@ def _validate_bearer_token_auth(auth: dict[str, Any], source: str) -> dict[str,
     if isinstance(token, str) and token.strip():
         return {"type": "bearer-token", "token": token}
     headers = auth.get("headers")
-    normalized_headers = _validate_headers_mapping(
-        headers, source=source, context="bearer-token authentication"
-    )
+    normalized_headers = _validate_headers_mapping(headers, source=source, context="bearer-token authentication")
     return {"type": "bearer-token", "headers": normalized_headers}
 
 
@@ -177,9 +171,7 @@ def _validate_api_key_auth(auth: dict[str, Any], source: str) -> dict[str, Any]:
     """
     headers = auth.get("headers")
     if headers is not None:
-        normalized_headers = _validate_headers_mapping(
-            headers, source=source, context="api-key authentication"
-        )
+        normalized_headers = _validate_headers_mapping(headers, source=source, context="api-key authentication")
         return {"type": "api-key", "headers": normalized_headers}
 
     key = auth.get("key")
@@ -215,9 +207,7 @@ def _validate_custom_header_auth(auth: dict[str, Any], source: str) -> dict[str,
         click.ClickException: If custom-header structure is invalid
     """
     headers = auth.get("headers")
-    normalized_headers = _validate_headers_mapping(
-        headers, source=source, context="custom-header authentication"
-    )
+    normalized_headers = _validate_headers_mapping(headers, source=source, context="custom-header authentication")
     return {"type": "custom-header", "headers": normalized_headers}
 
 

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(
         [
@@ -156,9 +155,7 @@ def _resolve_pager_command() -> tuple[list[str] | None, str | None]:
     return pager_cmd, less_path
 
 
-def _run_less_pager(
-    pager_cmd: list[str] | None, less_path: str | None, tmp_path: str
-) -> None:
+def _run_less_pager(pager_cmd: list[str] | None, less_path: str | None, tmp_path: str) -> None:
     """Run less pager with appropriate command and flags.
 
     Args:
@@ -195,9 +192,7 @@ def _run_more_pager(tmp_path: str) -> None:
         raise FileNotFoundError("more command not found")
 
 
-def _run_pager_with_temp_file(
-    pager_runner: Callable[[str], None], ansi_text: str
-) -> bool:
+def _run_pager_with_temp_file(pager_runner: Callable[[str], None], ansi_text: str) -> bool:
     """Run a pager using a temporary file containing the content.
 
     Args:
@@ -239,9 +234,7 @@ def _page_with_system_pager(ansi_text: str) -> bool:
     pager_cmd, less_path = _resolve_pager_command()
 
     if pager_cmd or less_path:
-        return _run_pager_with_temp_file(
-            lambda tmp_path: _run_less_pager(pager_cmd, less_path, tmp_path), ansi_text
-        )
+        return _run_pager_with_temp_file(lambda tmp_path: _run_less_pager(pager_cmd, less_path, tmp_path), ansi_text)
 
     if platform.system().lower().startswith("win"):
         return False
@@ -260,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/parsers/json_input.py

@@ -33,9 +33,7 @@ def _looks_like_file_path(value: str) -> bool:
     )
 
 
-def _format_file_error(
-    prefix: str, file_path_str: str, resolved_path: Path, *, detail: str | None = None
-) -> str:
+def _format_file_error(prefix: str, file_path_str: str, resolved_path: Path, *, detail: str | None = None) -> str:
     r"""Format a file-related error message with path context.
 
     Args:
@@ -77,16 +75,12 @@ def _parse_json_from_file(file_path_str: str) -> Any:
 
     # Check if file exists and is a regular file
     if not file_path.is_file():
-        raise click.ClickException(
-            _format_file_error("File not found or not a file", file_path_str, file_path)
-        )
+        raise click.ClickException(_format_file_error("File not found or not a file", file_path_str, file_path))
 
     # Check if file is readable
     if not os.access(file_path, os.R_OK):
         raise click.ClickException(
-            _format_file_error(
-                "File not readable (permission denied)", file_path_str, file_path
-            )
+            _format_file_error("File not readable (permission denied)", file_path_str, file_path)
         )
 
     # Read file content
@@ -94,16 +88,12 @@ def _parse_json_from_file(file_path_str: str) -> Any:
         content = file_path.read_text(encoding="utf-8")
     except Exception as e:
         raise click.ClickException(
-            _format_file_error(
-                "Error reading file", file_path_str, file_path, detail=f"Error: {e}"
-            )
-        )
+            _format_file_error("Error reading file", file_path_str, file_path, detail=f"Error: {e}")
+        ) from e
 
     # Check for empty content
     if not content.strip():
-        raise click.ClickException(
-            _format_file_error("File is empty", file_path_str, file_path)
-        )
+        raise click.ClickException(_format_file_error("File is empty", file_path_str, file_path))
 
     # Determine file format and parse accordingly
     file_ext = file_path.suffix.lower()
@@ -120,7 +110,7 @@ def _parse_json_from_file(file_path_str: str) -> Any:
                     file_path,
                     detail=f"Error: {e}",
                 )
-            )
+            ) from e
     else:
         # Default to JSON parsing
         try:
@@ -133,7 +123,7 @@ def _parse_json_from_file(file_path_str: str) -> Any:
                     file_path,
                     detail=f"Error: {e.msg} at line {e.lineno}, column {e.colno}",
                 )
-            )
+            ) from e
 
 
 def parse_json_input(value: str | None) -> Any:
@@ -180,9 +170,8 @@ def parse_json_input(value: str | None) -> Any:
                 f"Error: {e.msg} at line {e.lineno}, column {e.colno}\n"
                 f"\n💡 Did you mean to load this from a file? "
                 f"File-based config values should start with @ (e.g., @{trimmed})"
-            )
+            ) from e
 
         raise click.ClickException(
-            f"Invalid JSON in inline value\n"
-            f"Error: {e.msg} at line {e.lineno}, column {e.colno}"
-        )
+            f"Invalid JSON in inline value\nError: {e.msg} at line {e.lineno}, column {e.colno}"
+        ) from e

glaip_sdk/cli/resolution.py

@@ -51,14 +51,8 @@ def resolve_resource_reference(
         click.ClickException: If resolution fails.
     """
     try:
-        message = (
-            spinner_message
-            if spinner_message is not None
-            else f"[bold blue]Fetching {label}…[/bold blue]"
-        )
-        with spinner_context(
-            ctx, message, spinner_style=ACCENT_STYLE
-        ) as status_indicator:
+        message = spinner_message if spinner_message is not None else f"[bold blue]Fetching {label}…[/bold blue]"
+        with spinner_context(ctx, message, spinner_style=ACCENT_STYLE) as status_indicator:
             return resolve_resource(
                 ctx,
                 reference,
@@ -70,4 +64,4 @@ def resolve_resource_reference(
                 status_indicator=status_indicator,
             )
     except Exception as e:
-        raise click.ClickException(f"Failed to resolve {resource_type.lower()}: {e}")
+        raise click.ClickException(f"Failed to resolve {resource_type.lower()}: {e}") from e

glaip_sdk/cli/rich_helpers.py

@@ -21,9 +21,7 @@ def markup_text(message: str, **kwargs: Any) -> Text:
         return Text(message, **kwargs)
 
 
-def print_markup(
-    message: str, *, console: Console | None = None, **kwargs: Any
-) -> None:
+def print_markup(message: str, *, console: Console | None = None, **kwargs: Any) -> None:
     """Print markup-aware text to the provided console (default: new Console)."""
     target_console = console or Console()
     target_console.print(markup_text(message, **kwargs))

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",
 ]