Map 400 to 401 on s2s (#1)

Author: cosminacho

.github/workflows/cd-langchain.yml

@@ -2,9 +2,10 @@ name: Publish uipath-langchain-client to PyPI
 
 on:
   push:
-    tags:
-      # Publish on any tag starting with `langchain-v`, e.g., langchain-v1.0.0
-      - langchain-v*
+    branches:
+      - main
+    paths:
+      - 'packages/uipath_langchain_client/src/uipath_langchain_client/__version__.py'
   workflow_dispatch:
 
 jobs:

.github/workflows/cd.yml

@@ -2,9 +2,10 @@ name: Publish uipath-llm-client to PyPI
 
 on:
   push:
-    tags:
-      # Publish on any tag starting with a `v`, e.g., v1.0.0
-      - v*
+    branches:
+      - main
+    paths:
+      - 'src/uipath_llm_client/__version__.py'
   workflow_dispatch:
 
 jobs:

CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes to `uipath_llm_client` (core package) will be documented in this file.
 
+## [1.0.1] - 2025-01-30
+
+### Bug Fixes
+- Map 400 Bad requests on S2S to 401 Unauthorized for better readability
+
 ## [1.0.0] - 2025-01-30
 
 ### Official Release

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.0"
+__version__ = "1.0.1"

src/uipath_llm_client/settings/llmgateway/auth.py

@@ -1,11 +1,11 @@
 from collections.abc import Generator
 
-from httpx import Auth, Client, HTTPStatusError, Request, Response
+from httpx import Auth, Client, Request, Response
 
 from uipath_llm_client.settings.llmgateway.settings import LLMGatewayBaseSettings
 from uipath_llm_client.settings.llmgateway.utils import LLMGatewayEndpoints
 from uipath_llm_client.settings.utils import SingletonMeta
-from uipath_llm_client.utils.exceptions import UiPathAPIError
+from uipath_llm_client.utils.exceptions import UiPathAPIError, UiPathAuthenticationError
 
 
 class LLMGatewayS2SAuth(Auth, metaclass=SingletonMeta):
@@ -39,10 +39,15 @@ def get_llmgw_token_header(
         )
         with Client() as http_client:
             response = http_client.post(url_get_token, data=token_credentials)
-            try:
-                response.raise_for_status()
-            except HTTPStatusError as e:
-                raise UiPathAPIError.from_response(e.response)
+            if response.is_client_error:
+                raise UiPathAuthenticationError(
+                    message="Failed to authenticate with LLM Gateway, invalid credentials",
+                    request=response.request,
+                    response=response,
+                    body=response.json(),
+                )
+            elif response.is_error:
+                raise UiPathAPIError.from_response(response)
             llmgw_token_header = response.json().get("access_token")
             return llmgw_token_header
 

tests/core/test_base_client.py

@@ -506,9 +506,7 @@ def test_auth_flow_refreshes_on_401(self, agenthub_env_vars):
                 settings = AgentHubSettings()
 
                 # Mock _get_access_token to return a new token on refresh
-                with patch.object(
-                    AgentHubAuth, "_get_access_token", return_value="initial-token"
-                ):
+                with patch.object(AgentHubAuth, "_get_access_token", return_value="initial-token"):
                     auth = AgentHubAuth(settings=settings)
 
                 request = Request("GET", "https://example.com")