uipath-llm-client 1.0.0__tar.gz → 1.0.1__tar.gz

{uipath_llm_client-1.0.0 → uipath_llm_client-1.0.1}/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

{uipath_llm_client-1.0.0 → uipath_llm_client-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-llm-client
-Version: 1.0.0
+Version: 1.0.1
 Summary: UiPath LLM Client
 Author-email: Cosmin Maria <cosmin.maria@uipath.com>, Dragos Bobolea <dragos.bobolea@uipath.com>
 License-File: LICENSE

{uipath_llm_client-1.0.0 → uipath_llm_client-1.0.1}/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"

{uipath_llm_client-1.0.0 → uipath_llm_client-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 @@ class LLMGatewayS2SAuth(Auth, metaclass=SingletonMeta):
         )
         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
 

{uipath_llm_client-1.0.0 → uipath_llm_client-1.0.1}/tests/core/test_base_client.py

@@ -506,9 +506,7 @@ class TestAgentHubAuthRefresh:
                 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")