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 77e48d0f..26354b86 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.sgp_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 ede85d83..4791d6b2 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 + sgp_base_url: str | None = None TracingProcessorConfig = Annotated[