Updated included headers on LLMGW settings (#18)

Author: cosminacho

CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes to `uipath_llm_client` (core package) will be documented in this file.
 
+## [1.0.13] - 2026-02-05
+
+### Fix
+- Fixed headers on llmgw settings
+
 ## [1.0.12] - 2026-02-05
 
 ### Fix

packages/uipath_langchain_client/CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes to `uipath_langchain_client` will be documented in this file.
 
+## [1.0.13] - 2026-02-05
+
+### Fix
+- Bump version
+
 ## [1.0.12] - 2026-02-05
 
 ### Fix

packages/uipath_langchain_client/pyproject.toml

@@ -6,7 +6,7 @@ readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "langchain>=1.2.7",
-    "uipath-llm-client>=1.0.10",
+    "uipath-llm-client>=1.0.13",
 ]
 
 [project.optional-dependencies]

packages/uipath_langchain_client/src/uipath_langchain_client/__version__.py

@@ -1,3 +1,3 @@
 __title__ = "UiPath LangChain Client"
 __description__ = "A Python client for interacting with UiPath's LLM services via LangChain."
-__version__ = "1.0.12"
+__version__ = "1.0.13"

src/uipath_llm_client/__version__.py

@@ -1,3 +1,3 @@
 __titile__ = "UiPath LLM Client"
 __description__ = "A Python client for interacting with UiPath's LLM services."
-__version__ = "1.0.12"
+__version__ = "1.0.13"

src/uipath_llm_client/settings/agenthub/settings.py

@@ -35,10 +35,10 @@ class AgentHubBaseSettings(UiPathBaseSettings):
 
     model_config = SettingsConfigDict(validate_by_alias=True)
 
-    # Environment configuration
+    # Environment configuration: alpha, staging, cloud
     environment: str | None = Field(default=None, validation_alias="UIPATH_ENVIRONMENT")
 
-    # Authentication fields
+    # Authentication fields - retrieved from uipath auth as well
     access_token: SecretStr | None = Field(default=None, validation_alias="UIPATH_ACCESS_TOKEN")
     base_url: str | None = Field(default=None, validation_alias="UIPATH_URL")
     tenant_id: str | None = Field(default=None, validation_alias="UIPATH_TENANT_ID")

src/uipath_llm_client/settings/llmgateway/settings.py

@@ -42,9 +42,18 @@ class LLMGatewayBaseSettings(UiPathBaseSettings):
     requesting_product: str = Field(default=..., validation_alias="LLMGW_REQUESTING_PRODUCT")
     requesting_feature: str = Field(default=..., validation_alias="LLMGW_REQUESTING_FEATURE")
 
-    # Optional fields - used for tracking and billing
-    user_id: str | None = Field(default=None, validation_alias="LLMGW_SEMANTIC_USER_ID")
-    action_id: str | None = Field(default=None, validation_alias="LLMGW_ACTION_ID")
+    # Optional fields - situational usecase
+    user_id: str | None = Field(
+        default=None, validation_alias="LLMGW_SEMANTIC_USER_ID"
+    )  # used to apply governance rules
+    action_id: str | None = Field(
+        default=None, validation_alias="LLMGW_ACTION_ID"
+    )  # used to track the action
+    operation_code: str | None = Field(
+        default=None, validation_alias="LLMGW_OPERATION_CODE"
+    )  # used to correctly identify byo models
+
+    # additional headers for the request (e.g. X-UiPath-LlmGateway-Telemetry-SessionId)
     additional_headers: Mapping[str, str] = Field(
         default_factory=dict, validation_alias="LLMGW_ADDITIONAL_HEADERS"
     )
@@ -82,11 +91,15 @@ def build_auth_headers(
         headers = {
             "X-UiPath-LlmGateway-RequestingProduct": self.requesting_product,
             "X-UiPath-LlmGateway-RequestingFeature": self.requesting_feature,
+            "X-UiPath-Internal-AccountId": self.org_id,
+            "X-UiPath-Internal-TenantId": self.tenant_id,
         }
         if self.user_id:
             headers["X-UiPath-LlmGateway-UserId"] = self.user_id
         if self.action_id:
             headers["X-UiPath-LlmGateway-ActionId"] = self.action_id
+        if self.operation_code:
+            headers["X-UiPath-LlmGateway-OperationCode"] = self.operation_code
         if self.additional_headers:
             headers.update(self.additional_headers)
         return headers