glaip-sdk 0.0.20__py3-none-any.whl → 0.7.7__py3-none-any.whl

glaip_sdk/cli/commands/agents/sync_langflow.py

@@ -0,0 +1,72 @@
+"""Sync LangFlow agents command.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import click
+
+from glaip_sdk.cli.context import get_ctx_value, output_flags
+from glaip_sdk.cli.display import handle_json_output, handle_rich_output
+from glaip_sdk.cli.core.context import get_client
+from glaip_sdk.cli.rich_helpers import markup_text
+
+from ._common import _handle_command_exception, agents_group
+
+from glaip_sdk.branding import SUCCESS_STYLE
+
+
+@agents_group.command()
+@click.option(
+    "--base-url",
+    help="Custom LangFlow server base URL (overrides LANGFLOW_BASE_URL env var)",
+)
+@click.option("--api-key", help="Custom LangFlow API key (overrides LANGFLOW_API_KEY env var)")
+@output_flags()
+@click.pass_context
+def sync_langflow(ctx: Any, base_url: str | None, api_key: str | None) -> None:
+    r"""Sync agents with LangFlow server flows.
+
+    This command fetches all flows from the configured LangFlow server and
+    creates/updates corresponding agents in the platform.
+
+    The LangFlow server configuration can be provided via:
+    - Command options (--base-url, --api-key)
+    - Environment variables (LANGFLOW_BASE_URL, LANGFLOW_API_KEY)
+
+    \b
+    Examples:
+        aip agents sync-langflow
+        aip agents sync-langflow --base-url https://my-langflow.com --api-key my-key
+    """
+    try:
+        client = get_client(ctx)
+
+        # Perform the sync
+        result = client.sync_langflow_agents(base_url=base_url, api_key=api_key)
+
+        # Handle output format
+        handle_json_output(ctx, result)
+
+        # Show success message for non-JSON output
+        if get_ctx_value(ctx, "view") != "json":
+            # Extract some useful info from the result
+            success_count = result.get("data", {}).get("created_count", 0) + result.get("data", {}).get(
+                "updated_count", 0
+            )
+            total_count = result.get("data", {}).get("total_processed", 0)
+
+            handle_rich_output(
+                ctx,
+                markup_text(
+                    f"[{SUCCESS_STYLE}]✅ Successfully synced {success_count} LangFlow agents "
+                    f"({total_count} total processed)[/]"
+                ),
+            )
+
+    except Exception as e:
+        _handle_command_exception(ctx, e)

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

@@ -0,0 +1,104 @@
+"""Shared helpers for configuration/account flows."""
+
+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.branding import PRIMARY, SUCCESS_STYLE, WARNING_STYLE, AIPBranding
+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:
+    """Render the standard CLI branding header with a custom rule text."""
+    branding = AIPBranding.create_from_sdk(sdk_version=sdk_version(), package_name="glaip-sdk")
+    heading = "[bold]>_ GDP Labs AI Agents Package (AIP CLI)[/bold]"
+    console.print(heading)
+    console.print()
+    console.print(branding.get_welcome_banner())
+    console.rule(rule_text, style=PRIMARY)
+
+
+def check_connection(
+    api_url: str,
+    api_key: str,
+    console: Console,
+    *,
+    abort_on_error: bool = False,
+    extra_hint: str | None = None,
+) -> bool:
+    """Test connectivity and report results.
+
+    Returns True on success, False on handled failures. Raises click.Abort when
+    abort_on_error is True and a fatal error occurs.
+    """
+    console.print("\n🔌 Testing connection...")
+    client: Client | None = None
+    try:
+        # Import lazily to avoid pulling in SDK dependencies during CLI startup.
+        from glaip_sdk import Client  # noqa: PLC0415
+
+        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))
+            return True
+        except Exception as exc:  # pragma: no cover - API failures depend on network
+            console.print(Text(f"⚠️  Connection established but API call failed: {exc}", style=WARNING_STYLE))
+            console.print("   You may need to check your API permissions or network access")
+            if extra_hint:
+                console.print(extra_hint)
+            if abort_on_error:
+                raise click.Abort() from exc
+            return False
+    except Exception as exc:
+        console.print(Text(f"❌ Connection failed: {exc}"))
+        console.print("   Please check your API URL and key")
+        if extra_hint:
+            console.print(extra_hint)
+        if abort_on_error:
+            raise click.Abort() from exc
+        return False
+    finally:
+        if client is not None:
+            client.close()
+
+
+def check_connection_with_reason(
+    api_url: str,
+    api_key: str,
+    *,
+    abort_on_error: bool = False,
+) -> tuple[bool, str]:
+    """Test connectivity and return structured reason."""
+    client: Client | None = None
+    try:
+        # Import lazily to avoid pulling in SDK dependencies during CLI startup.
+        from glaip_sdk import Client  # noqa: PLC0415
+
+        client = Client(api_url=api_url, api_key=api_key)
+        try:
+            client.list_agents()
+            return True, ""
+        except Exception as exc:  # pragma: no cover - API failures depend on network
+            if abort_on_error:
+                raise click.Abort() from exc
+            return False, f"api_failed: {exc}"
+    except Exception as exc:
+        # Log unexpected exceptions in debug while keeping CLI-friendly messaging
+        logging.getLogger(__name__).debug("Unexpected connection error", exc_info=exc)
+        if abort_on_error:
+            raise click.Abort() from exc
+        return False, f"connection_failed: {exc}"
+    finally:
+        if client is not None:
+            try:
+                client.close()
+            except Exception:
+                pass