From 4b885b5a7589acf110878c131c4257b7f5b011c3 Mon Sep 17 00:00:00 2001
From: iliescucristian <cristian.iliescu@uipath.com>
Date: Mon, 29 Jun 2026 12:17:59 +0300
Subject: [PATCH] feat(platform): add source=agents baggage to guardrails trace
 context headers

Pass extra_baggage=["source=agents"] to build_trace_context_headers() in
GuardrailsService so the server can distinguish guardrail validation
requests originating from the agents runtime.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 packages/uipath-platform/pyproject.toml                       | 2 +-
 .../src/uipath/platform/guardrails/_guardrails_service.py     | 2 +-
 packages/uipath-platform/uv.lock                              | 2 +-
 packages/uipath/pyproject.toml                                | 4 ++--
 packages/uipath/uv.lock                                       | 4 ++--
 5 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/packages/uipath-platform/pyproject.toml b/packages/uipath-platform/pyproject.toml
index 21c693e7b..d46a6de4b 100644
--- a/packages/uipath-platform/pyproject.toml
+++ b/packages/uipath-platform/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "uipath-platform"
-version = "0.1.78"
+version = "0.1.79"
 description = "HTTP client library for programmatic access to UiPath Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
diff --git a/packages/uipath-platform/src/uipath/platform/guardrails/_guardrails_service.py b/packages/uipath-platform/src/uipath/platform/guardrails/_guardrails_service.py
index 5082843ab..1d860b8d1 100644
--- a/packages/uipath-platform/src/uipath/platform/guardrails/_guardrails_service.py
+++ b/packages/uipath-platform/src/uipath/platform/guardrails/_guardrails_service.py
@@ -129,7 +129,7 @@ def evaluate_guardrail(
         # the execution source (x-uipath-guardrails-source) and job key headers
         # for licensing/metering correlation. The execution source is read from
         # the execution context, propagated from the runtime context.
-        trace_headers = build_trace_context_headers()
+        trace_headers = build_trace_context_headers(extra_baggage=["source=agents"])
         source_headers: dict[str, str] = {}
         execution_source = self._execution_context.execution_source
         if execution_source:
diff --git a/packages/uipath-platform/uv.lock b/packages/uipath-platform/uv.lock
index c0552fa41..da9ec91d6 100644
--- a/packages/uipath-platform/uv.lock
+++ b/packages/uipath-platform/uv.lock
@@ -1095,7 +1095,7 @@ dev = [
 
 [[package]]
 name = "uipath-platform"
-version = "0.1.78"
+version = "0.1.79"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },
diff --git a/packages/uipath/pyproject.toml b/packages/uipath/pyproject.toml
index 62bb9a5e1..37b64b956 100644
--- a/packages/uipath/pyproject.toml
+++ b/packages/uipath/pyproject.toml
@@ -1,13 +1,13 @@
 [project]
 name = "uipath"
-version = "2.11.13"
+version = "2.11.14"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
   "uipath-core>=0.5.21, <0.6.0",
   "uipath-runtime>=0.11.4, <0.12.0",
-  "uipath-platform>=0.1.78, <0.2.0",
+  "uipath-platform>=0.1.79, <0.2.0",
   "click>=8.3.1",
   "httpx>=0.28.1",
   "pyjwt>=2.10.1",
diff --git a/packages/uipath/uv.lock b/packages/uipath/uv.lock
index f99dd06a0..1a3771d8b 100644
--- a/packages/uipath/uv.lock
+++ b/packages/uipath/uv.lock
@@ -2552,7 +2552,7 @@ wheels = [
 
 [[package]]
 name = "uipath"
-version = "2.11.13"
+version = "2.11.14"
 source = { editable = "." }
 dependencies = [
     { name = "applicationinsights" },
@@ -2691,7 +2691,7 @@ dev = [
 
 [[package]]
 name = "uipath-platform"
-version = "0.1.78"
+version = "0.1.79"
 source = { editable = "../uipath-platform" }
 dependencies = [
     { name = "httpx" },