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

glaip_sdk/__init__.py

@@ -4,12 +4,49 @@ Authors:
     Raymond Christopher (raymond.christopher@gdplabs.id)
 """
 
+from __future__ import annotations
+
+import importlib
+from typing import TYPE_CHECKING, Any
+
 from glaip_sdk._version import __version__
-from glaip_sdk.client import Client
-from glaip_sdk.exceptions import AIPError
-from glaip_sdk.agents import Agent
-from glaip_sdk.tools import Tool
-from glaip_sdk.mcps import MCP
 
+if TYPE_CHECKING:  # pragma: no cover - import only for type checking
+    from glaip_sdk.agents import Agent
+    from glaip_sdk.client import Client
+    from glaip_sdk.exceptions import AIPError
+    from glaip_sdk.mcps import MCP
+    from glaip_sdk.tools import Tool
 
 __all__ = ["Client", "Agent", "Tool", "MCP", "AIPError", "__version__"]
+
+_LAZY_IMPORTS: dict[str, tuple[str, str]] = {
+    "Client": ("glaip_sdk.client", "Client"),
+    "Agent": ("glaip_sdk.agents", "Agent"),
+    "Tool": ("glaip_sdk.tools", "Tool"),
+    "MCP": ("glaip_sdk.mcps", "MCP"),
+    "AIPError": ("glaip_sdk.exceptions", "AIPError"),
+}
+
+
+def __getattr__(name: str) -> Any:
+    """Lazy attribute access for public SDK symbols to defer heavy imports."""
+    if name == "__version__":
+        # Import __version__ when accessed via __getattr__
+        # This ensures coverage even if __version__ was removed from __dict__ for testing
+        from glaip_sdk._version import __version__ as version  # noqa: PLC0415
+
+        globals()["__version__"] = version
+        return version
+    if name in _LAZY_IMPORTS:
+        module_path, attr_name = _LAZY_IMPORTS[name]
+        module = importlib.import_module(module_path)
+        attr = getattr(module, attr_name)
+        globals()[name] = attr
+        return attr
+    raise AttributeError(f"module 'glaip_sdk' has no attribute {name!r}")
+
+
+def __dir__() -> list[str]:
+    """Return module attributes for dir()."""
+    return sorted(__all__)

glaip_sdk/agents/base.py

@@ -50,16 +50,10 @@ from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
 from glaip_sdk.registry import get_agent_registry, get_mcp_registry, get_tool_registry
-from glaip_sdk.runner import get_default_runner
-from glaip_sdk.runner.deps import (
-    check_local_runtime_available,
-    get_local_runtime_missing_message,
-)
-from glaip_sdk.utils.discovery import find_agent
 from glaip_sdk.utils.resource_refs import is_uuid
-from glaip_sdk.utils.runtime_config import normalize_runtime_config_keys
 
 if TYPE_CHECKING:
+    from glaip_sdk.client.schedules import AgentScheduleManager
     from glaip_sdk.models import AgentResponse
     from glaip_sdk.registry import AgentRegistry, MCPRegistry, ToolRegistry
 
@@ -518,6 +512,8 @@ class Agent:
         from glaip_sdk.utils.client import get_client  # noqa: PLC0415
 
         client = get_client()
+        from glaip_sdk.utils.discovery import find_agent  # noqa: PLC0415
+
         response = self._create_or_update_agent(config, client, find_agent)
 
         # Update self with deployed info
@@ -847,6 +843,36 @@ class Agent:
         self._client = client
         return self
 
+    @property
+    def schedule(self) -> AgentScheduleManager:
+        """Get the schedule manager for this agent.
+
+        Provides a convenient interface for managing schedules scoped to this agent.
+
+        Returns:
+            AgentScheduleManager for schedule operations
+
+        Raises:
+            ValueError: If agent is not deployed
+            RuntimeError: If agent is not bound to a client
+
+        Example:
+            >>> agent = client.get_agent_by_id("agent-id")
+            >>> schedules = agent.schedule.list()
+            >>> new_schedule = agent.schedule.create(
+            ...     input="Daily task",
+            ...     schedule="0 9 * * 1-5"
+            ... )
+        """
+        if not self.id:
+            raise ValueError(_AGENT_NOT_DEPLOYED_MSG)
+        if not self._client:
+            raise RuntimeError(_CLIENT_NOT_AVAILABLE_MSG)
+
+        from glaip_sdk.client.schedules import AgentScheduleManager  # noqa: PLC0415
+
+        return AgentScheduleManager(self, self._client.schedules)
+
     def _prepare_run_kwargs(
         self,
         message: str,
@@ -869,7 +895,7 @@ class Agent:
             ValueError: If the agent hasn't been deployed yet.
             RuntimeError: If client is not available.
         """
-        if not self.id:
+        if not self.id:  # pragma: no cover - defensive: called only when self.id is truthy
             raise ValueError(_AGENT_NOT_DEPLOYED_MSG)
         if not self._client:
             raise RuntimeError(_CLIENT_NOT_AVAILABLE_MSG)
@@ -883,6 +909,10 @@ class Agent:
         }
 
         if runtime_config is not None:
+            from glaip_sdk.utils.runtime_config import (  # noqa: PLC0415
+                normalize_runtime_config_keys,
+            )
+
             call_kwargs["runtime_config"] = normalize_runtime_config_keys(
                 runtime_config,
                 tool_registry=get_tool_registry(),
@@ -893,10 +923,68 @@ class Agent:
         call_kwargs.update(kwargs)
         return agent_client, call_kwargs
 
+    def _get_local_runner_or_raise(self) -> Any:
+        """Get the local runner if available, otherwise raise ValueError.
+
+        Returns:
+            The default local runner instance.
+
+        Raises:
+            ValueError: If local runtime is not available.
+        """
+        from glaip_sdk.runner import get_default_runner  # noqa: PLC0415
+        from glaip_sdk.runner.deps import (  # noqa: PLC0415
+            check_local_runtime_available,
+            get_local_runtime_missing_message,
+        )
+
+        if check_local_runtime_available():
+            return get_default_runner()
+
+        # If agent is not deployed, it *must* use local runtime
+        if not self.id:
+            raise ValueError(f"{_AGENT_NOT_DEPLOYED_MSG}\n\n{get_local_runtime_missing_message()}")
+
+        # If agent IS deployed but local execution was forced (local=True)
+        raise ValueError(
+            f"Local execution override was requested, but local runtime is missing.\n\n"
+            f"{get_local_runtime_missing_message()}"
+        )
+
+    def _prepare_local_runner_kwargs(
+        self,
+        message: str,
+        verbose: bool,
+        runtime_config: dict[str, Any] | None,
+        chat_history: list[dict[str, str]] | None,
+        **kwargs: Any,
+    ) -> dict[str, Any]:
+        """Prepare kwargs for local runner execution.
+
+        Args:
+            message: The message to send to the agent.
+            verbose: If True, print streaming output to console.
+            runtime_config: Optional runtime configuration.
+            chat_history: Optional list of prior conversation messages.
+            **kwargs: Additional arguments.
+
+        Returns:
+            Dictionary of prepared kwargs for runner.run() or runner.arun().
+        """
+        return {
+            "agent": self,
+            "message": message,
+            "verbose": verbose,
+            "runtime_config": runtime_config,
+            "chat_history": chat_history,
+            **kwargs,
+        }
+
     def run(
         self,
         message: str,
         verbose: bool = False,
+        local: bool = False,
         runtime_config: dict[str, Any] | None = None,
         chat_history: list[dict[str, str]] | None = None,
         **kwargs: Any,
@@ -909,9 +997,13 @@ class Agent:
         - **Local**: When the agent is not deployed and glaip-sdk[local] is installed,
           execution happens locally via aip-agents (no server required).
 
+        You can force local execution for a deployed agent by passing `local=True`.
+
         Args:
             message: The message to send to the agent.
             verbose: If True, print streaming output to console. Defaults to False.
+            local: If True, force local execution even if the agent is deployed.
+                Defaults to False.
             runtime_config: Optional runtime configuration for tools, MCPs, and agents.
                 Keys can be SDK objects, UUIDs, or names. Example:
                 {
@@ -933,42 +1025,47 @@ class Agent:
             RuntimeError: If server-backed execution fails due to client issues.
         """
         # Backend routing: deployed agents use server, undeployed use local (if available)
-        if self.id:
+        if self.id and not local:
             # Server-backed execution path (agent is deployed)
             agent_client, call_kwargs = self._prepare_run_kwargs(
-                message, verbose, runtime_config or kwargs.get("runtime_config"), **kwargs
+                message,
+                verbose,
+                runtime_config or kwargs.get("runtime_config"),
+                **kwargs,
             )
             if chat_history is not None:
                 call_kwargs["chat_history"] = chat_history
             return agent_client.run_agent(**call_kwargs)
 
-        # Local execution path (agent is not deployed)
-        if check_local_runtime_available():
-            runner = get_default_runner()
-            return runner.run(
-                agent=self,
-                message=message,
-                verbose=verbose,
-                runtime_config=runtime_config,
-                chat_history=chat_history,
-                **kwargs,
-            )
-
-        # Neither deployed nor local runtime available - provide actionable error
-        raise ValueError(f"{_AGENT_NOT_DEPLOYED_MSG}\n\n{get_local_runtime_missing_message()}")
+        # Local execution path (agent is not deployed OR local=True)
+        runner = self._get_local_runner_or_raise()
+        local_kwargs = self._prepare_local_runner_kwargs(message, verbose, runtime_config, chat_history, **kwargs)
+        return runner.run(**local_kwargs)
 
     async def arun(
         self,
         message: str,
         verbose: bool = False,
+        local: bool = False,
         runtime_config: dict[str, Any] | None = None,
+        chat_history: list[dict[str, str]] | None = None,
         **kwargs: Any,
     ) -> AsyncGenerator[dict, None]:
         """Run the agent asynchronously with streaming output.
 
+        Supports two execution modes:
+        - **Server-backed**: When the agent is deployed (has an ID), execution
+          happens via the AIP backend server with streaming.
+        - **Local**: When the agent is not deployed and glaip-sdk[local] is installed,
+          execution happens locally via aip-agents (no server required).
+
+        You can force local execution for a deployed agent by passing `local=True`.
+
         Args:
             message: The message to send to the agent.
-            verbose: If True, print streaming output to console.
+            verbose: If True, print streaming output to console. Defaults to False.
+            local: If True, force local execution even if the agent is deployed.
+                Defaults to False.
             runtime_config: Optional runtime configuration for tools, MCPs, and agents.
                 Keys can be SDK objects, UUIDs, or names. Example:
                 {
@@ -976,20 +1073,47 @@ class Agent:
                     "mcp_configs": {"mcp-id": {"setting": "on"}},
                     "agent_config": {"planning": True},
                 }
+                Defaults to None.
+            chat_history: Optional list of prior conversation messages for context.
+                Each message is a dict with "role" and "content" keys.
+                Defaults to None.
             **kwargs: Additional arguments to pass to the run API.
 
         Yields:
             Streaming response chunks from the agent.
 
         Raises:
-            ValueError: If the agent hasn't been deployed yet.
-            RuntimeError: If client is not available.
+            ValueError: If the agent is not deployed and local runtime is not available.
+            RuntimeError: If server-backed execution fails due to client issues.
         """
-        agent_client, call_kwargs = self._prepare_run_kwargs(
-            message, verbose, runtime_config or kwargs.get("runtime_config"), **kwargs
-        )
-        async for chunk in agent_client.arun_agent(**call_kwargs):
-            yield chunk
+        # Backend routing: deployed agents use server, undeployed use local (if available)
+        if self.id and not local:
+            # Server-backed execution path (agent is deployed)
+            agent_client, call_kwargs = self._prepare_run_kwargs(
+                message,
+                verbose,
+                runtime_config or kwargs.get("runtime_config"),
+                **kwargs,
+            )
+            if chat_history is not None:
+                call_kwargs["chat_history"] = chat_history
+
+            async for chunk in agent_client.arun_agent(**call_kwargs):
+                yield chunk
+            return
+
+        # Local execution path (agent is not deployed OR local=True)
+        runner = self._get_local_runner_or_raise()
+        local_kwargs = self._prepare_local_runner_kwargs(message, verbose, runtime_config, chat_history, **kwargs)
+        result = await runner.arun(**local_kwargs)
+        # Yield a final_response event for consistency with server-backed execution
+        # Include event_type for A2A event shape parity
+        yield {
+            "event_type": "final_response",
+            "metadata": {"kind": "final_response"},
+            "content": result,
+            "is_final": True,
+        }
 
     def update(self, **kwargs: Any) -> Agent:
         """Update the deployed agent with new configuration.

glaip_sdk/cli/auth.py

@@ -19,8 +19,7 @@ from rich.console import Console
 
 from glaip_sdk.branding import HINT_PREFIX_STYLE, WARNING_STYLE
 from glaip_sdk.cli.account_store import AccountNotFoundError, AccountStoreError, get_account_store
-from glaip_sdk.cli.hints import format_command_hint
-from glaip_sdk.cli.utils import command_hint
+from glaip_sdk.cli.hints import command_hint, format_command_hint
 
 
 def prepare_authentication_export(
@@ -526,12 +525,19 @@ def resolve_api_url_from_context(
     elif hasattr(ctx, "obj") and isinstance(ctx.obj, dict):
         account_name = ctx.obj.get("account_name")
 
-    resolved_url, _, _ = resolve_credentials(
-        account_name=account_name,
-        api_url=api_url,
-        api_key=None,
-        ignore_env_creds=True,
-    )
+    if isinstance(api_url, str) and api_url.strip():
+        return api_url.strip()
+
+    try:
+        resolved_url, _, _ = resolve_credentials(
+            account_name=account_name,
+            api_url=None,
+            api_key=None,
+            ignore_env_creds=True,
+        )
+    except Exception:
+        return None
+
     return resolved_url
 
 

glaip_sdk/cli/commands/accounts.py

@@ -33,7 +33,7 @@ from glaip_sdk.cli.commands.common_config import check_connection, render_brandi
 from glaip_sdk.cli.hints import format_command_hint
 from glaip_sdk.cli.masking import mask_api_key_display
 from glaip_sdk.cli.slash.accounts_shared import env_credentials_present
-from glaip_sdk.cli.utils import command_hint
+from glaip_sdk.cli.hints import command_hint
 from glaip_sdk.icons import ICON_TOOL
 from glaip_sdk.rich_components import AIPPanel, AIPTable
 

glaip_sdk/cli/commands/agents/__init__.py

@@ -0,0 +1,119 @@
+"""Agent CLI commands package.
+
+This package contains agent management commands split by operation.
+The package is the canonical import surface.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+# pylint: disable=duplicate-code
+# Import from submodules
+from glaip_sdk.cli.commands.agents._common import (  # noqa: E402
+    AGENT_NOT_FOUND_ERROR,
+    _get_agent_for_update,
+    _resolve_agent,
+    agents_group,
+    _coerce_mapping_candidate,
+    _display_agent_details,
+    _emit_verbose_guidance,
+    _fetch_full_agent_details,
+    _get_agent_model_name,
+    _get_language_model_display_name,
+    _model_from_config,
+    _prepare_agent_output,
+    _resolve_resources_by_name,
+    console,
+)
+from glaip_sdk.cli.commands.agents.create import create  # noqa: E402
+from glaip_sdk.cli.commands.agents.delete import delete  # noqa: E402
+from glaip_sdk.cli.commands.agents.get import get  # noqa: E402
+from glaip_sdk.cli.commands.agents.list import list_agents  # noqa: E402
+from glaip_sdk.cli.commands.agents.run import run, _maybe_attach_transcript_toggle  # noqa: E402
+from glaip_sdk.cli.commands.agents.sync_langflow import sync_langflow  # noqa: E402
+from glaip_sdk.cli.commands.agents.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.rendering import with_client_and_spinner  # noqa: E402
+
+# Import display functions for test compatibility
+from glaip_sdk.cli.display import (  # noqa: E402
+    handle_json_output,
+    handle_rich_output,
+)
+
+# Import core output functions for test compatibility
+from glaip_sdk.cli.core.output import (  # noqa: E402
+    handle_resource_export,
+    output_list,
+)
+
+# Import rendering functions for test compatibility
+from glaip_sdk.cli.core.rendering import (  # noqa: E402
+    build_renderer,
+)
+
+# Import display functions for test compatibility
+from glaip_sdk.cli.display import (  # noqa: E402
+    display_agent_run_suggestions,
+)
+
+# Import rich helpers for test compatibility
+from glaip_sdk.cli.rich_helpers import (  # noqa: E402
+    markup_text,
+)
+
+# Import transcript functions for test compatibility
+from glaip_sdk.cli.transcript import (  # noqa: E402
+    maybe_launch_post_run_viewer,
+    store_transcript_for_session,
+)
+
+# Import IO functions for test compatibility
+from glaip_sdk.cli.io import (  # noqa: E402
+    fetch_raw_resource_details,
+)
+
+# Import utils for test compatibility
+from glaip_sdk.utils import (  # noqa: E402
+    is_uuid,
+)
+
+__all__ = [
+    "AGENT_NOT_FOUND_ERROR",
+    "agents_group",
+    "create",
+    "delete",
+    "get",
+    "list_agents",
+    "run",
+    "sync_langflow",
+    "update",
+    "_get_agent_for_update",
+    "_resolve_agent",
+    "_coerce_mapping_candidate",
+    "_display_agent_details",
+    "_emit_verbose_guidance",
+    "_fetch_full_agent_details",
+    "_get_agent_model_name",
+    "_get_language_model_display_name",
+    "_model_from_config",
+    "_prepare_agent_output",
+    "_resolve_resources_by_name",
+    "_maybe_attach_transcript_toggle",
+    "get_client",
+    "with_client_and_spinner",
+    "console",
+    "handle_json_output",
+    "handle_rich_output",
+    "output_list",
+    "handle_resource_export",
+    "build_renderer",
+    "display_agent_run_suggestions",
+    "markup_text",
+    "maybe_launch_post_run_viewer",
+    "store_transcript_for_session",
+    "fetch_raw_resource_details",
+    "is_uuid",
+]