uipath-llm-client 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

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.2"

uipath_llm_client/httpx_client.py

@@ -193,7 +193,7 @@ class UiPathHttpxClient(Client):
             Response with patched raise_for_status() that raises UiPath exceptions.
         """
         if self._freeze_base_url:
-            request.url = URL(self.base_url)
+            request.url = URL(str(self.base_url).rstrip("/"))
         request.headers[self._streaming_header] = str(stream).lower()
         response = super().send(request, stream=stream, **kwargs)
         return patch_raise_for_status(response)
@@ -313,7 +313,7 @@ class UiPathHttpxAsyncClient(AsyncClient):
             Response with patched raise_for_status() that raises UiPath exceptions.
         """
         if self._freeze_base_url:
-            request.url = URL(self.base_url)
+            request.url = URL(str(self.base_url).rstrip("/"))
         request.headers[self._streaming_header] = str(stream).lower()
         response = await super().send(request, stream=stream, **kwargs)
         return patch_raise_for_status(response)

uipath_llm_client/settings/agenthub/settings.py

@@ -8,8 +8,8 @@ from dotenv import load_dotenv
 from pydantic import Field, SecretStr, model_validator
 from pydantic_settings import SettingsConfigDict
 from uipath._cli._auth._auth_service import AuthService
+from uipath.utils import EndpointManager
 
-from uipath_llm_client.settings.agenthub.utils import AgentHubEndpoints
 from uipath_llm_client.settings.base import UiPathAPIConfig, UiPathBaseSettings
 
 
@@ -99,13 +99,16 @@ class AgentHubBaseSettings(UiPathBaseSettings):
         api_config: UiPathAPIConfig | None = None,
     ) -> str:
         """Build the base URL for API requests."""
-        if api_config is not None and api_config.client_type == "normalized":
-            url = f"{self.base_url}/{AgentHubEndpoints.NORMALIZED_ENDPOINT.value.format(api_type=api_config.api_type)}"
+        assert model_name is not None
+        assert api_config is not None
+        if api_config.client_type == "normalized" and api_config.api_type == "completions":
+            url = f"{self.base_url}/{EndpointManager.get_normalized_endpoint()}"
+        elif api_config.client_type == "passthrough" and api_config.api_type == "embeddings":
+            assert api_config.api_version is not None
+            url = f"{self.base_url}/{EndpointManager.get_embeddings_endpoint().format(model=model_name, api_version=api_config.api_version)}"
         else:
-            assert api_config is not None
-            assert api_config.api_type is not None
             assert api_config.vendor_type is not None
-            url = f"{self.base_url}/{AgentHubEndpoints.PASSTHROUGH_ENDPOINT.value.format(model=model_name, vendor=api_config.vendor_type, api_type=api_config.api_type)}"
+            url = f"{self.base_url}/{EndpointManager.get_vendor_endpoint().format(model=model_name, vendor=api_config.vendor_type)}"
         return url
 
     @override

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.dist-info → uipath_llm_client-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-llm-client
-Version: 1.0.0
+Version: 1.0.2
 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.dist-info → uipath_llm_client-1.0.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 uipath_llm_client/__init__.py,sha256=K7pMecLOblh6GWVBkoOalC-ho5ozlw3WG6loiJNzSFo,2363
-uipath_llm_client/__version__.py,sha256=i-3gbkh2cjfZHmW7rn8P-fH6H8bytkCMMV7yK6auN6g,135
-uipath_llm_client/httpx_client.py,sha256=IscG5Qu5E60FooEktOmNVIjjBGBJoA6APNCabmSFSnA,12718
+uipath_llm_client/__version__.py,sha256=1nCovE7VmxdZ4cjSkRaZ6kp9WIZbQeEyQ6owNU7xYec,135
+uipath_llm_client/httpx_client.py,sha256=F7fDgbTOImu4zWehUAxlqXhOysnGyfmB_58wm53YLlE,12752
 uipath_llm_client/clients/anthropic/__init__.py,sha256=0KxbVGKSVWwC89PYPM7IXkP3qhJa2J0zyW6xsr0efWM,546
 uipath_llm_client/clients/anthropic/client.py,sha256=UjrXzsgOOgoIRjo7SiLThWtXAJIEH-BHuPy3umNcmC0,17967
 uipath_llm_client/clients/google/__init__.py,sha256=_JRzU_RdUnWvpPdeAOpWjUb-ccN2TBcReXCFF34Ouyw,100
@@ -13,16 +13,15 @@ uipath_llm_client/settings/base.py,sha256=rVloAQCvbhY-bBdKEOH5WuB-pay91BZ9gpNXT0
 uipath_llm_client/settings/utils.py,sha256=2Rmr7RiQjRchSDHg46tuFRV5scEDphIl-NMkmyPScwo,466
 uipath_llm_client/settings/agenthub/__init__.py,sha256=3aBfGa4YkfeylXGBALF-BFktakYRlideFajhvfq8YlQ,1538
 uipath_llm_client/settings/agenthub/auth.py,sha256=vWb0nipR3pITNxVQmQARCsxx5ZfUAfWS_SR5w0A7dEo,1591
-uipath_llm_client/settings/agenthub/settings.py,sha256=-tNiyg_AeKKNa-m2T2DG596eoCUayJrB-6KUswHncgo,5764
-uipath_llm_client/settings/agenthub/utils.py,sha256=JVpGC8SKNshCz_MCp-wcjPGTpGWTdP0TPXcPs5kDEH4,471
+uipath_llm_client/settings/agenthub/settings.py,sha256=Dr9hg6PeR2pjYamq-ERH88oXbjUXL6x6_6KpqkdUQck,5944
 uipath_llm_client/settings/llmgateway/__init__.py,sha256=QMViUqELA0LVle_WDjNivxxNaFSpr_X3yfJRsCEdZ4I,1525
-uipath_llm_client/settings/llmgateway/auth.py,sha256=BxZZtLvHCiFaNouS7eyYfO9GHKUllJMnVtxaPPHvClA,2441
+uipath_llm_client/settings/llmgateway/auth.py,sha256=JzN6czbLOQfLF0CMuHNHJjzUOSzKfr_tVrgeC5tXSxI,2710
 uipath_llm_client/settings/llmgateway/settings.py,sha256=ZhKk94UP2yJ6MNR8mSRWmoqYB_pRv9SKmV3OWVh0CWY,4733
 uipath_llm_client/settings/llmgateway/utils.py,sha256=gqyZXulcG0ifvlaxqdOGkf24jOtjCAalBme26BmoV3Q,527
 uipath_llm_client/utils/exceptions.py,sha256=IcHPQIf-Jzir_rG7gaD7ByD25Thcl-QpAjnC6xvZ0qQ,5973
 uipath_llm_client/utils/logging.py,sha256=pmC00QAt4LVXFb6PCHQDPoX7iLE9N2GeEytuAkUiQ1w,4641
 uipath_llm_client/utils/retry.py,sha256=cvM-OBEvga6mufCQ5G9rmxVzd33wOJOO6UGOHs9IWgk,8603
-uipath_llm_client-1.0.0.dist-info/METADATA,sha256=BcAbGn1ItN0S6gQI9ZwuSRPXgJ9fgmoaSTl-5snnJZU,22964
-uipath_llm_client-1.0.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-uipath_llm_client-1.0.0.dist-info/licenses/LICENSE,sha256=r4HMU0pdnVI7BlSYgDUCszn7IrCXJiN8o3yt2U1aOpc,375
-uipath_llm_client-1.0.0.dist-info/RECORD,,
+uipath_llm_client-1.0.2.dist-info/METADATA,sha256=7-_NIxmbpl2cBfqPDLIa8_ae_ZvzDNPE7qDJE83sSSY,22964
+uipath_llm_client-1.0.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+uipath_llm_client-1.0.2.dist-info/licenses/LICENSE,sha256=r4HMU0pdnVI7BlSYgDUCszn7IrCXJiN8o3yt2U1aOpc,375
+uipath_llm_client-1.0.2.dist-info/RECORD,,

uipath_llm_client/settings/agenthub/utils.py

@@ -1,14 +0,0 @@
-"""Utility classes for AgentHub settings."""
-
-from enum import StrEnum
-
-
-class AgentHubEndpoints(StrEnum):
-    """API endpoint paths for UiPath AgentHub.
-
-    Normalized endpoints provide a consistent API across all providers.
-    Passthrough endpoints expose vendor-specific APIs with formats for vendor/model.
-    """
-
-    NORMALIZED_ENDPOINT = "agenthub_/llm/api/chat/{api_type}"
-    PASSTHROUGH_ENDPOINT = "agenthub_/llm/raw/vendor/{vendor}/model/{model}/{api_type}"