From 0d1f2d317e64c2d1c53bd1b58a349c66de908fdc Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Mon, 18 May 2026 10:15:54 -0700 Subject: [PATCH] chore(generate): rename X-Comfy-Env header to Comfy-Env Co-Authored-By: Claude Opus 4.7 (1M context) --- comfy_cli/command/generate/app.py | 2 +- comfy_cli/command/generate/client.py | 2 +- tests/comfy_cli/command/generate/test_app.py | 2 +- tests/comfy_cli/command/generate/test_client.py | 2 +- tests/comfy_cli/command/generate/test_upload.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/comfy_cli/command/generate/app.py b/comfy_cli/command/generate/app.py index 5897c009..03b99243 100644 --- a/comfy_cli/command/generate/app.py +++ b/comfy_cli/command/generate/app.py @@ -305,7 +305,7 @@ def _refresh() -> None: url = spec.base_url() + "/openapi.yml" try: with httpx.Client(timeout=30.0, follow_redirects=True) as cli: - r = cli.get(url, headers={"X-Comfy-Env": "comfy-cli", "User-Agent": "comfy-cli/api"}) + r = cli.get(url, headers={"Comfy-Env": "comfy-cli", "User-Agent": "comfy-cli/api"}) r.raise_for_status() except httpx.HTTPError as e: rprint(f"[bold red]Failed to fetch {url}: {e}[/bold red]") diff --git a/comfy_cli/command/generate/client.py b/comfy_cli/command/generate/client.py index 10f6e31d..d312f134 100644 --- a/comfy_cli/command/generate/client.py +++ b/comfy_cli/command/generate/client.py @@ -85,7 +85,7 @@ def _auth_headers(api_key: str, extra: dict[str, str] | None = None) -> dict[str # - "comfyui-..." API keys → X-API-Key (validated by sha256 lookup) # - Firebase ID tokens → Authorization: Bearer (validated as a JWT) # See comfy-api server/middleware/authentication/comfy_firebase_auth.go. - headers = {"User-Agent": "comfy-cli/api", "X-Comfy-Env": "comfy-cli"} + headers = {"User-Agent": "comfy-cli/api", "Comfy-Env": "comfy-cli"} if api_key.startswith("comfyui-"): headers["X-API-Key"] = api_key else: diff --git a/tests/comfy_cli/command/generate/test_app.py b/tests/comfy_cli/command/generate/test_app.py index b82f618a..6efb7108 100644 --- a/tests/comfy_cli/command/generate/test_app.py +++ b/tests/comfy_cli/command/generate/test_app.py @@ -393,7 +393,7 @@ def get(self, url, headers=None): assert r.exit_code == 0, r.stdout assert "Refreshed" in r.stdout assert (tmp_path / "openapi-cache.yml").exists() - assert captured["headers"].get("X-Comfy-Env") == "comfy-cli" + assert captured["headers"].get("Comfy-Env") == "comfy-cli" def test_refresh_network_failure(runner, monkeypatch): diff --git a/tests/comfy_cli/command/generate/test_client.py b/tests/comfy_cli/command/generate/test_client.py index 9fed9f03..c3909143 100644 --- a/tests/comfy_cli/command/generate/test_client.py +++ b/tests/comfy_cli/command/generate/test_client.py @@ -74,7 +74,7 @@ def test_send_request_uses_x_api_key_for_comfyui_keys(monkeypatch): client.send_request(ep, {"prompt": "x", "width": 1, "height": 1}, flags, api_key="comfyui-abc") assert captured["headers"]["X-API-Key"] == "comfyui-abc" assert "Authorization" not in captured["headers"] - assert captured["headers"]["X-Comfy-Env"] == "comfy-cli" + assert captured["headers"]["Comfy-Env"] == "comfy-cli" def test_send_request_uses_bearer_for_firebase_tokens(monkeypatch): diff --git a/tests/comfy_cli/command/generate/test_upload.py b/tests/comfy_cli/command/generate/test_upload.py index f7cf30f9..b88102cf 100644 --- a/tests/comfy_cli/command/generate/test_upload.py +++ b/tests/comfy_cli/command/generate/test_upload.py @@ -28,7 +28,7 @@ def fake_post(url, *, json=None, headers=None, timeout=None): assert body["upload_url"] == "https://signed/up" assert captured["json"] == {"file_name": "cat.png", "content_type": "image/png", "file_hash": "deadbeef"} assert captured["headers"]["X-API-Key"] == "comfyui-test" - assert captured["headers"]["X-Comfy-Env"] == "comfy-cli" + assert captured["headers"]["Comfy-Env"] == "comfy-cli" assert captured["url"].endswith("/customers/storage")