lovelytics-cli 0.4.14__tar.gz → 0.4.16__tar.gz

{lovelytics_cli-0.4.14 → lovelytics_cli-0.4.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lovelytics-cli
-Version: 0.4.14
+Version: 0.4.16
 Summary: Development tools for Lovelytics Labs.
 License: LICENSE
 License-File: LICENSE

{lovelytics_cli-0.4.14 → lovelytics_cli-0.4.16}/love/cli.py

@@ -6,6 +6,7 @@ from love import forecast as forecast_commands
 from love import crate as crate_commands
 from love import genie as genie_commands
 from love import install as install_commands
+from love import conduit as conduit_commands
 import json
 import logging
 import click_log
@@ -212,7 +213,8 @@ def install_crate(crate_name, params_file, download):
     Downloads and installs the specified solution from the Lovelytics CodeArtifact
     repository via Lando. Parameters can be provided via a config file (--params-file) and/or
     environment variables prefixed with CRATE_ (e.g., CRATE_CATALOG_NAME). Environment
-    variables take precedence over values in the config file.
+    variables take precedence over values in the config file. Any parameter without a
+    dedicated crate flag is passed as --variable key=value.
 
     Downloaded solution archives are cached under ~/.lovelytics/crates/. Use --download
     to force a fresh download even when a cached archive exists.
@@ -267,6 +269,61 @@ def install_crate(crate_name, params_file, download):
     click.echo(f"Successfully installed solution '{crate_name}'.")
 
 
+@cli.group()
+def conduit():
+    """Entrypoint for Conduit commands."""
+    pass
+
+
+@conduit.command(name="list-budgets")
+def conduit_list_budgets():
+    """List your Conduit budgets, marking the active one."""
+    conduit_commands.list_budgets()
+
+
+@conduit.command(name="set-active-budget")
+@click.argument("budget_id")
+def conduit_set_active_budget(budget_id):
+    """Make BUDGET_ID the budget your Conduit usage is spent against."""
+    conduit_commands.set_active_budget(budget_id)
+
+
+@conduit.command(name="list-models")
+@click.option(
+    "--budget-id",
+    default=None,
+    help="Budget to list models for. Defaults to the active budget.",
+)
+def conduit_list_models(budget_id):
+    """List the models available on a Conduit budget."""
+    conduit_commands.list_models(budget_id)
+
+
+@conduit.command(
+    name="claude",
+    context_settings={"ignore_unknown_options": True, "allow_extra_args": True},
+)
+@click.option(
+    "--budget-id",
+    default=None,
+    help="Budget to spend against. Defaults to the active budget.",
+)
+@click.argument("claude_args", nargs=-1, type=click.UNPROCESSED)
+def conduit_claude(budget_id, claude_args):
+    """Start Claude Code against a Conduit budget.
+
+    The Conduit environment variables are scoped to the launched session and
+    your shell is left unchanged. Provider variables such as
+    CLAUDE_CODE_USE_BEDROCK are removed from the session. Extra arguments and
+    flags are passed straight to claude, e.g. `love conduit claude --resume` or
+    `love conduit claude -p "..." --model opus`. Use `--` only to forward
+    `--help` or a literal `--budget-id`, e.g. `love conduit claude -- --help`.
+
+    GATEWAY_URL / HEIMDALL_URL select the environment (production by default).
+    """
+    conduit_commands.launch_claude(budget_id, claude_args)
+
+
 @cli.group(name="list")
 def list_group():
     """Entrypoint for list commands."""

lovelytics_cli-0.4.16/love/conduit.py

@@ -0,0 +1,158 @@
+"""Logic for using Conduit budgets and running Claude Code against Conduit."""
+import os
+
+import click
+import requests
+
+from love import auth, install
+from love.config import CONDUIT_BUDGETS_URL, CONDUIT_URL
+
+DEFAULT_OPUS_MODEL = "claude-opus-5-5"
+DEFAULT_SONNET_MODEL = "claude-sonnet-5-5"
+DEFAULT_HAIKU_MODEL = "claude-haiku-4-5"
+CONDUIT_GRANT_HEADER = "X-Conduit-Grant-Id"
+
+# Variables that would redirect or override Claude Code's provider, auth or model
+# selection away from Conduit.
+CONFLICTING_ENV_VARS = (
+    "CLAUDE_CODE_USE_BEDROCK",
+    "CLAUDE_CODE_USE_VERTEX",
+    "CLAUDE_CODE_USE_FOUNDRY",
+    "CLAUDE_CODE_SKIP_BEDROCK_AUTH",
+    "CLAUDE_CODE_SKIP_VERTEX_AUTH",
+    "CLAUDE_CODE_SKIP_FOUNDRY_AUTH",
+    "ANTHROPIC_API_KEY",
+    "CLAUDE_CODE_OAUTH_TOKEN",
+    "ANTHROPIC_BEDROCK_BASE_URL",
+    "ANTHROPIC_VERTEX_BASE_URL",
+    "ANTHROPIC_VERTEX_PROJECT_ID",
+    "ANTHROPIC_FOUNDRY_BASE_URL",
+    "ANTHROPIC_FOUNDRY_RESOURCE",
+    "ANTHROPIC_FOUNDRY_API_KEY",
+    "AWS_BEARER_TOKEN_BEDROCK",
+    "ANTHROPIC_SMALL_FAST_MODEL",
+    "ANTHROPIC_SMALL_FAST_MODEL_AWS_REGION",
+    "CLAUDE_CODE_SUBAGENT_MODEL",
+)
+
+
+def _request(method, path, **kwargs):
+    """Call the Conduit budget API and return the parsed JSON response."""
+    token = auth.AuthHandler.get_auth()["token"]
+    res = requests.request(
+        method,
+        f"{CONDUIT_BUDGETS_URL}{path}",
+        headers={"Authorization": f"Bearer {token}"},
+        timeout=30,
+        **kwargs,
+    )
+    if not res.ok:
+        detail = res.text
+        try:
+            detail = res.json().get("detail") or detail
+        except (ValueError, AttributeError):
+            pass
+        raise click.ClickException(
+            f"Conduit request {method} {path} failed with HTTP {res.status_code}: "
+            f"{detail}"
+        )
+    return res.json()
+
+
+def _format_amount(minor, currency):
+    """Format an amount in minor units, or '-' when there is none."""
+    if minor is None:
+        return "-"
+    return f"{minor / 100:.2f} {currency or ''}".strip()
+
+
+def list_budgets():
+    """Print the caller's Conduit budgets, marking the active one."""
+    budgets = _request("GET", "/mine")
+    if not budgets:
+        click.echo("No Conduit budgets found.")
+        return
+
+    active = _request("GET", "/active-grant")
+    active_id = active.get("id") if active else None
+
+    for budget in budgets:
+        marker = "*" if budget.get("id") == active_id else " "
+        currency = budget.get("currency")
+        click.echo(
+            f"{marker} {str(budget.get('id')):<38} "
+            f"{budget.get('name') or budget.get('kind') or '':<30} "
+            f"{budget.get('status') or '':<12} "
+            f"spent {_format_amount(budget.get('spent_minor'), currency)} / "
+            f"approved {_format_amount(budget.get('approved_amount_minor'), currency)}"
+        )
+
+
+def set_active_budget(budget_id):
+    """Make the given budget the server-side active Conduit budget."""
+    grant = _request("PUT", "/active-grant", json={"grant_id": budget_id})
+    label = (grant or {}).get("name") or (grant or {}).get("id") or budget_id
+    click.echo(f"Active Conduit budget set to {label} ({budget_id}).")
+
+
+def get_available_models(budget_id=None):
+    """Fetch the models available on a budget, or the active budget if omitted."""
+    params = {"grant_id": budget_id} if budget_id else None
+    return _request("GET", "/available-models", params=params)
+
+
+def list_models(budget_id=None):
+    """Print the models available on a budget."""
+    info = get_available_models(budget_id)
+    label = info.get("grant_id") or info.get("grant_kind")
+    click.echo(f"Budget {label} ({info.get('grant_status')})")
+    for model in info.get("models", []):
+        availability = "available" if model.get("available") else "unavailable"
+        click.echo(
+            f"{str(model.get('id')):<40} {model.get('tier') or '':<12} {availability}"
+        )
+
+
+def build_claude_env(token, budget_id=None, base_env=None):
+    """Build the environment for a Claude Code session against Conduit.
+
+    Returns a new mapping and never mutates os.environ or base_env.
+    """
+    env = dict(os.environ if base_env is None else base_env)
+    for name in (*CONFLICTING_ENV_VARS, "ANTHROPIC_CUSTOM_HEADERS"):
+        env.pop(name, None)
+
+    env["ANTHROPIC_BASE_URL"] = CONDUIT_URL
+    env["ANTHROPIC_AUTH_TOKEN"] = token
+    env["ANTHROPIC_MODEL"] = DEFAULT_OPUS_MODEL
+    env["ANTHROPIC_DEFAULT_OPUS_MODEL"] = DEFAULT_OPUS_MODEL
+    env["ANTHROPIC_DEFAULT_SONNET_MODEL"] = DEFAULT_SONNET_MODEL
+    env["ANTHROPIC_DEFAULT_HAIKU_MODEL"] = DEFAULT_HAIKU_MODEL
+    if budget_id:
+        env["ANTHROPIC_CUSTOM_HEADERS"] = f"{CONDUIT_GRANT_HEADER}: {budget_id}"
+    return env
+
+
+def launch_claude(budget_id=None, claude_args=()):
+    """Replace this process with Claude Code configured to use Conduit.
+
+    execve is used instead of subprocess so Ctrl-C reaches Claude Code alone and
+    the child gets exactly the environment built here.
+    """
+    claude_binary = install.ensure_claude_binary()
+
+    info = get_available_models(budget_id)
+    status = info.get("grant_status")
+    click.echo(
+        "Starting Claude Code against Conduit budget "
+        f"{info.get('grant_id') or info.get('grant_kind') or budget_id} ({status})..."
+    )
+    if status != "available":
+        click.echo(
+            f"Warning: this budget's status is '{status}', so requests may be "
+            "rejected or exceed its allowance."
+        )
+
+    token = auth.AuthHandler.get_auth()["token"]
+    env = build_claude_env(token, budget_id)
+    os.execve(claude_binary, [claude_binary, *claude_args], env)  # nosec

{lovelytics_cli-0.4.14 → lovelytics_cli-0.4.16}/love/config.py

@@ -15,6 +15,8 @@ DBT_PROFILE_DIR = os.path.join(CONFIG_PATH, "profiles")
 LANDO_URL = os.getenv("LANDO_URL", "https://landoapi.saas.lovelytics.com")
 HEIMDALL_URL = os.getenv("HEIMDALL_URL", "https://heimdall.lovelytics.com")
 GATEWAY_URL = os.getenv("GATEWAY_URL", "https://api.lovelytics.com")
+CONDUIT_URL = f"{GATEWAY_URL}/v2/conduit"
+CONDUIT_BUDGETS_URL = f"{GATEWAY_URL}/v2/aiutils/budget-requests"
 
 CONFIG_OPTIONS = {
     "runtime": {

{lovelytics_cli-0.4.14 → lovelytics_cli-0.4.16}/love/crate.py

@@ -155,9 +155,12 @@ def build_crate_install_args(source_path: str, params: dict) -> list:
         if flag:
             args.extend([flag, str(value)])
         else:
-            logger.warning(
-                f"Ignoring unrecognized parameter '{key}' "
-                f"(no matching crate CLI flag)."
-            )
+            # crate's --variable splits on commas and quotes; such values can't round-trip.
+            if "," in str(value) or '"' in str(value):
+                raise click.ClickException(
+                    f"Parameter '{key}' must not contain a comma or double quote."
+                )
+            # Anything else is a crate variable (crate.yaml `variables:`).
+            args.extend(["--variable", f"{key}={value}"])
 
     return args

{lovelytics_cli-0.4.14 → lovelytics_cli-0.4.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lovelytics-cli"
-version = "0.4.14"
+version = "0.4.16"
 description = "Development tools for Lovelytics Labs."
 authors = ["Lovelytics Labs <labs@lovelytics.com>"]
 readme = "README.md"