From 7114e5b2a3ed149e99b84db9b8a4b45e7db77ef4 Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Mon, 18 Aug 2025 12:34:40 -0700 Subject: [PATCH 1/6] Update sgp_tracing_processor.py --- .../tracing/processors/sgp_tracing_processor.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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 f614f78e..f2580a9b 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -25,8 +25,15 @@ def __init__(self, config: SGPTracingProcessorConfig): ) self._spans: dict[str, SGPSpan] = {} + def _add_source_to_span(self, span: Span) -> None: + if span.data is None: + span.data = {} + span.data["source"] = "agentex" + @override def on_span_start(self, span: Span) -> None: + self._add_source_to_span(span) + sgp_span = create_span( name=span.name, span_id=span.id, @@ -50,6 +57,7 @@ def on_span_end(self, span: Span) -> None: ) return + self._add_source_to_span(span) sgp_span.output = span.output sgp_span.metadata = span.data sgp_span.end_time = span.end_time.isoformat() @@ -71,8 +79,14 @@ def __init__(self, config: SGPTracingProcessorConfig): else None ) + def _add_source_to_span(self, span: Span) -> None: + if span.data is None: + span.data = {} + span.data["source"] = "agentex" + @override async def on_span_start(self, span: Span) -> None: + self._add_source_to_span(span) sgp_span = create_span( name=span.name, span_id=span.id, @@ -101,6 +115,7 @@ async def on_span_end(self, span: Span) -> None: ) return + self._add_source_to_span(span) sgp_span.output = span.output sgp_span.metadata = span.data sgp_span.end_time = span.end_time.isoformat() From 17257bb768673a96ab448b4515b2620a747ea63e Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Mon, 8 Sep 2025 11:28:10 -0700 Subject: [PATCH 2/6] Update sgp_tracing_processor.py --- .../lib/core/tracing/processors/sgp_tracing_processor.py | 2 +- 1 file changed, 1 insertion(+), 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 f2580a9b..ba365ae0 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -82,7 +82,7 @@ def __init__(self, config: SGPTracingProcessorConfig): def _add_source_to_span(self, span: Span) -> None: if span.data is None: span.data = {} - span.data["source"] = "agentex" + span.data["__source__"] = "agentex" @override async def on_span_start(self, span: Span) -> None: From 5ce63935d4fe82b9c9acd6f17f81c258187bae94 Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Mon, 8 Sep 2025 11:29:17 -0700 Subject: [PATCH 3/6] Update sgp_tracing_processor.py --- .../lib/core/tracing/processors/sgp_tracing_processor.py | 2 +- 1 file changed, 1 insertion(+), 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 ba365ae0..ae7d0b0f 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -28,7 +28,7 @@ def __init__(self, config: SGPTracingProcessorConfig): def _add_source_to_span(self, span: Span) -> None: if span.data is None: span.data = {} - span.data["source"] = "agentex" + span.data["__source__"] = "agentex" @override def on_span_start(self, span: Span) -> None: From db16b17f5c94e4194edac4795dd99555eee8e576 Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Wed, 1 Oct 2025 11:16:16 -0700 Subject: [PATCH 4/6] Update sgp_tracing_processor.py --- .../tracing/processors/sgp_tracing_processor.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) 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 ae7d0b0f..c0094b4a 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -12,6 +12,7 @@ from agentex.types.span import Span from agentex.lib.types.tracing import SGPTracingProcessorConfig from agentex.lib.utils.logging import make_logger +from agentex.lib.environment_variables import EnvironmentVariables logger = make_logger(__name__) @@ -24,11 +25,18 @@ def __init__(self, config: SGPTracingProcessorConfig): disabled=disabled, ) self._spans: dict[str, SGPSpan] = {} + self.env_vars = EnvironmentVariables.refresh() def _add_source_to_span(self, span: Span) -> None: if span.data is None: span.data = {} span.data["__source__"] = "agentex" + if self.env_vars.ACP_TYPE is not None: + span.data["__acp_type__"] = self.env_vars.ACP_TYPE + if self.env_vars.AGENT_NAME is not None: + span.data["__agent_name__"] = self.env_vars.AGENT_NAME + if self.env_vars.AGENT_ID is not None: + span.data["__agent_id__"] = self.env_vars.AGENT_ID @override def on_span_start(self, span: Span) -> None: @@ -78,11 +86,18 @@ def __init__(self, config: SGPTracingProcessorConfig): if not self.disabled else None ) + self.env_vars = EnvironmentVariables.refresh() def _add_source_to_span(self, span: Span) -> None: if span.data is None: span.data = {} span.data["__source__"] = "agentex" + if self.env_vars.ACP_TYPE is not None: + span.data["__acp_type__"] = self.env_vars.ACP_TYPE + if self.env_vars.AGENT_NAME is not None: + span.data["__agent_name__"] = self.env_vars.AGENT_NAME + if self.env_vars.AGENT_ID is not None: + span.data["__agent_id__"] = self.env_vars.AGENT_ID @override async def on_span_start(self, span: Span) -> None: @@ -99,6 +114,7 @@ async def on_span_start(self, span: Span) -> None: sgp_span.start_time = span.start_time.isoformat() if self.disabled: + logger.warning("SGP is disabled, skipping span upsert") return await self.sgp_async_client.spans.upsert_batch( items=[sgp_span.to_request_params()] From 06a3db0c1a48333e9eb7f3aa926d1b044becbced Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Wed, 1 Oct 2025 11:23:48 -0700 Subject: [PATCH 5/6] Update sgp_tracing_processor.py --- .../lib/core/tracing/processors/sgp_tracing_processor.py | 2 +- 1 file changed, 1 insertion(+), 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 36b8ed3f..01d89553 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -8,11 +8,11 @@ from agentex.types.span import Span from agentex.lib.types.tracing import SGPTracingProcessorConfig from agentex.lib.utils.logging import make_logger +from agentex.lib.environment_variables import EnvironmentVariables from agentex.lib.core.tracing.processors.tracing_processor_interface import ( SyncTracingProcessor, AsyncTracingProcessor, ) -from agentex.lib.environment_variables import EnvironmentVariables logger = make_logger(__name__) From bb30569c26e72cef162d4e546876ef05ce4f3f13 Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Wed, 1 Oct 2025 11:32:36 -0700 Subject: [PATCH 6/6] Update sgp_tracing_processor.py --- .../processors/sgp_tracing_processor.py | 30 ++++++++++--------- 1 file changed, 16 insertions(+), 14 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 01d89553..77e48d0f 100644 --- a/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py +++ b/src/agentex/lib/core/tracing/processors/sgp_tracing_processor.py @@ -30,13 +30,14 @@ def __init__(self, config: SGPTracingProcessorConfig): def _add_source_to_span(self, span: Span) -> None: if span.data is None: span.data = {} - span.data["__source__"] = "agentex" - if self.env_vars.ACP_TYPE is not None: - span.data["__acp_type__"] = self.env_vars.ACP_TYPE - if self.env_vars.AGENT_NAME is not None: - span.data["__agent_name__"] = self.env_vars.AGENT_NAME - if self.env_vars.AGENT_ID is not None: - span.data["__agent_id__"] = self.env_vars.AGENT_ID + if isinstance(span.data, dict): + span.data["__source__"] = "agentex" + if self.env_vars.ACP_TYPE is not None: + span.data["__acp_type__"] = self.env_vars.ACP_TYPE + if self.env_vars.AGENT_NAME is not None: + span.data["__agent_name__"] = self.env_vars.AGENT_NAME + if self.env_vars.AGENT_ID is not None: + span.data["__agent_id__"] = self.env_vars.AGENT_ID @override def on_span_start(self, span: Span) -> None: @@ -91,13 +92,14 @@ def __init__(self, config: SGPTracingProcessorConfig): def _add_source_to_span(self, span: Span) -> None: if span.data is None: span.data = {} - span.data["__source__"] = "agentex" - if self.env_vars.ACP_TYPE is not None: - span.data["__acp_type__"] = self.env_vars.ACP_TYPE - if self.env_vars.AGENT_NAME is not None: - span.data["__agent_name__"] = self.env_vars.AGENT_NAME - if self.env_vars.AGENT_ID is not None: - span.data["__agent_id__"] = self.env_vars.AGENT_ID + if isinstance(span.data, dict): + span.data["__source__"] = "agentex" + if self.env_vars.ACP_TYPE is not None: + span.data["__acp_type__"] = self.env_vars.ACP_TYPE + if self.env_vars.AGENT_NAME is not None: + span.data["__agent_name__"] = self.env_vars.AGENT_NAME + if self.env_vars.AGENT_ID is not None: + span.data["__agent_id__"] = self.env_vars.AGENT_ID @override async def on_span_start(self, span: Span) -> None: