glaip-sdk 0.1.2__py3-none-any.whl → 0.6.5b3__py3-none-any.whl

glaip_sdk/cli/slash/accounts_shared.py

@@ -0,0 +1,75 @@
+"""Shared helpers for palette `/accounts`.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+import os
+from typing import Any
+
+from glaip_sdk.cli.masking import mask_api_key_display
+
+
+def build_account_status_string(row: dict[str, Any], *, use_markup: bool = False) -> str:
+    """Build status string for an account row (active/env-lock).
+
+    When `use_markup` is True, returns Rich markup strings for Textual/Rich rendering;
+    when False, returns plain text for console output.
+
+    Example:
+        build_account_status_string({"active": True, "env_lock": True}, use_markup=True)
+        returns "[bold green]● active[/] · [yellow]🔒 env-lock[/]"
+        use_markup=False returns "● active · 🔒 env-lock"
+    """
+    status_parts: list[str] = []
+    if row.get("active"):
+        status_parts.append("[bold green]● active[/]" if use_markup else "● active")
+    if row.get("env_lock"):
+        status_parts.append("[yellow]🔒 env-lock[/]" if use_markup else "🔒 env-lock")
+    return " · ".join(status_parts)
+
+
+def env_credentials_present(*, partial: bool = False) -> bool:
+    """Return True when env credentials are present.
+
+    Args:
+        partial: When True, treat either AIP_API_URL or AIP_API_KEY as present
+            (used by UIs that should lock on any env override). When False,
+            require both to be non-empty (used for context display).
+    """
+    api_url = (os.getenv("AIP_API_URL") or "").strip()
+    api_key = (os.getenv("AIP_API_KEY") or "").strip()
+    if partial:
+        return bool(api_url or api_key)
+    return bool(api_url and api_key)
+
+
+def build_account_rows(
+    accounts: dict[str, dict[str, str]],
+    active_account: str | None,
+    env_lock: bool,
+) -> list[dict[str, str | bool]]:
+    """Build account rows for display from accounts dict.
+
+    Args:
+        accounts: Dictionary mapping account names to account data.
+        active_account: Name of the currently active account.
+        env_lock: Whether environment credentials are locking account switching.
+
+    Returns:
+        List of account row dictionaries with name, api_url, masked_key, active, and env_lock.
+    """
+    rows: list[dict[str, str | bool]] = []
+    for name, account in sorted(accounts.items()):
+        rows.append(
+            {
+                "name": name,
+                "api_url": account.get("api_url", ""),
+                "masked_key": mask_api_key_display(account.get("api_key", "")),
+                "active": name == active_account,
+                "env_lock": env_lock,
+            }
+        )
+    return rows

glaip_sdk/cli/slash/agent_session.py

@@ -14,8 +14,10 @@ import click
 from glaip_sdk.branding import ERROR_STYLE, HINT_PREFIX_STYLE
 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.constants import DEFAULT_AGENT_INSTRUCTION_PREVIEW_LIMIT
+from glaip_sdk.cli.hints import format_command_hint
 from glaip_sdk.cli.slash.prompt import _HAS_PROMPT_TOOLKIT, FormattedText
-from glaip_sdk.cli.utils import format_command_hint
+from glaip_sdk.cli.utils import bind_slash_session_context
 
 if TYPE_CHECKING:  # pragma: no cover - type checking only
     from glaip_sdk.cli.slash.session import SlashSession
@@ -38,11 +40,13 @@ class AgentRunSession:
         self._agent_name = getattr(agent, "name", "") or self._agent_id
         self._prompt_placeholder: str = "Chat with this agent here; use / for shortcuts. Alt+Enter inserts a newline."
         self._contextual_completion_help: dict[str, str] = {
-            "details": "Show this agent's full configuration.",
+            "details": "Show this agent's configuration (+ expands prompt).",
             "help": "Display this context-aware menu.",
+            "runs": "✨ NEW · Browse remote run history for this agent.",
             "exit": "Return to the command palette.",
             "q": "Return to the command palette.",
         }
+        self._instruction_preview_limit = DEFAULT_AGENT_INSTRUCTION_PREVIEW_LIMIT
 
     def run(self) -> None:
         """Run the interactive agent session loop."""
@@ -86,6 +90,7 @@ class AgentRunSession:
         try:
 
             def _prompt_message() -> Any:
+                """Get formatted prompt message for agent session."""
                 prompt_prefix = f"{self._agent_name} ({self._agent_id}) "
 
                 # Use FormattedText if prompt_toolkit is available, otherwise use simple string
@@ -122,7 +127,7 @@ class AgentRunSession:
         if raw in {"/exit", "/back", "/q"}:
             return self._handle_exit_command()
 
-        if raw in {"/details", "/detail"}:
+        if raw == "/details":
             return self._handle_details_command(agent_id)
 
         if raw in {"/help", "/?"}:
@@ -151,16 +156,59 @@ class AgentRunSession:
         self.session.handle_command(raw, invoked_from_agent=True)
         return not self.session._should_exit
 
-    def _show_details(self, agent_id: str) -> None:
+    def _show_details(self, agent_id: str, *, enable_prompt: bool = True) -> None:
+        """Render the agent's configuration export inside the command palette."""
         try:
-            self.session.ctx.invoke(agents_get_command, agent_ref=agent_id)
-            self.console.print(
-                f"[{HINT_PREFIX_STYLE}]Tip:[/] Continue the conversation in this prompt, or use "
-                f"{format_command_hint('/help') or '/help'} for shortcuts."
+            self.session.ctx.invoke(
+                agents_get_command,
+                agent_ref=agent_id,
+                instruction_preview=self._instruction_preview_limit,
             )
+            if enable_prompt:
+                self._prompt_instruction_view_toggle(agent_id)
+                self.console.print(
+                    f"[{HINT_PREFIX_STYLE}]Tip:[/] Continue the conversation in this prompt, or use "
+                    f"{format_command_hint('/help') or '/help'} for shortcuts."
+                )
         except click.ClickException as exc:
             self.console.print(f"[{ERROR_STYLE}]{exc}[/]")
 
+    def _prompt_instruction_view_toggle(self, agent_id: str) -> None:
+        """Offer a prompt to expand or collapse the instruction preview after details."""
+        if not getattr(self.console, "is_terminal", False):
+            return
+
+        while True:
+            mode = "expanded" if self._instruction_preview_limit == 0 else "trimmed"
+            self.console.print(f"[dim]Instruction view is {mode}. Press Ctrl+T to toggle, Enter to continue.[/dim]")
+            try:
+                ch = click.getchar()
+            except (EOFError, KeyboardInterrupt):  # pragma: no cover - defensive guard
+                return
+
+            if not self._handle_instruction_toggle_input(agent_id, ch):
+                break
+
+        if self._instruction_preview_limit == 0:
+            self._instruction_preview_limit = DEFAULT_AGENT_INSTRUCTION_PREVIEW_LIMIT
+        self.console.print("")
+
+    def _handle_instruction_toggle_input(self, agent_id: str, ch: str) -> bool:
+        """Process a single toggle keypress; return False when the loop should exit."""
+        if ch in {"\r", "\n"}:
+            return False
+
+        lowered = ch.lower()
+        if lowered == "t" or ch == "\x14":  # support literal 't' or Ctrl+T
+            self._instruction_preview_limit = (
+                DEFAULT_AGENT_INSTRUCTION_PREVIEW_LIMIT if self._instruction_preview_limit == 0 else 0
+            )
+            self._show_details(agent_id, enable_prompt=False)
+            return True
+
+        # Ignore other keys and continue prompting.
+        return True
+
     def _after_agent_run(self) -> None:
         """Handle transcript viewer behaviour after a successful run."""
         payload, manifest = self.session._get_last_transcript()
@@ -206,21 +254,11 @@ class AgentRunSession:
     @contextmanager
     def _bind_session_context(self) -> Any:
         """Temporarily attach this slash session to the Click context."""
-        ctx_obj = getattr(self.session.ctx, "obj", None)
-        has_context = isinstance(ctx_obj, dict)
-        previous_session = ctx_obj.get("_slash_session") if has_context else None
-        if has_context:
-            ctx_obj["_slash_session"] = self.session
-        try:
+        with bind_slash_session_context(self.session.ctx, self.session):
             yield
-        finally:
-            if has_context:
-                if previous_session is None:
-                    ctx_obj.pop("_slash_session", None)
-                else:
-                    ctx_obj["_slash_session"] = previous_session
 
     def _run_agent(self, agent_id: str, message: str) -> None:
+        """Execute the agents run command for the active agent."""
         if not message:
             return
 

glaip_sdk/cli/slash/prompt.py

@@ -123,6 +123,11 @@ def _create_key_bindings(_session: SlashSession) -> Any:
     bindings = KeyBindings()
 
     def _refresh_completions(buffer: Any) -> None:  # type: ignore[no-any-return]
+        """Refresh completions when slash command is typed.
+
+        Args:
+            buffer: Prompt buffer instance.
+        """
         text = buffer.document.text_before_cursor or ""
         if text.startswith("/") and " " not in text:
             buffer.start_completion(select_first=False)
@@ -163,6 +168,7 @@ def _create_key_bindings(_session: SlashSession) -> Any:
 def _iter_command_completions(
     session: SlashSession, text: str
 ) -> Iterable[Completion]:  # pragma: no cover - thin wrapper
+    """Yield completions for global slash commands."""
     prefix = text[1:]
     seen: set[str] = set()
 
@@ -171,8 +177,11 @@ def _iter_command_completions(
         return []
 
     commands = sorted(session._unique_commands.values(), key=lambda c: c.name)
+    agent_context = bool(getattr(session, "_current_agent", None))
 
     for cmd in commands:
+        if getattr(cmd, "agent_only", False) and not agent_context:
+            continue
         yield from _generate_command_completions(cmd, prefix, text, seen)
 
 
@@ -203,6 +212,7 @@ def _generate_command_completions(cmd: Any, prefix: str, text: str, seen: set[st
 def _iter_contextual_completions(
     session: SlashSession, text: str
 ) -> Iterable[Completion]:  # pragma: no cover - thin wrapper
+    """Yield completions for context-specific slash commands."""
     prefix = text[1:]
     seen: set[str] = set()