feat: pass x-uipath-licensing-context header on LLM Gateway calls (#670)

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>

Author: cotovanu-cristian

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.8.6"
+version = "0.8.7"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath_langchain/chat/http_client/headers.py

@@ -3,6 +3,7 @@
 import os
 from urllib.parse import quote
 
+from uipath.platform.common._config import UiPathConfig
 from uipath.platform.common.constants import (
     ENV_FOLDER_KEY,
     ENV_JOB_KEY,
@@ -15,6 +16,7 @@
     HEADER_INTERNAL_ACCOUNT_ID,
     HEADER_INTERNAL_TENANT_ID,
     HEADER_JOB_KEY,
+    HEADER_LICENSING_CONTEXT,
     HEADER_LLMGATEWAY_BYO_CONNECTION_ID,
     HEADER_PROCESS_KEY,
     HEADER_TRACE_ID,
@@ -53,9 +55,13 @@ def build_uipath_headers(
         headers[HEADER_FOLDER_KEY] = folder_key
     if trace_id := os.getenv(ENV_UIPATH_TRACE_ID):
         headers[HEADER_TRACE_ID] = trace_id
+    if licensing_context := UiPathConfig.licensing_context:
+        headers[HEADER_LICENSING_CONTEXT] = licensing_context
+
     if inject_routing:
         if tenant_id := os.getenv(ENV_TENANT_ID):
             headers[HEADER_INTERNAL_TENANT_ID] = tenant_id
         if organization_id := os.getenv(ENV_ORGANIZATION_ID):
             headers[HEADER_INTERNAL_ACCOUNT_ID] = organization_id
+
     return headers

tests/chat/test_context_headers.py

@@ -1,5 +1,5 @@
 import os
-from unittest.mock import patch
+from unittest.mock import PropertyMock, patch
 
 from uipath_langchain.chat.http_client import build_uipath_headers
 
@@ -49,3 +49,27 @@ def test_optional_gateway_headers(self) -> None:
             )
         assert headers["x-uipath-agenthub-config"] == "config-abc"
         assert headers["x-uipath-llmgateway-byoisconnectionid"] == "conn-xyz"
+
+    def test_licensing_context_header_present(self) -> None:
+        with (
+            patch.dict(os.environ, {}, clear=True),
+            patch(
+                "uipath.platform.common._config.ConfigurationManager.licensing_context",
+                new_callable=PropertyMock,
+                return_value="robot:unattended",
+            ),
+        ):
+            headers = build_uipath_headers()
+        assert headers["x-uipath-licensing-context"] == "robot:unattended"
+
+    def test_licensing_context_header_absent_when_none(self) -> None:
+        with (
+            patch.dict(os.environ, {}, clear=True),
+            patch(
+                "uipath.platform.common._config.ConfigurationManager.licensing_context",
+                new_callable=PropertyMock,
+                return_value=None,
+            ),
+        ):
+            headers = build_uipath_headers()
+        assert "x-uipath-licensing-context" not in headers