fixes

Author: cosminacho

packages/uipath_langchain_client/src/uipath_langchain_client/utils.py

@@ -0,0 +1,33 @@
+from uipath.llm_client.utils.exceptions import (
+    UiPathAPIError,
+    UiPathAuthenticationError,
+    UiPathBadRequestError,
+    UiPathConflictError,
+    UiPathGatewayTimeoutError,
+    UiPathInternalServerError,
+    UiPathNotFoundError,
+    UiPathPermissionDeniedError,
+    UiPathRateLimitError,
+    UiPathRequestTooLargeError,
+    UiPathServiceUnavailableError,
+    UiPathTooManyRequestsError,
+    UiPathUnprocessableEntityError,
+)
+from uipath.llm_client.utils.retry import RetryConfig
+
+__all__ = [
+    "RetryConfig",
+    "UiPathAPIError",
+    "UiPathBadRequestError",
+    "UiPathAuthenticationError",
+    "UiPathPermissionDeniedError",
+    "UiPathNotFoundError",
+    "UiPathConflictError",
+    "UiPathRequestTooLargeError",
+    "UiPathUnprocessableEntityError",
+    "UiPathRateLimitError",
+    "UiPathInternalServerError",
+    "UiPathServiceUnavailableError",
+    "UiPathGatewayTimeoutError",
+    "UiPathTooManyRequestsError",
+]

src/uipath/llm_client/utils/retry.py

@@ -49,13 +49,6 @@
 
 from uipath.llm_client.utils.exceptions import UiPathAPIError, UiPathRateLimitError
 
-__all__ = [
-    "RetryConfig",
-    "RetryableHTTPTransport",
-    "RetryableAsyncHTTPTransport",
-]
-
-
 # Default retry configuration values
 _DEFAULT_RETRY_ON_EXCEPTIONS: tuple[type[Exception], ...] = (UiPathRateLimitError,)
 _DEFAULT_INITIAL_DELAY: float = 2.0
@@ -147,7 +140,7 @@ class RetryConfig(TypedDict):
         ... }
     """
 
-    retry_on_exceptions: NotRequired[tuple[type[UiPathAPIError], ...]]
+    retry_on_exceptions: NotRequired[tuple[type[Exception], ...]]
     initial_delay: NotRequired[float]
     max_delay: NotRequired[float]
     exp_base: NotRequired[float]
@@ -327,3 +320,10 @@ async def _send() -> Response:
                 return await _send()
         except UiPathAPIError as e:
             return e.response
+
+
+__all__ = [
+    "RetryConfig",
+    "RetryableHTTPTransport",
+    "RetryableAsyncHTTPTransport",
+]