From d0412958e3a59d0dc62ca74c258e240d6c23a205 Mon Sep 17 00:00:00 2001 From: ViditOstwal Date: Fri, 18 Sep 2026 19:35:48 +0530 Subject: [PATCH] fix(platform): retain integration token environment key --- lib/cli/src/crewai_cli/create_json_crew.py | 14 ++++++-------- lib/cli/tests/test_create_crew.py | 2 +- .../tools/crewai_platform_tools/misc.py | 8 ++------ .../test_integrations_client.py | 2 +- 4 files changed, 10 insertions(+), 16 deletions(-) diff --git a/lib/cli/src/crewai_cli/create_json_crew.py b/lib/cli/src/crewai_cli/create_json_crew.py index a1a310cdd..16b63645d 100644 --- a/lib/cli/src/crewai_cli/create_json_crew.py +++ b/lib/cli/src/crewai_cli/create_json_crew.py @@ -910,7 +910,7 @@ def _prompt_platform_token() -> str: ) return str( click.prompt( - click.style(" CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN", fg="cyan"), + click.style(" CREWAI_PLATFORM_INTEGRATION_TOKEN", fg="cyan"), hide_input=True, prompt_suffix=click.style(" > ", fg="bright_white"), ) @@ -1082,9 +1082,7 @@ def _setup_platform_auth(agents: list[dict[str, Any]]) -> str | None: "`pip install 'crewai[tools]'`." ) from error - token = os.environ.get("CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN") or os.environ.get( - "CREWAI_PLATFORM_INTEGRATION_TOKEN", "" - ) + token = os.environ.get("CREWAI_PLATFORM_INTEGRATION_TOKEN", "") while True: if not token: token = _prompt_platform_token() @@ -1096,7 +1094,7 @@ def _setup_platform_auth(agents: list[dict[str, Any]]) -> str | None: ) continue - os.environ["CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN"] = token + os.environ["CREWAI_PLATFORM_INTEGRATION_TOKEN"] = token failed_apps, token_invalid = _validate_platform_apps( apps, ApplicationSelector, client_for_selector ) @@ -1112,7 +1110,7 @@ def _setup_platform_auth(agents: list[dict[str, Any]]) -> str | None: replacement_token = _prompt_platform_revalidation_token() if replacement_token: token = replacement_token - os.environ["CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN"] = token + os.environ["CREWAI_PLATFORM_INTEGRATION_TOKEN"] = token # ── Main ──────────────────────────────────────────────────────── @@ -1180,9 +1178,9 @@ def create_json_crew( copy_assistant_instructions(folder_path) if platform_token: - os.environ["CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN"] = platform_token + os.environ["CREWAI_PLATFORM_INTEGRATION_TOKEN"] = platform_token env_vars = load_env_vars(folder_path) - env_vars["CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN"] = platform_token + env_vars["CREWAI_PLATFORM_INTEGRATION_TOKEN"] = platform_token write_env_file(folder_path, env_vars) _success("CrewAI Platform Enterprise Action Auth Token saved to .env") diff --git a/lib/cli/tests/test_create_crew.py b/lib/cli/tests/test_create_crew.py index 6407ae59d..de93410a6 100644 --- a/lib/cli/tests/test_create_crew.py +++ b/lib/cli/tests/test_create_crew.py @@ -1042,7 +1042,7 @@ def test_json_create_saves_platform_token_to_env_file(tmp_path, monkeypatch): json_crew.create_json_crew("Platform Crew", skip_provider=True) env_file = tmp_path / "platform_crew" / ".env" - assert "CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN=token" in env_file.read_text() + assert "CREWAI_PLATFORM_INTEGRATION_TOKEN=token" in env_file.read_text() def test_json_crew_uses_template_files(): diff --git a/lib/crewai-tools/src/crewai_tools/tools/crewai_platform_tools/misc.py b/lib/crewai-tools/src/crewai_tools/tools/crewai_platform_tools/misc.py index 574ead030..4fa6c972b 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/crewai_platform_tools/misc.py +++ b/lib/crewai-tools/src/crewai_tools/tools/crewai_platform_tools/misc.py @@ -3,14 +3,10 @@ import os def get_platform_integration_token() -> str: """Get the Platform Enterprise Action Auth Token from the environment.""" - token = ( - os.getenv("CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN") - or os.getenv("CREWAI_PLATFORM_INTEGRATION_TOKEN") - or "" - ) + token = os.getenv("CREWAI_PLATFORM_INTEGRATION_TOKEN") or "" if not token: raise ValueError( "No Platform Enterprise Action Auth Token found, please set the " - "CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN environment variable" + "CREWAI_PLATFORM_INTEGRATION_TOKEN environment variable" ) return token # TODO: Use context manager to get token diff --git a/lib/crewai-tools/tests/tools/crewai_platform_tools/test_integrations_client.py b/lib/crewai-tools/tests/tools/crewai_platform_tools/test_integrations_client.py index 7d012961f..7d57a5d01 100644 --- a/lib/crewai-tools/tests/tools/crewai_platform_tools/test_integrations_client.py +++ b/lib/crewai-tools/tests/tools/crewai_platform_tools/test_integrations_client.py @@ -367,7 +367,7 @@ def test_clipper_client_requires_platform_integration_token( parameters={}, ) - with pytest.raises(ValueError, match="CREWAI_ENTERPRISE_ACTION_AUTH_TOKEN"): + with pytest.raises(ValueError, match="CREWAI_PLATFORM_INTEGRATION_TOKEN"): ClipperClient().execute_action(tool, {}) mock_post.assert_not_called()