uipath 2.1.132__tar.gz → 2.1.133__tar.gz

{uipath-2.1.132 → uipath-2.1.133}/.github/workflows/test.yml

@@ -7,6 +7,7 @@ jobs:
   test:
     name: Test
     runs-on: ${{ matrix.os }}
+    timeout-minutes: 10
     strategy:
       matrix:
         python-version: ["3.10", "3.11", "3.12", "3.13"]

{uipath-2.1.132 → uipath-2.1.133}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.1.132
+Version: 2.1.133
 Summary: Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools.
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python

{uipath-2.1.132 → uipath-2.1.133}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.1.132"
+version = "2.1.133"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
@@ -72,6 +72,7 @@ dev = [
   "toml>=0.10.2",
   "inflection>=0.5.1",
   "types-toml>=0.10.8",
+  "pytest-timeout>=2.4.0",
 ]
 
 [tool.hatch.build.targets.wheel]

{uipath-2.1.132 → uipath-2.1.133}/src/uipath/utils/_endpoints_manager.py

@@ -1,196 +1,184 @@
-import logging
-import os
-from enum import Enum
-from typing import Optional
-
-import httpx
-
-from uipath._utils._ssl_context import get_httpx_client_kwargs
-
-loggger = logging.getLogger(__name__)
-
-
-class UiPathEndpoints(Enum):
-    AH_NORMALIZED_COMPLETION_ENDPOINT = "agenthub_/llm/api/chat/completions"
-    AH_PASSTHROUGH_COMPLETION_ENDPOINT = "agenthub_/llm/openai/deployments/{model}/chat/completions?api-version={api_version}"
-    AH_EMBEDDING_ENDPOINT = (
-        "agenthub_/llm/openai/deployments/{model}/embeddings?api-version={api_version}"
-    )
-    AH_CAPABILITIES_ENDPOINT = "agenthub_/llm/api/capabilities"
-
-    OR_NORMALIZED_COMPLETION_ENDPOINT = "orchestrator_/llm/api/chat/completions"
-    OR_PASSTHROUGH_COMPLETION_ENDPOINT = "orchestrator_/llm/openai/deployments/{model}/chat/completions?api-version={api_version}"
-    OR_EMBEDDING_ENDPOINT = "orchestrator_/llm/openai/deployments/{model}/embeddings?api-version={api_version}"
-    OR_CAPABILITIES_ENDPOINT = "orchestrator_/llm/api/capabilities"
-
-    LG_NORMALIZED_COMPLETION_ENDPOINT = "llmgateway_/api/chat/completions"
-    LG_PASSTHROUGH_COMPLETION_ENDPOINT = "llmgateway_/openai/deployments/{model}/chat/completions?api-version={api_version}"
-    LG_EMBEDDING_ENDPOINT = (
-        "llmgateway_/openai/deployments/{model}/embeddings?api-version={api_version}"
-    )
-
-
-class EndpointManager:
-    """Manages and caches the UiPath endpoints.
-    This class provides functionality to determine which UiPath endpoints to use based on
-    the availability of AgentHub and Orchestrator. It checks for capabilities and caches
-    the results to avoid repeated network calls.
-
-    The endpoint selection follows a fallback order:
-    1. AgentHub (if available)
-    2. Orchestrator (if available)
-    3. LLMGateway (default fallback)
-
-    Environment Variable Override:
-    The fallback behavior can be bypassed using the UIPATH_LLM_SERVICE environment variable:
-    - 'agenthub' or 'ah': Force use of AgentHub endpoints (skips capability checks)
-    - 'orchestrator' or 'or': Force use of Orchestrator endpoints (skips capability checks)
-    - 'llmgateway' or 'gateway': Force use of LLMGateway endpoints (skips capability checks)
-
-    Class Attributes:
-        _base_url (str): The base URL for UiPath services, retrieved from the UIPATH_URL
-                         environment variable.
-        _agenthub_available (Optional[bool]): Cached result of AgentHub availability check.
-        _orchestrator_available (Optional[bool]): Cached result of Orchestrator availability check.
-
-    Methods:
-        is_agenthub_available(): Checks if AgentHub is available, caching the result.
-        is_orchestrator_available(): Checks if Orchestrator is available, caching the result.
-        get_passthrough_endpoint(): Returns the appropriate passthrough completion endpoint.
-        get_normalized_endpoint(): Returns the appropriate normalized completion endpoint.
-        get_embeddings_endpoint(): Returns the appropriate embeddings endpoint.
-    All endpoint methods automatically select the best available endpoint using the fallback order,
-    unless overridden by the UIPATH_LLM_SERVICE environment variable.
-    """  # noqa: D205
-
-    _base_url = os.getenv("UIPATH_URL", "")
-    _agenthub_available: Optional[bool] = None
-    _orchestrator_available: Optional[bool] = None
-
-    @classmethod
-    def is_agenthub_available(cls) -> bool:
-        """Check if AgentHub is available and cache the result."""
-        if cls._agenthub_available is None:
-            cls._agenthub_available = cls._check_agenthub()
-        return cls._agenthub_available
-
-    @classmethod
-    def is_orchestrator_available(cls) -> bool:
-        """Check if Orchestrator is available and cache the result."""
-        if cls._orchestrator_available is None:
-            cls._orchestrator_available = cls._check_orchestrator()
-        return cls._orchestrator_available
-
-    @classmethod
-    def _check_capabilities(cls, endpoint: UiPathEndpoints, service_name: str) -> bool:
-        """Perform the actual check for service capabilities.
-
-        Args:
-            endpoint: The capabilities endpoint to check
-            service_name: Human-readable service name for logging
-
-        Returns:
-            bool: True if the service is available and has valid capabilities
-        """
-        try:
-            with httpx.Client(**get_httpx_client_kwargs()) as http_client:
-                base_url = os.getenv("UIPATH_URL", "")
-                capabilities_url = f"{base_url.rstrip('/')}/{endpoint.value}"
-                loggger.debug(
-                    f"Checking {service_name} capabilities at {capabilities_url}"
-                )
-                response = http_client.get(capabilities_url)
-
-                if response.status_code != 200:
-                    return False
-
-                capabilities = response.json()
-
-                # Validate structure and required fields
-                if not isinstance(capabilities, dict) or "version" not in capabilities:
-                    return False
-
-                return True
-
-        except Exception as e:
-            loggger.error(
-                f"Error checking {service_name} capabilities: {e}", exc_info=True
-            )
-            return False
-
-    @classmethod
-    def _check_agenthub(cls) -> bool:
-        """Perform the actual check for AgentHub capabilities."""
-        return cls._check_capabilities(
-            UiPathEndpoints.AH_CAPABILITIES_ENDPOINT, "AgentHub"
-        )
-
-    @classmethod
-    def _check_orchestrator(cls) -> bool:
-        """Perform the actual check for Orchestrator capabilities."""
-        return cls._check_capabilities(
-            UiPathEndpoints.OR_CAPABILITIES_ENDPOINT, "Orchestrator"
-        )
-
-    @classmethod
-    def _select_endpoint(
-        cls, ah: UiPathEndpoints, orc: UiPathEndpoints, gw: UiPathEndpoints
-    ) -> str:
-        """Select an endpoint based on UIPATH_LLM_SERVICE override or capability checks."""
-        service_override = os.getenv("UIPATH_LLM_SERVICE", "").lower()
-
-        if service_override in ("agenthub", "ah"):
-            return ah.value
-        if service_override in ("orchestrator", "or"):
-            return orc.value
-        if service_override in ("llmgateway", "gateway"):
-            return gw.value
-
-        # Determine fallback order based on environment hints
-        hdens_env = os.getenv("HDENS_ENV", "").lower()
-
-        # Default order: AgentHub -> Orchestrator
-        check_order = [
-            ("ah", ah, cls.is_agenthub_available),
-            ("orc", orc, cls.is_orchestrator_available),
-        ]
-
-        # Prioritize Orchestrator if HDENS_ENV is 'sf'
-        # Note: The default order already prioritizes AgentHub
-        if hdens_env == "sf":
-            check_order.reverse()
-
-        # Execute fallback checks in the determined order
-        for _, endpoint, is_available in check_order:
-            if is_available():
-                return endpoint.value
-
-        # Final fallback to LLMGateway
-        return gw.value
-
-    @classmethod
-    def get_passthrough_endpoint(cls) -> str:
-        """Get the passthrough completion endpoint."""
-        return cls._select_endpoint(
-            UiPathEndpoints.AH_PASSTHROUGH_COMPLETION_ENDPOINT,
-            UiPathEndpoints.OR_PASSTHROUGH_COMPLETION_ENDPOINT,
-            UiPathEndpoints.LG_PASSTHROUGH_COMPLETION_ENDPOINT,
-        )
-
-    @classmethod
-    def get_normalized_endpoint(cls) -> str:
-        """Get the normalized completion endpoint."""
-        return cls._select_endpoint(
-            UiPathEndpoints.AH_NORMALIZED_COMPLETION_ENDPOINT,
-            UiPathEndpoints.OR_NORMALIZED_COMPLETION_ENDPOINT,
-            UiPathEndpoints.LG_NORMALIZED_COMPLETION_ENDPOINT,
-        )
-
-    @classmethod
-    def get_embeddings_endpoint(cls) -> str:
-        """Get the embeddings endpoint."""
-        return cls._select_endpoint(
-            UiPathEndpoints.AH_EMBEDDING_ENDPOINT,
-            UiPathEndpoints.OR_EMBEDDING_ENDPOINT,
-            UiPathEndpoints.LG_EMBEDDING_ENDPOINT,
-        )
+import logging
+import os
+from enum import Enum
+from typing import Optional
+
+import httpx
+
+from uipath._utils._ssl_context import get_httpx_client_kwargs
+
+loggger = logging.getLogger(__name__)
+
+
+class UiPathEndpoints(Enum):
+    AH_NORMALIZED_COMPLETION_ENDPOINT = "agenthub_/llm/api/chat/completions"
+    AH_PASSTHROUGH_COMPLETION_ENDPOINT = "agenthub_/llm/openai/deployments/{model}/chat/completions?api-version={api_version}"
+    AH_EMBEDDING_ENDPOINT = (
+        "agenthub_/llm/openai/deployments/{model}/embeddings?api-version={api_version}"
+    )
+    AH_CAPABILITIES_ENDPOINT = "agenthub_/llm/api/capabilities"
+
+    OR_NORMALIZED_COMPLETION_ENDPOINT = "orchestrator_/llm/api/chat/completions"
+    OR_PASSTHROUGH_COMPLETION_ENDPOINT = "orchestrator_/llm/openai/deployments/{model}/chat/completions?api-version={api_version}"
+    OR_EMBEDDING_ENDPOINT = "orchestrator_/llm/openai/deployments/{model}/embeddings?api-version={api_version}"
+    OR_CAPABILITIES_ENDPOINT = "orchestrator_/llm/api/capabilities"
+
+
+class EndpointManager:
+    """Manages and caches the UiPath endpoints.
+    This class provides functionality to determine which UiPath endpoints to use based on
+    the availability of AgentHub and Orchestrator. It checks for capabilities and caches
+    the results to avoid repeated network calls.
+
+    The endpoint selection follows a fallback order:
+    1. AgentHub (if available)
+    2. Orchestrator (if available)
+
+    Environment Variable Override:
+    The fallback behavior can be bypassed using the UIPATH_LLM_SERVICE environment variable:
+    - 'agenthub' or 'ah': Force use of AgentHub endpoints (skips capability checks)
+    - 'orchestrator' or 'or': Force use of Orchestrator endpoints (skips capability checks)
+
+    Class Attributes:
+        _base_url (str): The base URL for UiPath services, retrieved from the UIPATH_URL
+                         environment variable.
+        _agenthub_available (Optional[bool]): Cached result of AgentHub availability check.
+        _orchestrator_available (Optional[bool]): Cached result of Orchestrator availability check.
+
+    Methods:
+        is_agenthub_available(): Checks if AgentHub is available, caching the result.
+        is_orchestrator_available(): Checks if Orchestrator is available, caching the result.
+        get_passthrough_endpoint(): Returns the appropriate passthrough completion endpoint.
+        get_normalized_endpoint(): Returns the appropriate normalized completion endpoint.
+        get_embeddings_endpoint(): Returns the appropriate embeddings endpoint.
+    All endpoint methods automatically select the best available endpoint using the fallback order,
+    unless overridden by the UIPATH_LLM_SERVICE environment variable.
+    """  # noqa: D205
+
+    _base_url = os.getenv("UIPATH_URL", "")
+    _agenthub_available: Optional[bool] = None
+    _orchestrator_available: Optional[bool] = None
+
+    @classmethod
+    def is_agenthub_available(cls) -> bool:
+        """Check if AgentHub is available and cache the result."""
+        if cls._agenthub_available is None:
+            cls._agenthub_available = cls._check_agenthub()
+        return cls._agenthub_available
+
+    @classmethod
+    def is_orchestrator_available(cls) -> bool:
+        """Check if Orchestrator is available and cache the result."""
+        if cls._orchestrator_available is None:
+            cls._orchestrator_available = cls._check_orchestrator()
+        return cls._orchestrator_available
+
+    @classmethod
+    def _check_capabilities(cls, endpoint: UiPathEndpoints, service_name: str) -> bool:
+        """Perform the actual check for service capabilities.
+
+        Args:
+            endpoint: The capabilities endpoint to check
+            service_name: Human-readable service name for logging
+
+        Returns:
+            bool: True if the service is available and has valid capabilities
+        """
+        try:
+            with httpx.Client(**get_httpx_client_kwargs()) as http_client:
+                base_url = os.getenv("UIPATH_URL", "")
+                capabilities_url = f"{base_url.rstrip('/')}/{endpoint.value}"
+                loggger.debug(
+                    f"Checking {service_name} capabilities at {capabilities_url}"
+                )
+                response = http_client.get(capabilities_url)
+
+                if response.status_code != 200:
+                    return False
+
+                capabilities = response.json()
+
+                # Validate structure and required fields
+                if not isinstance(capabilities, dict) or "version" not in capabilities:
+                    return False
+
+                return True
+
+        except Exception as e:
+            loggger.error(
+                f"Error checking {service_name} capabilities: {e}", exc_info=True
+            )
+            return False
+
+    @classmethod
+    def _check_agenthub(cls) -> bool:
+        """Perform the actual check for AgentHub capabilities."""
+        return cls._check_capabilities(
+            UiPathEndpoints.AH_CAPABILITIES_ENDPOINT, "AgentHub"
+        )
+
+    @classmethod
+    def _check_orchestrator(cls) -> bool:
+        """Perform the actual check for Orchestrator capabilities."""
+        return cls._check_capabilities(
+            UiPathEndpoints.OR_CAPABILITIES_ENDPOINT, "Orchestrator"
+        )
+
+    @classmethod
+    def _select_endpoint(cls, ah: UiPathEndpoints, orc: UiPathEndpoints) -> str:
+        """Select an endpoint based on UIPATH_LLM_SERVICE override or capability checks."""
+        service_override = os.getenv("UIPATH_LLM_SERVICE", "").lower()
+
+        if service_override in ("agenthub", "ah"):
+            return ah.value
+        if service_override in ("orchestrator", "or"):
+            return orc.value
+
+        # Determine fallback order based on environment hints
+        hdens_env = os.getenv("HDENS_ENV", "").lower()
+
+        # Default order: AgentHub -> Orchestrator
+        check_order = [
+            ("ah", ah, cls.is_agenthub_available),
+            ("orc", orc, cls.is_orchestrator_available),
+        ]
+
+        # Prioritize Orchestrator if HDENS_ENV is 'sf'
+        # Note: The default order already prioritizes AgentHub
+        if hdens_env == "sf":
+            check_order.reverse()
+
+        # Execute fallback checks in the determined order
+        for _, endpoint, is_available in check_order:
+            if is_available():
+                return endpoint.value
+
+        url = os.getenv("UIPATH_URL", "")
+        if ".uipath.com" in url:
+            return ah.value
+        else:
+            return orc.value
+
+    @classmethod
+    def get_passthrough_endpoint(cls) -> str:
+        """Get the passthrough completion endpoint."""
+        return cls._select_endpoint(
+            UiPathEndpoints.AH_PASSTHROUGH_COMPLETION_ENDPOINT,
+            UiPathEndpoints.OR_PASSTHROUGH_COMPLETION_ENDPOINT,
+        )
+
+    @classmethod
+    def get_normalized_endpoint(cls) -> str:
+        """Get the normalized completion endpoint."""
+        return cls._select_endpoint(
+            UiPathEndpoints.AH_NORMALIZED_COMPLETION_ENDPOINT,
+            UiPathEndpoints.OR_NORMALIZED_COMPLETION_ENDPOINT,
+        )
+
+    @classmethod
+    def get_embeddings_endpoint(cls) -> str:
+        """Get the embeddings endpoint."""
+        return cls._select_endpoint(
+            UiPathEndpoints.AH_EMBEDDING_ENDPOINT,
+            UiPathEndpoints.OR_EMBEDDING_ENDPOINT,
+        )

{uipath-2.1.132 → uipath-2.1.133}/tests/cli/eval/mocks/test_input_mocker.py

@@ -67,7 +67,8 @@ async def test_generate_llm_input_with_model_settings(
     )
 
     httpx_mock.add_response(
-        url="https://example.com/api/chat/completions?api-version=2024-08-01-preview",
+        url="https://example.com/llm/api/chat/completions"
+        "?api-version=2024-08-01-preview",
         status_code=200,
         json={
             "role": "assistant",

{uipath-2.1.132 → uipath-2.1.133}/tests/cli/eval/mocks/test_mocks.py

@@ -179,7 +179,8 @@ def test_llm_mockable_sync(httpx_mock: HTTPXMock, monkeypatch: MonkeyPatch):
     )
 
     httpx_mock.add_response(
-        url="https://example.com/api/chat/completions?api-version=2024-08-01-preview",
+        url="https://example.com/llm/api/chat/completions"
+        "?api-version=2024-08-01-preview",
         status_code=200,
         json={
             "id": "response-id",
@@ -211,7 +212,8 @@ def test_llm_mockable_sync(httpx_mock: HTTPXMock, monkeypatch: MonkeyPatch):
     with pytest.raises(NotImplementedError):
         assert foofoo()
     httpx_mock.add_response(
-        url="https://example.com/api/chat/completions?api-version=2024-08-01-preview",
+        url="https://example.com/llm/api/chat/completions"
+        "?api-version=2024-08-01-preview",
         status_code=200,
         json={},
     )
@@ -220,6 +222,7 @@ def test_llm_mockable_sync(httpx_mock: HTTPXMock, monkeypatch: MonkeyPatch):
 
 
 @pytest.mark.asyncio
+@pytest.mark.httpx_mock(assert_all_responses_were_requested=False)
 async def test_llm_mockable_async(httpx_mock: HTTPXMock, monkeypatch: MonkeyPatch):
     monkeypatch.setenv("UIPATH_URL", "https://example.com")
     monkeypatch.setenv("UIPATH_ACCESS_TOKEN", "1234567890")
@@ -251,8 +254,21 @@ async def test_llm_mockable_async(httpx_mock: HTTPXMock, monkeypatch: MonkeyPatc
     evaluation = LegacyEvaluationItem(**evaluation_item)
     assert isinstance(evaluation.mocking_strategy, LLMMockingStrategy)
 
+    # Mock capability checks
+    httpx_mock.add_response(
+        url="https://example.com/agenthub_/llm/api/capabilities",
+        status_code=200,
+        json={},
+    )
+    httpx_mock.add_response(
+        url="https://example.com/orchestrator_/llm/api/capabilities",
+        status_code=200,
+        json={},
+    )
+
     httpx_mock.add_response(
-        url="https://example.com/api/chat/completions?api-version=2024-08-01-preview",
+        url="https://example.com/llm/api/chat/completions"
+        "?api-version=2024-08-01-preview",
         status_code=200,
         json={
             "id": "response-id",
@@ -285,7 +301,8 @@ async def test_llm_mockable_async(httpx_mock: HTTPXMock, monkeypatch: MonkeyPatc
         assert await foofoo()
 
     httpx_mock.add_response(
-        url="https://example.com/api/chat/completions?api-version=2024-08-01-preview",
+        url="https://example.com/llm/api/chat/completions"
+        "?api-version=2024-08-01-preview",
         status_code=200,
         json={},
     )

{uipath-2.1.132 → uipath-2.1.133}/tests/sdk/services/test_uipath_llm_integration.py

@@ -5,10 +5,7 @@ import pytest
 
 from uipath._config import Config
 from uipath._execution_context import ExecutionContext
-from uipath._services.llm_gateway_service import (
-    ChatModels,
-    UiPathLlmChatService,
-)
+from uipath._services.llm_gateway_service import ChatModels, UiPathLlmChatService
 from uipath.models.llm_gateway import (
     AutoToolChoice,
     SpecificToolChoice,
@@ -278,7 +275,7 @@ class TestUiPathLLMServiceMocked:
 
         # Verify the correct endpoint and payload
         args, kwargs = mock_request.call_args
-        assert "/llmgateway_/api/chat/completions" in args[1]
+        assert "/orchestrator_/llm/api/chat/completions" in args[1]
         assert kwargs["json"]["messages"] == messages
         assert kwargs["json"]["max_tokens"] == 50
         assert kwargs["json"]["temperature"] == 0

{uipath-2.1.132 → uipath-2.1.133}/uv.lock

@@ -2635,6 +2635,18 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/5a/cc/06253936f4a7fa2e0f48dfe6d851d9c56df896a9ab09ac019d70b760619c/pytest_mock-3.15.1-py3-none-any.whl", hash = "sha256:0a25e2eb88fe5168d535041d09a4529a188176ae608a6d249ee65abc0949630d", size = 10095, upload-time = "2025-09-16T16:37:25.734Z" },
 ]
 
+[[package]]
+name = "pytest-timeout"
+version = "2.4.0"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "pytest" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/ac/82/4c9ecabab13363e72d880f2fb504c5f750433b2b6f16e99f4ec21ada284c/pytest_timeout-2.4.0.tar.gz", hash = "sha256:7e68e90b01f9eff71332b25001f85c75495fc4e3a836701876183c4bcfd0540a", size = 17973, upload-time = "2025-05-05T19:44:34.99Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/fa/b6/3127540ecdf1464a00e5a01ee60a1b09175f6913f0644ac748494d9c4b21/pytest_timeout-2.4.0-py3-none-any.whl", hash = "sha256:c42667e5cdadb151aeb5b26d114aff6bdf5a907f176a007a30b940d3d865b5c2", size = 14382, upload-time = "2025-05-05T19:44:33.502Z" },
+]
+
 [[package]]
 name = "pytest-trio"
 version = "0.8.0"
@@ -3047,7 +3059,7 @@ wheels = [
 
 [[package]]
 name = "uipath"
-version = "2.1.132"
+version = "2.1.133"
 source = { editable = "." }
 dependencies = [
     { name = "azure-monitor-opentelemetry" },
@@ -3087,6 +3099,7 @@ dev = [
     { name = "pytest-cov" },
     { name = "pytest-httpx" },
     { name = "pytest-mock" },
+    { name = "pytest-timeout" },
     { name = "pytest-trio" },
     { name = "ruff" },
     { name = "rust-just" },
@@ -3134,6 +3147,7 @@ dev = [
     { name = "pytest-cov", specifier = ">=4.1.0" },
     { name = "pytest-httpx", specifier = ">=0.35.0" },
     { name = "pytest-mock", specifier = ">=3.11.1" },
+    { name = "pytest-timeout", specifier = ">=2.4.0" },
     { name = "pytest-trio", specifier = ">=0.8.0" },
     { name = "ruff", specifier = ">=0.9.4" },
     { name = "rust-just", specifier = ">=1.39.0" },