diff --git a/tests/integration/telemetry/conftest.py b/tests/integration/telemetry/conftest.py index 8cfed5d4e..f8a7ff771 100644 --- a/tests/integration/telemetry/conftest.py +++ b/tests/integration/telemetry/conftest.py @@ -17,17 +17,9 @@ from tests.integration.telemetry.collectors import InMemoryTelemetryManager, Otl @pytest.fixture(scope="session") -def telemetry_test_collector(request): - # Determine stack mode from --stack-config argument - stack_config = request.session.config.getoption("--stack-config", default=None) - if not stack_config: - stack_config = os.environ.get("LLAMA_STACK_CONFIG", "") - - # Check if running in server or docker mode (both need server-side telemetry) - if stack_config.startswith("server:") or stack_config.startswith("docker:"): - stack_mode = "server" - else: - stack_mode = os.environ.get("LLAMA_STACK_TEST_STACK_CONFIG_TYPE", "library_client") +def telemetry_test_collector(): + # Stack mode is set by integration-tests.sh based on STACK_CONFIG + stack_mode = os.environ.get("LLAMA_STACK_TEST_STACK_CONFIG_TYPE", "library_client") if stack_mode == "server": try: