fix: pass job key to is activities for licensing

Author: andreitava-uip

packages/uipath-platform/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-platform"
-version = "0.1.45"
+version = "0.1.46"
 description = "HTTP client library for programmatic access to UiPath Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

packages/uipath-platform/src/uipath/platform/common/constants.py

@@ -36,6 +36,9 @@
 HEADER_INTERNAL_TENANT_ID = "x-uipath-internal-tenantid"
 HEADER_INTERNAL_ACCOUNT_ID = "x-uipath-internal-accountid"
 HEADER_JOB_KEY = "x-uipath-jobkey"
+# Sent on outbound Integration Service activity invocations so GenAI activities
+# can be stitched back to the parent job for licensing attribution.
+HEADER_ACTIVITY_JOB_ID = "x-uipath-job-id"
 HEADER_PROCESS_KEY = "x-uipath-processkey"
 HEADER_TRACE_ID = "x-uipath-traceid"
 HEADER_AGENTHUB_CONFIG = "x-uipath-agenthub-config"

packages/uipath-platform/src/uipath/platform/connections/_connections_service.py

@@ -1,5 +1,6 @@
 import json
 import logging
+import os
 from typing import Any, Dict, List, Optional
 from urllib.parse import parse_qsl, quote, urlsplit
 
@@ -12,6 +13,7 @@
 from ..common._execution_context import UiPathExecutionContext
 from ..common._folder_context import header_folder
 from ..common._models import Endpoint, RequestSpec
+from ..common.constants import ENV_JOB_KEY, HEADER_ACTIVITY_JOB_ID
 from ..orchestrator._folder_service import FolderService
 from .connections import (
     ActivityMetadata,
@@ -773,6 +775,8 @@ def _build_activity_request_spec(
             **header_folder(folder_key, None),
             **header_params,
         }
+        if job_key := os.getenv(ENV_JOB_KEY):
+            headers[HEADER_ACTIVITY_JOB_ID] = job_key
 
         # body and files handling
         json_data: Dict[str, Any] | None = None

packages/uipath-platform/tests/services/test_connections_service.py

@@ -8,7 +8,11 @@
 
 from uipath.platform import UiPathApiConfig, UiPathExecutionContext
 from uipath.platform.common import jsonschema_to_pydantic
-from uipath.platform.common.constants import HEADER_FOLDER_KEY, HEADER_USER_AGENT
+from uipath.platform.common.constants import (
+    HEADER_ACTIVITY_JOB_ID,
+    HEADER_FOLDER_KEY,
+    HEADER_USER_AGENT,
+)
 from uipath.platform.connections import (
     ActivityMetadata,
     ActivityParameterLocationInfo,
@@ -1421,6 +1425,60 @@ def test_invoke_activity_sets_standard_headers(
         assert sent_request.headers["x-uipath-originator"] == "uipath-python"
         assert sent_request.headers["x-uipath-source"] == "uipath-python"
 
+    def test_invoke_activity_propagates_job_id_header(
+        self,
+        httpx_mock: HTTPXMock,
+        service: ConnectionsService,
+        simple_activity_metadata: ActivityMetadata,
+        monkeypatch: pytest.MonkeyPatch,
+    ) -> None:
+        """Activity invocations carry x-uipath-job-id so GenAI calls can be stitched for licensing."""
+        monkeypatch.setenv("UIPATH_JOB_KEY", "job-key-abc")
+        connection_id = "test-connection-123"
+
+        httpx_mock.add_response(
+            method="GET",
+            status_code=200,
+            json={"id": connection_id, "name": "Test", "elementInstanceId": 1},
+        )
+        httpx_mock.add_response(method="POST", status_code=200, json={})
+
+        service.invoke_activity(
+            activity_metadata=simple_activity_metadata,
+            connection_id=connection_id,
+            activity_input={"body_field1": "x"},
+        )
+
+        sent_request = httpx_mock.get_requests()[1]
+        assert sent_request.headers[HEADER_ACTIVITY_JOB_ID] == "job-key-abc"
+
+    def test_invoke_activity_omits_job_id_header_when_unset(
+        self,
+        httpx_mock: HTTPXMock,
+        service: ConnectionsService,
+        simple_activity_metadata: ActivityMetadata,
+        monkeypatch: pytest.MonkeyPatch,
+    ) -> None:
+        """No job-id header is sent when UIPATH_JOB_KEY is not set."""
+        monkeypatch.delenv("UIPATH_JOB_KEY", raising=False)
+        connection_id = "test-connection-123"
+
+        httpx_mock.add_response(
+            method="GET",
+            status_code=200,
+            json={"id": connection_id, "name": "Test", "elementInstanceId": 1},
+        )
+        httpx_mock.add_response(method="POST", status_code=200, json={})
+
+        service.invoke_activity(
+            activity_metadata=simple_activity_metadata,
+            connection_id=connection_id,
+            activity_input={"body_field1": "x"},
+        )
+
+        sent_request = httpx_mock.get_requests()[1]
+        assert HEADER_ACTIVITY_JOB_ID not in sent_request.headers
+
     def test_invoke_activity_with_body_fields(
         self,
         httpx_mock: HTTPXMock,