glaip-sdk 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

glaip_sdk/branding.py

@@ -31,8 +31,8 @@ except Exception:  # pragma: no cover
 
 
 # ---- minimal, readable styles (light blue + white theme) -----------------
-PRIMARY = "bright_blue"  # Light blue for main elements
-BORDER = "bright_blue"  # Light blue borders
+PRIMARY = "#15a2d8"  # GDP Labs brand blue
+BORDER = PRIMARY  # Keep borders aligned with brand tone
 TITLE_STYLE = f"bold {PRIMARY}"
 LABEL = "bold"
 
@@ -95,7 +95,7 @@ GDP Labs AI Agents Package
     # ---- public API -----------------------------------------------------------
     def get_welcome_banner(self) -> str:
         """Get AIP banner with version info."""
-        banner = self.AIP_LOGO
+        banner = f"[{PRIMARY}]{self.AIP_LOGO}[/{PRIMARY}]"
         line = f"Version: {self.version}"
         banner = f"{banner}\n{line}"
         return banner

glaip_sdk/cli/commands/agents.py

@@ -6,6 +6,7 @@ Authors:
 
 import json
 import os
+from collections.abc import Mapping
 from pathlib import Path
 from typing import Any
 
@@ -53,6 +54,7 @@ from glaip_sdk.cli.utils import (
     output_flags,
     output_list,
     output_result,
+    spinner_context,
 )
 from glaip_sdk.cli.validators import (
     validate_agent_instruction_cli as validate_agent_instruction,
@@ -76,6 +78,91 @@ console = Console()
 AGENT_NOT_FOUND_ERROR = "Agent not found"
 
 
+def _safe_agent_attribute(agent: Any, name: str) -> Any:
+    """Return attribute value for ``name`` while filtering Mock sentinels."""
+
+    try:
+        value = getattr(agent, name)
+    except Exception:
+        return None
+
+    if hasattr(value, "_mock_name"):
+        return None
+    return value
+
+
+def _coerce_mapping_candidate(candidate: Any) -> dict[str, Any] | None:
+    """Convert a mapping-like candidate to a plain dict when possible."""
+
+    if candidate is None:
+        return None
+    if isinstance(candidate, Mapping):
+        return dict(candidate)
+    return None
+
+
+def _call_agent_method(agent: Any, method_name: str) -> dict[str, Any] | None:
+    """Attempt to call the named method and coerce its output to a dict."""
+
+    method = getattr(agent, method_name, None)
+    if not callable(method):
+        return None
+    try:
+        candidate = method()
+    except Exception:
+        return None
+    return _coerce_mapping_candidate(candidate)
+
+
+def _coerce_agent_via_methods(agent: Any) -> dict[str, Any] | None:
+    """Try standard serialisation helpers to produce a mapping."""
+
+    for attr in ("model_dump", "dict", "to_dict"):
+        mapping = _call_agent_method(agent, attr)
+        if mapping is not None:
+            return mapping
+    return None
+
+
+def _build_fallback_agent_mapping(agent: Any) -> dict[str, Any]:
+    """Construct a minimal mapping from well-known agent attributes."""
+
+    fallback_fields = (
+        "id",
+        "name",
+        "instruction",
+        "description",
+        "model",
+        "agent_config",
+        "tools",
+        "agents",
+        "mcps",
+        "timeout",
+    )
+
+    fallback: dict[str, Any] = {}
+    for field in fallback_fields:
+        value = _safe_agent_attribute(agent, field)
+        if value is not None:
+            fallback[field] = value
+
+    return fallback or {"name": str(agent)}
+
+
+def _prepare_agent_output(agent: Any) -> dict[str, Any]:
+    """Build a JSON-serialisable mapping for CLI output."""
+
+    method_mapping = _coerce_agent_via_methods(agent)
+    if method_mapping is not None:
+        return method_mapping
+
+    intrinsic = _coerce_mapping_candidate(agent)
+    if intrinsic is not None:
+        return intrinsic
+
+    return _build_fallback_agent_mapping(agent)
+
+
 def _fetch_full_agent_details(client: Any, agent: Any) -> Any | None:
     """Fetch full agent details by ID to ensure all fields are populated."""
     try:
@@ -199,7 +286,12 @@ def _display_agent_details(ctx: Any, client: Any, agent: Any) -> None:
         return
 
     # Try to fetch and format raw agent data first
-    formatted_data = _fetch_and_format_raw_agent_data(client, agent)
+    with spinner_context(
+        ctx,
+        "[bold blue]Loading agent details…[/bold blue]",
+        console_override=console,
+    ):
+        formatted_data = _fetch_and_format_raw_agent_data(client, agent)
 
     if formatted_data:
         # Use raw API data - this preserves ALL fields including account_id
@@ -216,7 +308,12 @@ def _display_agent_details(ctx: Any, client: Any, agent: Any) -> None:
             ctx, Text("[yellow]Falling back to Pydantic model data[/yellow]")
         )
 
-        result_data = _format_fallback_agent_data(client, agent)
+        with spinner_context(
+            ctx,
+            "[bold blue]Preparing fallback agent details…[/bold blue]",
+            console_override=console,
+        ):
+            result_data = _format_fallback_agent_data(client, agent)
 
         # Display using output_result
         output_result(
@@ -289,13 +386,18 @@ def list_agents(
     """List agents with optional filtering."""
     try:
         client = get_client(ctx)
-        agents = client.agents.list_agents(
-            agent_type=agent_type,
-            framework=framework,
-            name=name,
-            version=version,
-            sync_langflow_agents=sync_langflow,
-        )
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching agents…[/bold blue]",
+            console_override=console,
+        ):
+            agents = client.agents.list_agents(
+                agent_type=agent_type,
+                framework=framework,
+                name=name,
+                version=version,
+                sync_langflow_agents=sync_langflow,
+            )
 
         # Define table columns: (data_key, header, style, width)
         columns = [
@@ -364,7 +466,12 @@ def get(ctx: Any, agent_ref: str, select: int | None, export: str | None) -> Non
 
             # Always export comprehensive data - re-fetch agent with full details
             try:
-                agent = client.agents.get_agent_by_id(agent.id)
+                with spinner_context(
+                    ctx,
+                    "[bold blue]Fetching complete agent data…[/bold blue]",
+                    console_override=console,
+                ):
+                    agent = client.agents.get_agent_by_id(agent.id)
             except Exception as e:
                 handle_rich_output(
                     ctx,
@@ -776,7 +883,7 @@ def _get_language_model_display_name(agent: Any, model: str | None) -> str:
 
 def _handle_successful_creation(ctx: Any, agent: Any, model: str | None) -> None:
     """Handle successful agent creation output."""
-    handle_json_output(ctx, agent.model_dump())
+    handle_json_output(ctx, _prepare_agent_output(agent))
 
     lm_display = _get_language_model_display_name(agent, model)
 
@@ -1080,7 +1187,7 @@ def update(
 
         updated_agent = client.agents.update_agent(agent.id, **update_data)
 
-        handle_json_output(ctx, updated_agent.model_dump())
+        handle_json_output(ctx, _prepare_agent_output(updated_agent))
         handle_rich_output(ctx, display_update_success("Agent", updated_agent.name))
         handle_rich_output(ctx, display_agent_run_suggestions(updated_agent))
 

glaip_sdk/cli/commands/mcps.py

@@ -36,6 +36,7 @@ from glaip_sdk.cli.utils import (
     output_flags,
     output_list,
     output_result,
+    spinner_context,
 )
 from glaip_sdk.rich_components import AIPPanel
 from glaip_sdk.utils import format_datetime
@@ -72,7 +73,12 @@ def list_mcps(ctx: Any) -> None:
     """List all MCPs."""
     try:
         client = get_client(ctx)
-        mcps = client.mcps.list_mcps()
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching MCPs…[/bold blue]",
+            console_override=console,
+        ):
+            mcps = client.mcps.list_mcps()
 
         # Define table columns: (data_key, header, style, width)
         columns = [
@@ -123,13 +129,18 @@ def create(
             except json.JSONDecodeError:
                 raise click.ClickException("Invalid JSON in --config")
 
-        mcp = client.mcps.create_mcp(
-            name=name,
-            type="server",  # MCPs are always server type
-            transport=transport,
-            description=description,
-            config=mcp_config,
-        )
+        with spinner_context(
+            ctx,
+            "[bold blue]Creating MCP…[/bold blue]",
+            console_override=console,
+        ):
+            mcp = client.mcps.create_mcp(
+                name=name,
+                type="server",  # MCPs are always server type
+                transport=transport,
+                description=description,
+                config=mcp_config,
+            )
 
         # Handle JSON output
         handle_json_output(ctx, mcp.model_dump())
@@ -183,7 +194,12 @@ def get(ctx: Any, mcp_ref: str, export: str | None) -> None:
 
             # Always export comprehensive data - re-fetch MCP with full details if needed
             try:
-                mcp = client.mcps.get_mcp_by_id(mcp.id)
+                with spinner_context(
+                    ctx,
+                    "[bold blue]Fetching complete MCP details…[/bold blue]",
+                    console_override=console,
+                ):
+                    mcp = client.mcps.get_mcp_by_id(mcp.id)
             except Exception as e:
                 console.print(
                     Text(f"[yellow]⚠️  Could not fetch full MCP details: {e}[/yellow]")
@@ -192,7 +208,12 @@ def get(ctx: Any, mcp_ref: str, export: str | None) -> None:
                     Text("[yellow]⚠️  Proceeding with available data[/yellow]")
                 )
 
-            export_resource_to_file(mcp, export_path, detected_format)
+            with spinner_context(
+                ctx,
+                "[bold blue]Exporting MCP configuration…[/bold blue]",
+                console_override=console,
+            ):
+                export_resource_to_file(mcp, export_path, detected_format)
             console.print(
                 Text(
                     f"[green]✅ Complete MCP configuration exported to: {export_path} (format: {detected_format})[/green]"
@@ -200,7 +221,12 @@ def get(ctx: Any, mcp_ref: str, export: str | None) -> None:
             )
 
         # Try to fetch raw API data first to preserve ALL fields
-        raw_mcp_data = fetch_raw_resource_details(client, mcp, "mcps")
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching detailed MCP data…[/bold blue]",
+            console_override=console,
+        ):
+            raw_mcp_data = fetch_raw_resource_details(client, mcp, "mcps")
 
         if raw_mcp_data:
             # Use raw API data - this preserves ALL fields
@@ -257,7 +283,12 @@ def list_tools(ctx: Any, mcp_ref: str) -> None:
         mcp = _resolve_mcp(ctx, client, mcp_ref)
 
         # Get tools from MCP
-        tools = client.mcps.get_mcp_tools(mcp.id)
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching MCP tools…[/bold blue]",
+            console_override=console,
+        ):
+            tools = client.mcps.get_mcp_tools(mcp.id)
 
         # Define table columns: (data_key, header, style, width)
         columns = [
@@ -311,7 +342,12 @@ def connect(ctx: Any, config_file: str) -> None:
             )
 
         # Test connection using config
-        result = client.mcps.test_mcp_connection_from_config(config)
+        with spinner_context(
+            ctx,
+            "[bold blue]Connecting to MCP…[/bold blue]",
+            console_override=console,
+        ):
+            result = client.mcps.test_mcp_connection_from_config(config)
 
         view = get_ctx_value(ctx, "view", "rich")
         if view == "json":
@@ -366,7 +402,12 @@ def update(
             raise click.ClickException("No update fields specified")
 
         # Update MCP (automatically chooses PUT or PATCH based on provided fields)
-        updated_mcp = client.mcps.update_mcp(mcp.id, **update_data)
+        with spinner_context(
+            ctx,
+            "[bold blue]Updating MCP…[/bold blue]",
+            console_override=console,
+        ):
+            updated_mcp = client.mcps.update_mcp(mcp.id, **update_data)
 
         handle_json_output(ctx, updated_mcp.model_dump())
         handle_rich_output(ctx, display_update_success("MCP", updated_mcp.name))
@@ -395,7 +436,12 @@ def delete(ctx: Any, mcp_ref: str, yes: bool) -> None:
         if not yes and not display_confirmation_prompt("MCP", mcp.name):
             return
 
-        client.mcps.delete_mcp(mcp.id)
+        with spinner_context(
+            ctx,
+            "[bold blue]Deleting MCP…[/bold blue]",
+            console_override=console,
+        ):
+            client.mcps.delete_mcp(mcp.id)
 
         handle_json_output(
             ctx,

glaip_sdk/cli/commands/models.py

@@ -9,7 +9,12 @@ from typing import Any
 import click
 from rich.console import Console
 
-from glaip_sdk.cli.utils import get_client, output_flags, output_list
+from glaip_sdk.cli.utils import (
+    get_client,
+    output_flags,
+    output_list,
+    spinner_context,
+)
 
 console = Console()
 
@@ -27,7 +32,12 @@ def list_models(ctx: Any) -> None:
     """List available language models."""
     try:
         client = get_client(ctx)
-        models = client.list_language_models()
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching language models…[/bold blue]",
+            console_override=console,
+        ):
+            models = client.list_language_models()
 
         # Define table columns: (data_key, header, style, width)
         columns = [

glaip_sdk/cli/commands/tools.py

@@ -40,6 +40,7 @@ from glaip_sdk.cli.utils import (
     output_flags,
     output_list,
     output_result,
+    spinner_context,
 )
 from glaip_sdk.utils import format_datetime
 from glaip_sdk.utils.import_export import merge_import_with_cli_args
@@ -193,7 +194,12 @@ def list_tools(ctx: Any, tool_type: str | None) -> None:
     """List all tools."""
     try:
         client = get_client(ctx)
-        tools = client.list_tools(tool_type=tool_type)
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching tools…[/bold blue]",
+            console_override=console,
+        ):
+            tools = client.list_tools(tool_type=tool_type)
 
         # Define table columns: (data_key, header, style, width)
         columns = [
@@ -276,7 +282,12 @@ def create(
         _validate_creation_parameters(file, import_file)
 
         # Create tool from file (either direct file or import file)
-        tool = _create_tool_from_file(client, file, name, description, tags)
+        with spinner_context(
+            ctx,
+            "[bold blue]Creating tool…[/bold blue]",
+            console_override=console,
+        ):
+            tool = _create_tool_from_file(client, file, name, description, tags)
 
         # Handle JSON output
         handle_json_output(ctx, tool.model_dump())
@@ -333,7 +344,12 @@ def get(ctx: Any, tool_ref: str, select: int | None, export: str | None) -> None
 
             # Always export comprehensive data - re-fetch tool with full details if needed
             try:
-                tool = client.get_tool_by_id(tool.id)
+                with spinner_context(
+                    ctx,
+                    "[bold blue]Fetching complete tool details…[/bold blue]",
+                    console_override=console,
+                ):
+                    tool = client.get_tool_by_id(tool.id)
             except Exception as e:
                 console.print(
                     Text(f"[yellow]⚠️  Could not fetch full tool details: {e}[/yellow]")
@@ -342,7 +358,12 @@ def get(ctx: Any, tool_ref: str, select: int | None, export: str | None) -> None
                     Text("[yellow]⚠️  Proceeding with available data[/yellow]")
                 )
 
-            export_resource_to_file(tool, export_path, detected_format)
+            with spinner_context(
+                ctx,
+                "[bold blue]Exporting tool configuration…[/bold blue]",
+                console_override=console,
+            ):
+                export_resource_to_file(tool, export_path, detected_format)
             console.print(
                 Text(
                     f"[green]✅ Complete tool configuration exported to: {export_path} (format: {detected_format})[/green]"
@@ -350,7 +371,12 @@ def get(ctx: Any, tool_ref: str, select: int | None, export: str | None) -> None
             )
 
         # Try to fetch raw API data first to preserve ALL fields
-        raw_tool_data = fetch_raw_resource_details(client, tool, "tools")
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching detailed tool data…[/bold blue]",
+            console_override=console,
+        ):
+            raw_tool_data = fetch_raw_resource_details(client, tool, "tools")
 
         if raw_tool_data:
             # Use raw API data - this preserves ALL fields
@@ -418,7 +444,12 @@ def update(
 
         # Get tool by ID (no ambiguity handling needed)
         try:
-            tool = client.get_tool_by_id(tool_id)
+            with spinner_context(
+                ctx,
+                "[bold blue]Fetching tool…[/bold blue]",
+                console_override=console,
+            ):
+                tool = client.get_tool_by_id(tool_id)
         except Exception as e:
             raise click.ClickException(f"Tool with ID '{tool_id}' not found: {e}")
 
@@ -435,9 +466,14 @@ def update(
                 raise click.ClickException(
                     f"File updates are only supported for custom tools. Tool '{tool.name}' is of type '{tool.tool_type}'."
                 )
-            updated_tool = client.tools.update_tool_via_file(
-                tool.id, file, framework=tool.framework
-            )
+            with spinner_context(
+                ctx,
+                "[bold blue]Uploading new tool code…[/bold blue]",
+                console_override=console,
+            ):
+                updated_tool = client.tools.update_tool_via_file(
+                    tool.id, file, framework=tool.framework
+                )
             handle_rich_output(
                 ctx, Text(f"[green]✓[/green] Tool code updated from {file}")
             )
@@ -447,7 +483,12 @@ def update(
                 raise click.ClickException(
                     f"Metadata updates are only supported for native tools. Tool '{tool.name}' is of type '{tool.tool_type}'."
                 )
-            updated_tool = tool.update(**update_data)
+            with spinner_context(
+                ctx,
+                "[bold blue]Updating tool metadata…[/bold blue]",
+                console_override=console,
+            ):
+                updated_tool = tool.update(**update_data)
             handle_rich_output(ctx, Text("[green]✓[/green] Tool metadata updated"))
         else:
             handle_rich_output(ctx, Text("[yellow]No updates specified[/yellow]"))
@@ -475,7 +516,12 @@ def delete(ctx: Any, tool_id: str, yes: bool) -> None:
 
         # Get tool by ID (no ambiguity handling needed)
         try:
-            tool = client.get_tool_by_id(tool_id)
+            with spinner_context(
+                ctx,
+                "[bold blue]Fetching tool…[/bold blue]",
+                console_override=console,
+            ):
+                tool = client.get_tool_by_id(tool_id)
         except Exception as e:
             raise click.ClickException(f"Tool with ID '{tool_id}' not found: {e}")
 
@@ -483,7 +529,12 @@ def delete(ctx: Any, tool_id: str, yes: bool) -> None:
         if not yes and not display_confirmation_prompt("Tool", tool.name):
             return
 
-        tool.delete()
+        with spinner_context(
+            ctx,
+            "[bold blue]Deleting tool…[/bold blue]",
+            console_override=console,
+        ):
+            tool.delete()
 
         handle_json_output(
             ctx,
@@ -509,7 +560,12 @@ def script(ctx: Any, tool_id: str) -> None:
     """Get tool script content."""
     try:
         client = get_client(ctx)
-        script_content = client.get_tool_script(tool_id)
+        with spinner_context(
+            ctx,
+            "[bold blue]Fetching tool script…[/bold blue]",
+            console_override=console,
+        ):
+            script_content = client.get_tool_script(tool_id)
 
         if get_ctx_value(ctx, "view") == "json":
             click.echo(json.dumps({"script": script_content}, indent=2))

glaip_sdk/cli/display.py

@@ -247,9 +247,10 @@ def display_agent_run_suggestions(agent: Any) -> Panel:
 
     return AIPPanel(
         f"[bold blue]💡 Next Steps:[/bold blue]\n\n"
-        f"🚀 Run this agent:\n"
-        f'   [green]aip agents run {agent.id} "Your message here"[/green]\n\n'
-        f"📋 Or use the agent name:\n"
+        f"🚀 Start chatting with [bold]{agent.name}[/bold] right here:\n"
+        f"   Type your message below and press Enter to run it immediately.\n\n"
+        f"📋 Prefer the CLI instead?\n"
+        f'   [green]aip agents run {agent.id} "Your message here"[/green]\n'
         f'   [green]aip agents run "{agent.name}" "Your message here"[/green]\n\n'
         f"🔧 Available options:\n"
         f"   [dim]--chat-history[/dim]  Include previous conversation\n"

glaip_sdk/cli/main.py

@@ -24,11 +24,21 @@ from glaip_sdk.cli.commands.configure import (
 from glaip_sdk.cli.commands.mcps import mcps_group
 from glaip_sdk.cli.commands.models import models_group
 from glaip_sdk.cli.commands.tools import tools_group
-from glaip_sdk.config.constants import DEFAULT_AGENT_RUN_TIMEOUT
+from glaip_sdk.cli.utils import spinner_context, update_spinner
+from glaip_sdk.config.constants import (
+    DEFAULT_AGENT_RUN_TIMEOUT,
+)
 from glaip_sdk.rich_components import AIPPanel, AIPTable
 
+# Import SlashSession for potential mocking in tests
+try:
+    from glaip_sdk.cli.slash import SlashSession
+except ImportError:  # pragma: no cover - optional slash dependencies
+    # Slash dependencies might not be available in all environments
+    SlashSession = None
+
 
-@click.group()
+@click.group(invoke_without_command=True)
 @click.version_option(version=_SDK_VERSION, prog_name="aip")
 @click.option("--api-url", envvar="AIP_API_URL", help="AIP API URL")
 @click.option("--api-key", envvar="AIP_API_KEY", help="AIP API Key")
@@ -72,6 +82,15 @@ def main(
 
     ctx.obj["tty"] = not no_tty
 
+    if ctx.invoked_subcommand is None and not ctx.resilient_parsing:
+        if _should_launch_slash(ctx) and SlashSession is not None:
+            session = SlashSession(ctx)
+            session.run()
+            ctx.exit()
+        else:
+            click.echo(ctx.get_help())
+            ctx.exit()
+
 
 # Add command groups
 main.add_command(agents_group)
@@ -87,6 +106,19 @@ main.add_command(configure_command)
 # Tip: `--version` is provided by click.version_option above.
 
 
+def _should_launch_slash(ctx: click.Context) -> bool:
+    """Determine whether to open the command palette automatically."""
+
+    ctx_obj = ctx.obj or {}
+    if not bool(ctx_obj.get("tty", True)):
+        return False
+
+    if not (sys.stdin.isatty() and sys.stdout.isatty()):
+        return False
+
+    return True
+
+
 @main.command()
 @click.pass_context
 def status(ctx: Any) -> None:
@@ -152,9 +184,26 @@ def status(ctx: Any) -> None:
 
         # Test connection by listing resources
         try:
-            agents = client.list_agents()
-            tools = client.list_tools()
-            mcps = client.list_mcps()
+            with spinner_context(
+                ctx,
+                "[bold blue]Checking GL AIP status…[/bold blue]",
+                console_override=console,
+                spinner_style="cyan",
+            ) as status_indicator:
+                update_spinner(
+                    status_indicator, "[bold blue]Fetching agents…[/bold blue]"
+                )
+                agents = client.list_agents()
+
+                update_spinner(
+                    status_indicator, "[bold blue]Fetching tools…[/bold blue]"
+                )
+                tools = client.list_tools()
+
+                update_spinner(
+                    status_indicator, "[bold blue]Fetching MCPs…[/bold blue]"
+                )
+                mcps = client.list_mcps()
 
             # Create status table
             table = AIPTable(title="🔗 GL AIP Status")