gatewaysdk 0.3.2__py3-none-any.whl

gatewaysdk/cli/agent_runner.py

@@ -0,0 +1,132 @@
+# Copyright (c) Gateway. All rights reserved.
+
+"""gateway-agent CLI — runs a LitAgent inside a platform container.
+
+This eliminates the need for hand-written entrypoint.py files in agent
+containers. The orchestrator sets ``GATEWAY_STORE_URL`` and
+``GATEWAY_RUN_ID``; this command handles everything else.
+
+Usage::
+
+    # In Dockerfile:
+    CMD ["gateway-agent", "--agent", "crm_agent:crm_agent"]
+
+    # Or manually:
+    GATEWAY_STORE_URL=http://host:9876 GATEWAY_RUN_ID=run-abc \
+        gateway-agent --agent my_module:my_agent
+"""
+
+from __future__ import annotations
+
+import asyncio
+import importlib
+import logging
+import sys
+from typing import Optional
+
+
+def _resolve_agent(agent_ref: str):
+    """Resolve 'module:attribute' to a LitAgent instance.
+
+    Args:
+        agent_ref: Reference like 'crm_agent:crm_agent' (module:attr).
+
+    Returns:
+        The resolved LitAgent instance.
+
+    Raises:
+        ValueError: If the reference format is invalid.
+        ImportError: If the module can't be imported.
+        AttributeError: If the attribute doesn't exist in the module.
+    """
+    if ":" not in agent_ref:
+        raise ValueError(f"Agent reference must be in 'module:attribute' format, got: '{agent_ref}'")
+
+    module_path, attr_name = agent_ref.rsplit(":", 1)
+    module = importlib.import_module(module_path)
+    agent = getattr(module, attr_name)
+    return agent
+
+
+async def _run_agent(agent_ref: str, worker_id: int = 0) -> None:
+    """Initialize store, tracer, runner, and execute rollouts.
+
+    Args:
+        agent_ref: Module:attribute reference to the LitAgent.
+        worker_id: Worker ID for this runner instance.
+    """
+    logger = logging.getLogger(__name__)
+
+    # 1. Resolve agent
+    logger.info(f"Resolving agent: {agent_ref}")
+    agent = _resolve_agent(agent_ref)
+    logger.info(f"Agent resolved: {type(agent).__name__}")
+
+    # 2. Create store client (connects to worker's StoreServer via HTTP)
+    from gatewaysdk.store.gateway import create_platform_store_client_from_env
+
+    store = create_platform_store_client_from_env()
+    logger.info(f"Store client created")
+
+    # 3. Create tracer
+    from gatewaysdk.tracer.otel import OtelTracer
+
+    tracer = OtelTracer()
+
+    # 4. Create and initialize runner
+    from gatewaysdk.runner import LitAgentRunner
+
+    runner = LitAgentRunner(tracer=tracer)
+    runner.init(agent)
+    runner.init_worker(worker_id=worker_id, store=store)
+    logger.info(f"Runner initialized (worker_id={worker_id})")
+
+    # 5. Poll and execute rollouts
+    await runner.iter()
+
+
+def main() -> None:
+    """CLI entry point for ``gateway-agent``."""
+    try:
+        import typer
+    except ImportError:
+        print(
+            "typer is required for gateway-agent. " "Install with: pip install gatewaysdk[agent]",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s %(levelname)-8s [%(name)s] %(message)s",
+    )
+
+    app = typer.Typer(
+        name="gateway-agent",
+        help="Gateway agent runner — executes rollouts inside a platform container.",
+        no_args_is_help=True,
+    )
+
+    @app.command()
+    def run(
+        agent: str = typer.Option(
+            ...,
+            "--agent",
+            "-a",
+            help="Agent reference in module:attribute format (e.g., crm_agent:crm_agent).",
+        ),
+        worker_id: int = typer.Option(
+            0,
+            "--worker-id",
+            "-w",
+            help="Worker ID for this runner instance.",
+        ),
+    ) -> None:
+        """Start the agent runner, polling for rollouts from the store server."""
+        asyncio.run(_run_agent(agent, worker_id=worker_id))
+
+    app()
+
+
+if __name__ == "__main__":
+    main()

gatewaysdk/cli/http_client.py

@@ -0,0 +1,115 @@
+# Copyright (c) Gateway. All rights reserved.
+
+"""Shared HTTP client with centralized auth and config for the gateway CLI.
+
+Credential priority:
+1. CLI flags (--host, --api-key)
+2. Environment variables (GATEWAY_HOST, GATEWAY_SECRET_KEY)
+3. Config file (~/.gateway/config.json)
+4. Error with helpful message
+"""
+
+from __future__ import annotations
+
+import json
+import os
+import sys
+from pathlib import Path
+from typing import Any, Dict, Optional, Tuple
+
+CONFIG_DIR = Path.home() / ".gateway"
+CONFIG_FILE = CONFIG_DIR / "config.json"
+
+
+def get_config() -> Dict[str, Any]:
+    """Read config from ~/.gateway/config.json."""
+    if not CONFIG_FILE.exists():
+        return {}
+    return json.loads(CONFIG_FILE.read_text())
+
+
+def save_config(config: Dict[str, Any]) -> None:
+    """Write config to ~/.gateway/config.json."""
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True, mode=0o700)
+    CONFIG_FILE.write_text(json.dumps(config, indent=2))
+    CONFIG_FILE.chmod(0o600)
+
+
+def get_credentials(
+    host: Optional[str] = None,
+    api_key: Optional[str] = None,
+    public_key: Optional[str] = None,
+) -> Tuple[str, str, str]:
+    """Resolve host, public_key, and secret_key from flags, env vars, or config.
+
+    Args:
+        host: Explicit host override (from CLI flag).
+        api_key: Explicit secret key override (from CLI flag).
+        public_key: Explicit public key override (from CLI flag).
+
+    Returns:
+        Tuple of (host, public_key, secret_key).
+
+    Raises:
+        SystemExit: If credentials cannot be resolved.
+    """
+    config = get_config()
+
+    resolved_host = host or os.getenv("GATEWAY_HOST") or config.get("host")
+    resolved_secret = api_key or os.getenv("GATEWAY_SECRET_KEY") or config.get("api_key")
+    resolved_public = public_key or os.getenv("GATEWAY_PUBLIC_KEY") or config.get("public_key")
+
+    if not resolved_host:
+        print(
+            "Error: Host not configured. Use --host, set GATEWAY_HOST, " "or run `gateway auth login`.",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    if not resolved_secret:
+        print(
+            "Error: Secret key not configured. Use --api-key, set GATEWAY_SECRET_KEY, " "or run `gateway auth login`.",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    if not resolved_public:
+        print(
+            "Error: Public key not configured. Use --public-key, set GATEWAY_PUBLIC_KEY, "
+            "or run `gateway auth login`.",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    return resolved_host, resolved_public, resolved_secret
+
+
+def create_client(
+    host: Optional[str] = None,
+    api_key: Optional[str] = None,
+) -> Any:
+    """Create an httpx.Client with Basic auth.
+
+    Args:
+        host: Explicit host override.
+        api_key: Explicit api_key override.
+
+    Returns:
+        httpx.Client configured with base_url and auth.
+    """
+    try:
+        import httpx
+    except ImportError:
+        print(
+            "httpx is required. Install with: pip install gatewaysdk[platform]",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    resolved_host, resolved_public, resolved_secret = get_credentials(host, api_key)
+
+    return httpx.Client(
+        base_url=resolved_host.rstrip("/"),
+        auth=(resolved_public, resolved_secret),
+        timeout=30.0,
+    )