feat: add confirmToolCall support to CAS chat bridge [JAR-8666]

Author: JoshParkSJ

.github/scripts/test_check_version_uniqueness.py

@@ -5,7 +5,6 @@
 import urllib.error
 from unittest import mock
 
-import pytest
 
 from check_version_uniqueness import (
     get_package_info,

packages/uipath-core/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-core"
-version = "0.5.11"
+version = "0.5.13"
 description = "UiPath Core abstractions"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

packages/uipath-core/src/uipath/core/chat/__init__.py

@@ -77,20 +77,6 @@
     UiPathConversationExchangeEvent,
     UiPathConversationExchangeStartEvent,
 )
-from .interrupt import (
-    InterruptTypeEnum,
-    UiPathConversationGenericInterruptEndEvent,
-    UiPathConversationGenericInterruptStartEvent,
-    UiPathConversationInterrupt,
-    UiPathConversationInterruptData,
-    UiPathConversationInterruptEndEvent,
-    UiPathConversationInterruptEvent,
-    UiPathConversationInterruptStartEvent,
-    UiPathConversationToolCallConfirmationEndValue,
-    UiPathConversationToolCallConfirmationInterruptEndEvent,
-    UiPathConversationToolCallConfirmationInterruptStartEvent,
-    UiPathConversationToolCallConfirmationValue,
-)
 from .message import (
     UiPathConversationMessage,
     UiPathConversationMessageData,
@@ -108,6 +94,7 @@
 )
 from .tool import (
     UiPathConversationToolCall,
+    UiPathConversationToolCallConfirmationEvent,
     UiPathConversationToolCallData,
     UiPathConversationToolCallEndEvent,
     UiPathConversationToolCallEvent,
@@ -141,19 +128,6 @@
     "UiPathConversationMessageEvent",
     "UiPathConversationMessageData",
     "UiPathConversationMessage",
-    # Interrupt
-    "InterruptTypeEnum",
-    "UiPathConversationInterruptStartEvent",
-    "UiPathConversationInterruptEndEvent",
-    "UiPathConversationInterruptEvent",
-    "UiPathConversationToolCallConfirmationValue",
-    "UiPathConversationToolCallConfirmationEndValue",
-    "UiPathConversationToolCallConfirmationInterruptStartEvent",
-    "UiPathConversationToolCallConfirmationInterruptEndEvent",
-    "UiPathConversationGenericInterruptStartEvent",
-    "UiPathConversationGenericInterruptEndEvent",
-    "UiPathConversationInterruptData",
-    "UiPathConversationInterrupt",
     # Content
     "UiPathConversationContentPartChunkEvent",
     "UiPathConversationContentPartStartEvent",
@@ -178,6 +152,7 @@
     # Tool
     "UiPathConversationToolCallStartEvent",
     "UiPathConversationToolCallEndEvent",
+    "UiPathConversationToolCallConfirmationEvent",
     "UiPathConversationToolCallEvent",
     "UiPathConversationToolCallResult",
     "UiPathConversationToolCallData",

packages/uipath-core/src/uipath/core/chat/interrupt.py

@@ -10,11 +10,6 @@
     UiPathConversationContentPartEvent,
 )
 from .error import UiPathConversationErrorEvent
-from .interrupt import (
-    UiPathConversationInterrupt,
-    UiPathConversationInterruptData,
-    UiPathConversationInterruptEvent,
-)
 from .tool import (
     UiPathConversationToolCall,
     UiPathConversationToolCallData,
@@ -53,7 +48,6 @@ class UiPathConversationMessageEvent(BaseModel):
         None, alias="contentPart"
     )
     tool_call: UiPathConversationToolCallEvent | None = Field(None, alias="toolCall")
-    interrupt: UiPathConversationInterruptEvent | None = None
     meta_event: dict[str, Any] | None = Field(None, alias="metaEvent")
     error: UiPathConversationErrorEvent | None = Field(None, alias="messageError")
 
@@ -68,7 +62,7 @@ class UiPathConversationMessageData(BaseModel):
         ..., alias="contentParts"
     )
     tool_calls: Sequence[UiPathConversationToolCallData] = Field(..., alias="toolCalls")
-    interrupts: Sequence[UiPathConversationInterruptData]
+    interrupts: Sequence[Any] = Field(default_factory=list)
 
     model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
 
@@ -86,6 +80,5 @@ class UiPathConversationMessage(UiPathConversationMessageData):
         ..., alias="contentParts"
     )
     tool_calls: Sequence[UiPathConversationToolCall] = Field(..., alias="toolCalls")
-    interrupts: Sequence[UiPathConversationInterrupt]
 
     model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)

packages/uipath-core/src/uipath/core/chat/message.py

@@ -25,6 +25,8 @@ class UiPathConversationToolCallStartEvent(BaseModel):
     timestamp: str | None = None
     input: dict[str, Any] | None = None
     metadata: dict[str, Any] | None = Field(None, alias="metaData")
+    require_confirmation: bool | None = Field(None, alias="requireConfirmation")
+    input_schema: Any | None = Field(None, alias="inputSchema")
 
     model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
 
@@ -41,6 +43,15 @@ class UiPathConversationToolCallEndEvent(BaseModel):
     model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
 
 
+class UiPathConversationToolCallConfirmationEvent(BaseModel):
+    """Represents a tool call confirmation event (approve/reject)."""
+
+    approved: bool
+    input: Any | None = None
+
+    model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
+
+
 class UiPathConversationToolCallEvent(BaseModel):
     """Encapsulates the data related to a tool call event."""
 
@@ -49,6 +60,9 @@ class UiPathConversationToolCallEvent(BaseModel):
         None, alias="startToolCall"
     )
     end: UiPathConversationToolCallEndEvent | None = Field(None, alias="endToolCall")
+    confirm: UiPathConversationToolCallConfirmationEvent | None = Field(
+        None, alias="confirmToolCall"
+    )
     meta_event: dict[str, Any] | None = Field(None, alias="metaEvent")
     error: UiPathConversationErrorEvent | None = Field(None, alias="toolCallError")
 

packages/uipath-core/src/uipath/core/chat/tool.py

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-platform"
-version = "0.1.26"
+version = "0.1.25"
 description = "HTTP client library for programmatic access to UiPath Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

packages/uipath-core/uv.lock

@@ -1115,17 +1115,12 @@ def _list_records_spec(
         start: Optional[int] = None,
         limit: Optional[int] = None,
     ) -> RequestSpec:
-        params: dict[str, Any] = {}
-        if start is not None:
-            params["start"] = start
-        if limit is not None:
-            params["limit"] = limit
         return RequestSpec(
             method="GET",
             endpoint=Endpoint(
                 f"datafabric_/api/EntityService/entity/{entity_key}/read"
             ),
-            params=params,
+            params=({"start": start, "limit": limit}),
         )
 
     def _query_entity_records_spec(

packages/uipath-platform/pyproject.toml

@@ -267,41 +267,6 @@ def test_retrieve_records_with_optional_fields(
                     limit=1,
                 )
 
-    def test_retrieve_records_without_start_and_limit(
-        self,
-        httpx_mock: HTTPXMock,
-        service: EntitiesService,
-        base_url: str,
-        org: str,
-        tenant: str,
-        version: str,
-    ) -> None:
-        entity_key = uuid.uuid4()
-        httpx_mock.add_response(
-            url=f"{base_url}{org}{tenant}/datafabric_/api/EntityService/entity/{str(entity_key)}/read",
-            status_code=200,
-            json={
-                "totalCount": 1,
-                "value": [
-                    {"Id": "12345", "name": "record_name", "integer_field": 10},
-                ],
-            },
-        )
-
-        records = service.list_records(entity_key=str(entity_key))
-
-        sent_request = httpx_mock.get_request()
-        if sent_request is None:
-            raise Exception("No request was sent")
-
-        # Verify no start or limit query params are sent
-        assert "start" not in str(sent_request.url.params)
-        assert "limit" not in str(sent_request.url.params)
-
-        assert isinstance(records, list)
-        assert len(records) == 1
-        assert records[0].id == "12345"
-
     @pytest.mark.parametrize(
         "sql_query",
         [