Skip to content

Commit 03e4078

Browse files
Allow base_url in SGP tracer (#149)
* allow base_url in sgp tracer * consistent naming
1 parent 996380c commit 03e4078

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,11 @@ class SGPSyncTracingProcessor(SyncTracingProcessor):
2121
def __init__(self, config: SGPTracingProcessorConfig):
2222
disabled = config.sgp_api_key == "" or config.sgp_account_id == ""
2323
tracing.init(
24-
SGPClient(api_key=config.sgp_api_key, account_id=config.sgp_account_id),
24+
SGPClient(
25+
api_key=config.sgp_api_key,
26+
account_id=config.sgp_account_id,
27+
base_url=config.sgp_base_url,
28+
),
2529
disabled=disabled,
2630
)
2731
self._spans: dict[str, SGPSpan] = {}

src/agentex/lib/types/tracing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ class SGPTracingProcessorConfig(BaseModel):
2626
type: Literal["sgp"] = "sgp"
2727
sgp_api_key: str
2828
sgp_account_id: str
29+
sgp_base_url: str | None = None
2930

3031

3132
TracingProcessorConfig = Annotated[

0 commit comments

Comments
 (0)