From 05cccaffa24a0c47247975e92ec60ac6eb42b4e7 Mon Sep 17 00:00:00 2001 From: BennyKok Date: Sun, 11 May 2025 10:41:10 +0800 Subject: [PATCH 1/2] support for API_KEY_COMFY_ORG --- custom_routes.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/custom_routes.py b/custom_routes.py index b43ba80..6685cfd 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -158,6 +158,8 @@ from logging import basicConfig, getLogger # Check for an environment variable to enable/disable Logfire use_logfire = os.environ.get("USE_LOGFIRE", "false").lower() == "true" +API_KEY_COMFY_ORG = os.environ.get("API_KEY_COMFY_ORG", "false") + if use_logfire: try: import logfire @@ -283,6 +285,9 @@ def post_prompt(json_data): if "extra_data" in json_data: extra_data = json_data["extra_data"] + if API_KEY_COMFY_ORG is not None: + extra_data["API_KEY_COMFY_ORG"] = API_KEY_COMFY_ORG + if "client_id" in json_data: extra_data["client_id"] = json_data["client_id"] if valid[0]: From a557788e70abf5714a7595b1fafc765fd42d6e7e Mon Sep 17 00:00:00 2001 From: BennyKok Date: Sun, 11 May 2025 10:57:07 +0800 Subject: [PATCH 2/2] fix --- custom_routes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/custom_routes.py b/custom_routes.py index 6685cfd..81ffcdc 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -158,7 +158,7 @@ from logging import basicConfig, getLogger # Check for an environment variable to enable/disable Logfire use_logfire = os.environ.get("USE_LOGFIRE", "false").lower() == "true" -API_KEY_COMFY_ORG = os.environ.get("API_KEY_COMFY_ORG", "false") +API_KEY_COMFY_ORG = os.environ.get("API_KEY_COMFY_ORG", None) if use_logfire: try: @@ -286,7 +286,7 @@ def post_prompt(json_data): extra_data = json_data["extra_data"] if API_KEY_COMFY_ORG is not None: - extra_data["API_KEY_COMFY_ORG"] = API_KEY_COMFY_ORG + extra_data["api_key_comfy_org"] = API_KEY_COMFY_ORG if "client_id" in json_data: extra_data["client_id"] = json_data["client_id"]