feat(err_license): testing

Author: giuliastf

.github/workflows/integration_tests_powershell.yml

@@ -14,6 +14,7 @@
     UiPathBaseRuntime,
     UiPathErrorCategory,
     UiPathRuntimeResult,
+    UiPathRuntimeError,
 )
 
 from ..._utils import _instrument_traceable_attributes
@@ -113,6 +114,15 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
                         self._pretty_print(stream_chunk)
                         final_chunk = stream_chunk
 
+                    # 
+                    raise UiPathRuntimeError(
+                        code="LICENSE_NOT_AVAILABLE",
+                        title=body.get('title', 'License Not Available'),
+                        detail=body.get('detail', 'License not available for this service'),
+                        category=UiPathErrorCategory.DEPLOYMENT,
+                        prefix="UIPATH"
+                    )
+                
                     self.context.output = self._extract_graph_result(final_chunk, graph)
                 else:
                     # Execute the graph normally at runtime
@@ -135,6 +145,8 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
         except Exception as e:
             if isinstance(e, LangGraphRuntimeError):
                 raise
+            if isinstance(e, UiPathRuntimeError):
+                raise
 
             detail = f"Error: {str(e)}"
 

src/uipath_langchain/_cli/_runtime/_runtime.py

@@ -4,7 +4,7 @@
 import os
 import time
 from typing import Any, Dict, List, Mapping, Optional
-
+from contextlib import suppress
 import httpx
 import openai
 from langchain_core.embeddings import Embeddings
@@ -25,7 +25,7 @@
     get_uipath_token_header,
 )
 from uipath_langchain._utils._sleep_policy import before_sleep_log
-
+from uipath._cli._runtime._contracts import UiPathRuntimeError, UiPathErrorCategory
 
 def get_from_uipath_url():
     url = os.getenv("UIPATH_URL")
@@ -234,7 +234,7 @@ async def _arequest(
             # Handle HTTP errors and map them to OpenAI exceptions
             try:
                 response.raise_for_status()
-            except httpx.HTTPStatusError as err:
+            except httpx.HTTPStatusError as err: 
                 if self.logger:
                     self.logger.error(
                         "Error querying LLM: %s (%s)",
@@ -348,6 +348,24 @@ def _make_status_error(
             return openai.AuthenticationError(err_msg, response=response, body=data)
 
         if response.status_code == 403:
+            # Check if this is a license-specific error
+            if isinstance(body, dict):
+                title = body.get('title', '').lower()
+                detail = body.get('detail', '').lower()
+                
+                # Smart detection of license errors
+                if ('license' in title or 'license' in detail) and \
+                ('not available' in title or 'not available' in detail or
+                    'unavailable' in title or 'unavailable' in detail):
+                    # Raise UiPathRuntimeError for license issues
+                    raise UiPathRuntimeError(
+                        code="LICENSE_NOT_AVAILABLE",
+                        title=body.get('title', 'License Not Available'),
+                        detail=body.get('detail', 'License not available for this service'),
+                        category=UiPathErrorCategory.DEPLOYMENT,
+                        prefix="UIPATH"
+                    )
+            
             return openai.PermissionDeniedError(err_msg, response=response, body=data)
 
         if response.status_code == 404: