From 410e35d74212b90e6eef5bc4ce93861fca9898f2 Mon Sep 17 00:00:00 2001 From: alvinkam2001 Date: Tue, 25 Aug 2026 15:48:55 -0700 Subject: [PATCH 1/2] add optional metrics_headers param to AgentexWorker and get_temporal_client --- src/agentex/lib/core/temporal/workers/worker.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/agentex/lib/core/temporal/workers/worker.py b/src/agentex/lib/core/temporal/workers/worker.py index 0cfe01185..52d9b8510 100644 --- a/src/agentex/lib/core/temporal/workers/worker.py +++ b/src/agentex/lib/core/temporal/workers/worker.py @@ -95,6 +95,7 @@ def _validate_interceptors(interceptors: list) -> None: async def get_temporal_client( temporal_address: str, metrics_url: str | None = None, + metrics_headers: dict[str, str] | None = None, plugins: list = [], payload_codec: PayloadCodec | None = None, data_converter: DataConverter | None = None, @@ -143,7 +144,10 @@ async def get_temporal_client( if not metrics_url: client = await Client.connect(**connect_kwargs) else: - runtime = Runtime(telemetry=TelemetryConfig(metrics=OpenTelemetryConfig(url=metrics_url))) + runtime = Runtime(telemetry=TelemetryConfig(metrics=OpenTelemetryConfig( + url=metrics_url, + headers=metrics_headers or {}, + ))) connect_kwargs["runtime"] = runtime client = await Client.connect(**connect_kwargs) return client @@ -159,6 +163,7 @@ def __init__( plugins: list = [], interceptors: list = [], metrics_url: str | None = None, + metrics_headers: dict[str, str] | None = None, payload_codec: PayloadCodec | None = None, data_converter: DataConverter | None = None, ): @@ -174,6 +179,7 @@ def __init__( self.plugins = plugins self.interceptors = interceptors self.metrics_url = metrics_url + self.metrics_headers = metrics_headers self.payload_codec = payload_codec self.data_converter = data_converter @@ -211,6 +217,7 @@ async def run( temporal_address=os.environ.get("TEMPORAL_ADDRESS", "localhost:7233"), plugins=self.plugins, metrics_url=self.metrics_url, + metrics_headers=self.metrics_headers, payload_codec=self.payload_codec, data_converter=self.data_converter, ) From 32ca2d2b8ebb82b4fce97133a2941293f8262977 Mon Sep 17 00:00:00 2001 From: alvinkam2001 Date: Tue, 25 Aug 2026 16:09:58 -0700 Subject: [PATCH 2/2] fix param ordering --- src/agentex/lib/core/temporal/workers/worker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/agentex/lib/core/temporal/workers/worker.py b/src/agentex/lib/core/temporal/workers/worker.py index 52d9b8510..3cf6cdc77 100644 --- a/src/agentex/lib/core/temporal/workers/worker.py +++ b/src/agentex/lib/core/temporal/workers/worker.py @@ -95,10 +95,10 @@ def _validate_interceptors(interceptors: list) -> None: async def get_temporal_client( temporal_address: str, metrics_url: str | None = None, - metrics_headers: dict[str, str] | None = None, plugins: list = [], payload_codec: PayloadCodec | None = None, data_converter: DataConverter | None = None, + metrics_headers: dict[str, str] | None = None, ) -> Client: if plugins != []: # We don't need to validate the plugins if they are empty _validate_plugins(plugins) @@ -163,9 +163,9 @@ def __init__( plugins: list = [], interceptors: list = [], metrics_url: str | None = None, - metrics_headers: dict[str, str] | None = None, payload_codec: PayloadCodec | None = None, data_converter: DataConverter | None = None, + metrics_headers: dict[str, str] | None = None, ): self.task_queue = task_queue self.activity_handles = []