From 12334b40aefa03657898efdd81263c0a904ab90f Mon Sep 17 00:00:00 2001 From: d42me Date: Thu, 25 Dec 2025 22:39:43 +0100 Subject: [PATCH 1/4] Add global team notice. --- packages/prime/src/prime_cli/commands/env.py | 6 ++++++ packages/prime/src/prime_cli/main.py | 14 +++++++++++++- 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/packages/prime/src/prime_cli/commands/env.py b/packages/prime/src/prime_cli/commands/env.py index b51ac7f0..2f2c1f87 100644 --- a/packages/prime/src/prime_cli/commands/env.py +++ b/packages/prime/src/prime_cli/commands/env.py @@ -2163,6 +2163,12 @@ def run_eval( # If a team is configured, pass it to vf-eval via header if config.team_id: cmd += ["--header", f"X-Prime-Team-ID: {config.team_id}"] + if config.team_id_from_env: + console.print(f"[cyan]ℹ Using team account: {config.team_id} (from env var)[/cyan]") + elif config.team_name: + console.print(f"[cyan]ℹ Using team account: {config.team_name}[/cyan]") + else: + console.print(f"[cyan]ℹ Using team account: {config.team_id}[/cyan]") console.print(f"[dim]Eval job_id: {job_id}[/dim]") diff --git a/packages/prime/src/prime_cli/main.py b/packages/prime/src/prime_cli/main.py index 58e7d325..874c70d6 100644 --- a/packages/prime/src/prime_cli/main.py +++ b/packages/prime/src/prime_cli/main.py @@ -72,9 +72,10 @@ def callback( # Check for updates (only when a subcommand is being executed) if ctx.invoked_subcommand is not None: + console = Console(stderr=True, force_terminal=sys.stderr.isatty()) + update_available, latest = check_for_update() if update_available and latest: - console = Console(stderr=True, force_terminal=sys.stderr.isatty()) console.print( f"[yellow]A new version of prime is available: {latest} " f"(installed: {__version__})[/yellow]" @@ -82,6 +83,17 @@ def callback( console.print("[dim]Run: uv pip install --upgrade prime or uv tool upgrade prime[/dim]") console.print("[dim]Set PRIME_DISABLE_VERSION_CHECK=1 to disable this check[/dim]\n") + config = Config() + if config.team_id: + if config.team_id_from_env: + console.print( + f"[cyan]ℹ Using team account: {config.team_id} (from env var)[/cyan]\n" + ) + elif config.team_name: + console.print(f"[cyan]ℹ Using team account: {config.team_name}[/cyan]\n") + else: + console.print(f"[cyan]ℹ Using team account: {config.team_id}[/cyan]\n") + def run() -> None: """Entry point for the CLI""" From 6ee9e37eb32d08d0a4c64e5a1bce0c1d556c6792 Mon Sep 17 00:00:00 2001 From: d42me Date: Tue, 30 Dec 2025 17:09:07 -0600 Subject: [PATCH 2/4] Remove duplicate notice for env. --- packages/prime/src/prime_cli/commands/env.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/packages/prime/src/prime_cli/commands/env.py b/packages/prime/src/prime_cli/commands/env.py index 2f2c1f87..b51ac7f0 100644 --- a/packages/prime/src/prime_cli/commands/env.py +++ b/packages/prime/src/prime_cli/commands/env.py @@ -2163,12 +2163,6 @@ def run_eval( # If a team is configured, pass it to vf-eval via header if config.team_id: cmd += ["--header", f"X-Prime-Team-ID: {config.team_id}"] - if config.team_id_from_env: - console.print(f"[cyan]ℹ Using team account: {config.team_id} (from env var)[/cyan]") - elif config.team_name: - console.print(f"[cyan]ℹ Using team account: {config.team_name}[/cyan]") - else: - console.print(f"[cyan]ℹ Using team account: {config.team_id}[/cyan]") console.print(f"[dim]Eval job_id: {job_id}[/dim]") From b84b6fdfba6c54e59a3538c5b5b971e608b374a9 Mon Sep 17 00:00:00 2001 From: d42me Date: Fri, 2 Jan 2026 13:50:44 -0600 Subject: [PATCH 3/4] Format team notice. --- packages/prime/README.md | 9 +++++++++ packages/prime/src/prime_cli/main.py | 16 +++++----------- 2 files changed, 14 insertions(+), 11 deletions(-) diff --git a/packages/prime/README.md b/packages/prime/README.md index 92f62513..ba92e431 100644 --- a/packages/prime/README.md +++ b/packages/prime/README.md @@ -208,6 +208,15 @@ prime config set-ssh-key-path ~/.ssh/id_rsa.pub prime config view ``` +### Environment Variables + +| Variable | Description | +|----------|-------------| +| `PRIME_API_KEY` | API key for authentication | +| `PRIME_TEAM_ID` | Team ID to use for all operations | +| `PRIME_DISABLE_VERSION_CHECK` | Set to `1` to disable update notifications | +| `PRIME_DISABLE_TEAM_NOTICE` | Set to `1` to disable the team account notice | + ## Python SDK Prime also provides a Python SDK for programmatic access: diff --git a/packages/prime/src/prime_cli/main.py b/packages/prime/src/prime_cli/main.py index 874c70d6..f26601f9 100644 --- a/packages/prime/src/prime_cli/main.py +++ b/packages/prime/src/prime_cli/main.py @@ -1,3 +1,4 @@ +import os import sys from typing import Optional @@ -56,8 +57,6 @@ def callback( raise typer.Exit() if context: - import os - config = Config() # Check if the context exists if context.lower() != "production" and context not in config.list_environments(): @@ -84,15 +83,10 @@ def callback( console.print("[dim]Set PRIME_DISABLE_VERSION_CHECK=1 to disable this check[/dim]\n") config = Config() - if config.team_id: - if config.team_id_from_env: - console.print( - f"[cyan]ℹ Using team account: {config.team_id} (from env var)[/cyan]\n" - ) - elif config.team_name: - console.print(f"[cyan]ℹ Using team account: {config.team_name}[/cyan]\n") - else: - console.print(f"[cyan]ℹ Using team account: {config.team_id}[/cyan]\n") + if config.team_id and not os.getenv("PRIME_DISABLE_TEAM_NOTICE"): + team_display = config.team_name or config.team_id + suffix = " (from env var)" if config.team_id_from_env else "" + console.print(f"[dim]Using team account: {team_display}{suffix}[/dim]\n") def run() -> None: From 3cb5a111d3b16f7ecc74c2da825d20ca86b3ac4c Mon Sep 17 00:00:00 2001 From: d42me Date: Fri, 2 Jan 2026 14:49:34 -0600 Subject: [PATCH 4/4] Fix bugbot comments. --- packages/prime/src/prime_cli/main.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/packages/prime/src/prime_cli/main.py b/packages/prime/src/prime_cli/main.py index f26601f9..a3acaf7d 100644 --- a/packages/prime/src/prime_cli/main.py +++ b/packages/prime/src/prime_cli/main.py @@ -83,8 +83,14 @@ def callback( console.print("[dim]Set PRIME_DISABLE_VERSION_CHECK=1 to disable this check[/dim]\n") config = Config() - if config.team_id and not os.getenv("PRIME_DISABLE_TEAM_NOTICE"): - team_display = config.team_name or config.team_id + if config.team_id and os.environ.get("PRIME_DISABLE_TEAM_NOTICE", "").lower() not in ( + "1", + "true", + "yes", + ): + team_display = ( + config.team_id if config.team_id_from_env else (config.team_name or config.team_id) + ) suffix = " (from env var)" if config.team_id_from_env else "" console.print(f"[dim]Using team account: {team_display}{suffix}[/dim]\n")