From a7afff2c0e06749922d2388a9fdb81ce37cdf76d Mon Sep 17 00:00:00 2001 From: Adam Nolte Date: Tue, 24 Jun 2025 11:00:56 -0500 Subject: [PATCH 1/3] Update env variable --- autoblocks/_impl/config/constants.py | 4 +++- autoblocks/_impl/util.py | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/autoblocks/_impl/config/constants.py b/autoblocks/_impl/config/constants.py index 78e68d6a..185053dc 100644 --- a/autoblocks/_impl/config/constants.py +++ b/autoblocks/_impl/config/constants.py @@ -1,5 +1,7 @@ +from autoblocks._impl.util import AutoblocksEnvVar + API_ENDPOINT = "https://api.autoblocks.ai" -API_ENDPOINT_V2 = "https://api-v2.autoblocks.ai" +API_ENDPOINT_V2 = AutoblocksEnvVar.API_ENDPOINT_V2.get() or "https://api-v2.autoblocks.ai" INGESTION_ENDPOINT = "https://ingest-event.autoblocks.ai" REVISION_LATEST = "latest" REVISION_UNDEPLOYED = "undeployed" diff --git a/autoblocks/_impl/util.py b/autoblocks/_impl/util.py index 65099e4c..b7f60b7b 100644 --- a/autoblocks/_impl/util.py +++ b/autoblocks/_impl/util.py @@ -76,6 +76,7 @@ class AutoblocksEnvVar(StrEnum): ALIGN_TEST_EXTERNAL_ID = "AUTOBLOCKS_ALIGN_TEST_EXTERNAL_ID" API_KEY = "AUTOBLOCKS_API_KEY" V2_API_KEY = "AUTOBLOCKS_V2_API_KEY" + API_ENDPOINT_V2 = "AUTOBLOCKS_API_ENDPOINT_V2" CLI_SERVER_ADDRESS = "AUTOBLOCKS_CLI_SERVER_ADDRESS" INGESTION_KEY = "AUTOBLOCKS_INGESTION_KEY" FILTERS_TEST_SUITES = "AUTOBLOCKS_FILTERS_TEST_SUITES" From 9e3e5df10aca2a2c558d800985afe8d391c0f5e9 Mon Sep 17 00:00:00 2001 From: Adam Nolte Date: Tue, 24 Jun 2025 11:01:35 -0500 Subject: [PATCH 2/3] Update env variable --- autoblocks/_impl/config/constants.py | 2 +- autoblocks/_impl/util.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/autoblocks/_impl/config/constants.py b/autoblocks/_impl/config/constants.py index 185053dc..b6d83d96 100644 --- a/autoblocks/_impl/config/constants.py +++ b/autoblocks/_impl/config/constants.py @@ -1,7 +1,7 @@ from autoblocks._impl.util import AutoblocksEnvVar API_ENDPOINT = "https://api.autoblocks.ai" -API_ENDPOINT_V2 = AutoblocksEnvVar.API_ENDPOINT_V2.get() or "https://api-v2.autoblocks.ai" +API_ENDPOINT_V2 = AutoblocksEnvVar.V2_API_ENDPOINT_V2.get() or "https://api-v2.autoblocks.ai" INGESTION_ENDPOINT = "https://ingest-event.autoblocks.ai" REVISION_LATEST = "latest" REVISION_UNDEPLOYED = "undeployed" diff --git a/autoblocks/_impl/util.py b/autoblocks/_impl/util.py index b7f60b7b..8c69f782 100644 --- a/autoblocks/_impl/util.py +++ b/autoblocks/_impl/util.py @@ -76,7 +76,7 @@ class AutoblocksEnvVar(StrEnum): ALIGN_TEST_EXTERNAL_ID = "AUTOBLOCKS_ALIGN_TEST_EXTERNAL_ID" API_KEY = "AUTOBLOCKS_API_KEY" V2_API_KEY = "AUTOBLOCKS_V2_API_KEY" - API_ENDPOINT_V2 = "AUTOBLOCKS_API_ENDPOINT_V2" + V2_API_ENDPOINT_V2 = "AUTOBLOCKS_V2_API_ENDPOINT" CLI_SERVER_ADDRESS = "AUTOBLOCKS_CLI_SERVER_ADDRESS" INGESTION_KEY = "AUTOBLOCKS_INGESTION_KEY" FILTERS_TEST_SUITES = "AUTOBLOCKS_FILTERS_TEST_SUITES" From 16c7927743b62d2cd4f2bef65a327ccc2caaee44 Mon Sep 17 00:00:00 2001 From: Adam Nolte Date: Tue, 24 Jun 2025 11:06:02 -0500 Subject: [PATCH 3/3] Update auto tracer --- autoblocks/_impl/tracer/auto_tracer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autoblocks/_impl/tracer/auto_tracer.py b/autoblocks/_impl/tracer/auto_tracer.py index 0953e321..4b4e087a 100644 --- a/autoblocks/_impl/tracer/auto_tracer.py +++ b/autoblocks/_impl/tracer/auto_tracer.py @@ -24,7 +24,6 @@ def init_auto_tracer( *, api_key: Optional[str] = None, - api_endpoint: Optional[str] = f"{API_ENDPOINT_V2}/otel/v1/traces", is_batch_disabled: Optional[bool] = False, ) -> None: """ @@ -33,6 +32,7 @@ def init_auto_tracer( if is_auto_tracer_initialized(): log.debug("Skipping auto tracer initialization because it is already initialized") return + api_endpoint = f"{API_ENDPOINT_V2}/otel/v1/traces" log.debug(f"Initializing Autoblocks auto tracer with api_endpoint={api_endpoint}") set_global_textmap( CompositePropagator(