glaip-sdk 0.6.24__py3-none-any.whl → 0.6.26__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/mcps/__init__.py

@@ -0,0 +1,98 @@
+"""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,
+    _assemble_update_data_from_import_payload,
+    _assemble_update_data_from_cli_options,
+    _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",
+    "_assemble_update_data_from_import_payload",
+    "_assemble_update_data_from_cli_options",
+    "_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",
+]