glaip-sdk 0.6.5b6__py3-none-any.whl → 0.7.12__py3-none-any.whl

glaip_sdk/cli/commands/agents/update.py

@@ -0,0 +1,112 @@
+"""Update agent command.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import click
+
+from glaip_sdk.cli.context import output_flags
+from glaip_sdk.cli.display import (
+    display_agent_run_suggestions,
+    display_update_success,
+    handle_json_output,
+    handle_rich_output,
+)
+from glaip_sdk.cli.core.context import get_client
+
+from ._common import (
+    _get_agent_for_update,
+    _handle_click_exception_for_json,
+    _handle_command_exception,
+    _prepare_agent_output,
+    _resolve_resources_by_name,
+    _split_comma_separated_refs,
+    agents_group,
+)
+
+
+@agents_group.command()
+@click.argument("agent_id")
+@click.option("--name", help="New agent name")
+@click.option("--instruction", help="New instruction")
+@click.option("--tools", multiple=True, help="New tool names or IDs")
+@click.option("--agents", multiple=True, help="New sub-agent names")
+@click.option("--mcps", multiple=True, help="New MCP names or IDs")
+@click.option("--timeout", type=int, help="New timeout value")
+@click.option(
+    "--import",
+    "import_file",
+    type=click.Path(exists=True, dir_okay=False),
+    help="Import agent configuration from JSON file",
+)
+@output_flags()
+@click.pass_context
+def update(
+    ctx: Any,
+    agent_id: str,
+    name: str | None,
+    instruction: str | None,
+    tools: tuple[str, ...] | None,
+    agents: tuple[str, ...] | None,
+    mcps: tuple[str, ...] | None,
+    timeout: float | None,
+    import_file: str | None,
+) -> None:
+    r"""Update an existing agent.
+
+    \b
+    Examples:
+        aip agents update my-agent --instruction "New instruction"
+        aip agents update my-agent --import agent.json
+    """
+    try:
+        client = get_client(ctx)
+        tools = _split_comma_separated_refs(tools)
+        agents = _split_comma_separated_refs(agents)
+        mcps = _split_comma_separated_refs(mcps)
+
+        has_updates = bool(import_file) or any(
+            [
+                name is not None,
+                instruction is not None,
+                bool(tools),
+                bool(agents),
+                bool(mcps),
+                timeout is not None,
+            ]
+        )
+        if not has_updates:
+            raise click.ClickException("No update fields specified")
+
+        agent = _get_agent_for_update(client, agent_id)
+
+        resolved_tools = _resolve_resources_by_name(client, tools, "tool", client.find_tools, "Tool") if tools else None
+        resolved_agents = (
+            _resolve_resources_by_name(client, agents, "agent", client.find_agents, "Agent") if agents else None
+        )
+        resolved_mcps = _resolve_resources_by_name(client, mcps, "mcp", client.find_mcps, "MCP") if mcps else None
+
+        updated_agent = client.agents.update_agent(
+            agent.id,
+            file=import_file,
+            name=name,
+            instruction=instruction,
+            tools=resolved_tools,
+            agents=resolved_agents,
+            mcps=resolved_mcps,
+            timeout=timeout,
+        )
+
+        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))
+
+    except click.ClickException as e:
+        _handle_click_exception_for_json(ctx, e)
+    except Exception as e:
+        _handle_command_exception(ctx, e)

glaip_sdk/cli/commands/common_config.py

@@ -1,13 +1,18 @@
 """Shared helpers for configuration/account flows."""
 
-import click
+from __future__ import annotations
+
 import logging
+from typing import TYPE_CHECKING
+
+import click
 from rich.console import Console
 from rich.text import Text
-
-from glaip_sdk import Client
 from glaip_sdk.branding import PRIMARY, SUCCESS_STYLE, WARNING_STYLE, AIPBranding
-from glaip_sdk.cli.utils import sdk_version
+from glaip_sdk.cli.core.output import sdk_version
+
+if TYPE_CHECKING:  # pragma: no cover - type checking only
+    from glaip_sdk import Client
 
 
 def render_branding_header(console: Console, rule_text: str) -> None:
@@ -36,11 +41,10 @@ def check_connection(
     console.print("\n🔌 Testing connection...")
     client: Client | None = None
     try:
-        # Import lazily so test patches targeting glaip_sdk.Client are honored
-        from importlib import import_module  # noqa: PLC0415
+        # Import lazily to avoid pulling in SDK dependencies during CLI startup.
+        from glaip_sdk import Client  # noqa: PLC0415
 
-        client_module = import_module("glaip_sdk")
-        client = client_module.Client(api_url=api_url, api_key=api_key)
+        client = Client(api_url=api_url, api_key=api_key)
         try:
             agents = client.list_agents()
             console.print(Text(f"✅ Connection successful! Found {len(agents)} agents", style=SUCCESS_STYLE))
@@ -75,11 +79,10 @@ def check_connection_with_reason(
     """Test connectivity and return structured reason."""
     client: Client | None = None
     try:
-        # Import lazily so test patches targeting glaip_sdk.Client are honored
-        from importlib import import_module  # noqa: PLC0415
+        # Import lazily to avoid pulling in SDK dependencies during CLI startup.
+        from glaip_sdk import Client  # noqa: PLC0415
 
-        client_module = import_module("glaip_sdk")
-        client = client_module.Client(api_url=api_url, api_key=api_key)
+        client = Client(api_url=api_url, api_key=api_key)
         try:
             client.list_agents()
             return True, ""

glaip_sdk/cli/commands/configure.py

@@ -22,15 +22,14 @@ from glaip_sdk.branding import ACCENT_STYLE, ERROR_STYLE, INFO, NEUTRAL, SUCCESS
 # Optional import for gitignore support; warn when missing to avoid silent expansion
 try:
     import pathspec  # type: ignore[import-untyped]  # noqa: PLC0415
-except ImportError:
+except ImportError:  # pragma: no cover - optional dependency
     pathspec = None  # type: ignore[assignment]
 from glaip_sdk.cli.account_store import get_account_store
 from glaip_sdk.cli.commands.common_config import check_connection, render_branding_header
 from glaip_sdk.cli.config import CONFIG_FILE, load_config, save_config
-from glaip_sdk.cli.hints import format_command_hint
+from glaip_sdk.cli.hints import command_hint, format_command_hint
 from glaip_sdk.cli.masking import mask_api_key_display
 from glaip_sdk.cli.rich_helpers import markup_text
-from glaip_sdk.cli.utils import command_hint
 from glaip_sdk.rich_components import AIPTable
 
 console = Console()

glaip_sdk/cli/commands/mcps/__init__.py

@@ -0,0 +1,94 @@
+"""MCP CLI commands package.
+
+This package contains MCP management commands split by operation.
+The package is the canonical import surface.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+# pylint: disable=duplicate-code
+from glaip_sdk.cli.commands.mcps._common import (  # noqa: E402
+    mcps_group,
+    _resolve_mcp,
+    console,
+    _load_import_ready_payload,
+    _merge_import_payload,
+    _strip_server_only_fields,
+    _coerce_cli_string,
+    _collect_cli_overrides,
+    _merge_config_field,
+    _merge_auth_field,
+    _get_config_transport,
+    _generate_update_preview,
+)
+from glaip_sdk.cli.commands.mcps.connect import connect  # noqa: E402
+from glaip_sdk.cli.commands.mcps.create import create  # noqa: E402
+from glaip_sdk.cli.commands.mcps.delete import delete  # noqa: E402
+from glaip_sdk.cli.commands.mcps.get import get  # noqa: E402
+from glaip_sdk.cli.commands.mcps.list import list_mcps  # noqa: E402
+from glaip_sdk.cli.commands.mcps.tools import list_tools, _get_tools_from_config  # noqa: E402
+from glaip_sdk.cli.commands.mcps.update import update  # noqa: E402
+
+# Import core functions for test compatibility
+from glaip_sdk.cli.core.context import get_client  # noqa: E402
+from glaip_sdk.cli.core.output import (  # noqa: E402
+    output_list,
+    output_result,
+)
+from glaip_sdk.cli.core.rendering import (  # noqa: E402
+    spinner_context,
+    with_client_and_spinner,
+)
+
+# Import validators for test compatibility
+from glaip_sdk.cli.mcp_validators import (  # noqa: E402
+    validate_mcp_auth_structure,
+    validate_mcp_config_structure,
+)
+
+# Import import/export utilities for test compatibility
+from glaip_sdk.utils.import_export import convert_export_to_import_format  # noqa: E402
+from glaip_sdk.cli.io import load_resource_from_file_with_validation  # noqa: E402
+
+# Import shared formatters for test compatibility
+from glaip_sdk.cli.commands.shared.formatters import (  # noqa: E402
+    _format_dict_value,
+    _redact_sensitive_dict,
+    _is_sensitive_data,
+)
+
+__all__ = [
+    "mcps_group",
+    "connect",
+    "create",
+    "delete",
+    "get",
+    "list_mcps",
+    "list_tools",
+    "update",
+    "_resolve_mcp",
+    "console",
+    "_load_import_ready_payload",
+    "_merge_import_payload",
+    "_strip_server_only_fields",
+    "_coerce_cli_string",
+    "_collect_cli_overrides",
+    "_merge_config_field",
+    "_merge_auth_field",
+    "_get_config_transport",
+    "_generate_update_preview",
+    "_get_tools_from_config",
+    "_format_dict_value",
+    "_redact_sensitive_dict",
+    "_is_sensitive_data",
+    "get_client",
+    "output_list",
+    "output_result",
+    "spinner_context",
+    "with_client_and_spinner",
+    "validate_mcp_auth_structure",
+    "validate_mcp_config_structure",
+    "convert_export_to_import_format",
+    "load_resource_from_file_with_validation",
+]