From 1804787a3a102a1ba9582415b9985aba1f27bdb4 Mon Sep 17 00:00:00 2001 From: Erich Woo Date: Wed, 15 Oct 2025 13:11:57 -0400 Subject: [PATCH 1/3] allow base_url in sgp tracer --- .../lib/core/tracing/processors/sgp_tracing_processor.py | 6 +++++- src/agentex/lib/types/tracing.py | 1 + 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py index 77e48d0f1..64a8f61f5 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -21,7 +21,11 @@ class SGPSyncTracingProcessor(SyncTracingProcessor): def __init__(self, config: SGPTracingProcessorConfig): disabled = config.sgp_api_key == "" or config.sgp_account_id == "" tracing.init( - SGPClient(api_key=config.sgp_api_key, account_id=config.sgp_account_id), + SGPClient( + api_key=config.sgp_api_key, + account_id=config.sgp_account_id, + base_url=config.base_url, + ), disabled=disabled, ) self._spans: dict[str, SGPSpan] = {} diff --git a/src/agentex/lib/types/tracing.py b/src/agentex/lib/types/tracing.py index ede85d83f..bc1766576 100644 --- a/src/agentex/lib/types/tracing.py +++ b/src/agentex/lib/types/tracing.py @@ -26,6 +26,7 @@ class SGPTracingProcessorConfig(BaseModel): type: Literal["sgp"] = "sgp" sgp_api_key: str sgp_account_id: str + base_url: str | None = None TracingProcessorConfig = Annotated[ From 91a3ef67d0ed59b4ed74e5505059bd0ac5d8a928 Mon Sep 17 00:00:00 2001 From: Erich Woo Date: Wed, 15 Oct 2025 14:50:50 -0400 Subject: [PATCH 2/3] consistent naming --- .../lib/core/tracing/processors/sgp_tracing_processor.py | 2 +- src/agentex/lib/types/tracing.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py index 64a8f61f5..26354b86d 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -24,7 +24,7 @@ def __init__(self, config: SGPTracingProcessorConfig): SGPClient( api_key=config.sgp_api_key, account_id=config.sgp_account_id, - base_url=config.base_url, + base_url=config.sgp_base_url, ), disabled=disabled, ) diff --git a/src/agentex/lib/types/tracing.py b/src/agentex/lib/types/tracing.py index bc1766576..4791d6b2b 100644 --- a/src/agentex/lib/types/tracing.py +++ b/src/agentex/lib/types/tracing.py @@ -26,7 +26,7 @@ class SGPTracingProcessorConfig(BaseModel): type: Literal["sgp"] = "sgp" sgp_api_key: str sgp_account_id: str - base_url: str | None = None + sgp_base_url: str | None = None TracingProcessorConfig = Annotated[ From a1dd94f3ff69fe3101cda1904afed855756edaff Mon Sep 17 00:00:00 2001 From: Erich Woo Date: Wed, 15 Oct 2025 15:04:57 -0400 Subject: [PATCH 3/3] add base_url to async too --- .../lib/core/tracing/processors/sgp_tracing_processor.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py index 26354b86d..8a298121d 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -87,7 +87,11 @@ def __init__(self, config: SGPTracingProcessorConfig): self.disabled = config.sgp_api_key == "" or config.sgp_account_id == "" self._spans: dict[str, SGPSpan] = {} self.sgp_async_client = ( - AsyncSGPClient(api_key=config.sgp_api_key, account_id=config.sgp_account_id) + AsyncSGPClient( + api_key=config.sgp_api_key, + account_id=config.sgp_account_id, + base_url=config.sgp_base_url, + ) if not self.disabled else None )