send exchange end

Author: GabrielVasilescu04

src/uipath/runtime/chat/protocol.py

@@ -1,6 +1,6 @@
 """Abstract conversation bridge interface."""
 
-from typing import Protocol
+from typing import Protocol, Any
 
 from uipath.core.chat import (
     UiPathConversationMessageEvent,
@@ -48,6 +48,6 @@ async def emit_exchange_end_event(self) -> None:
         """Send an exchange end event."""
         ...
 
-    async def wait_for_resume(self) -> None:
+    async def wait_for_resume(self) -> dict[str, Any]:
         """Wait for the interrupt_end event to be received."""
         ...

src/uipath/runtime/chat/runtime.py

@@ -103,6 +103,8 @@ async def stream(
                 else:
                     yield event
 
+            await self.chat_bridge.emit_exchange_end_event()
+
     async def get_schema(self) -> UiPathRuntimeSchema:
         """Get schema from the delegate runtime."""
         return await self.delegate.get_schema()

tests/test_chat_runtime.py

@@ -146,7 +146,7 @@ async def stream(
                     status=UiPathRuntimeStatus.SUSPENDED,
                     trigger=UiPathResumeTrigger(
                         trigger_type=UiPathResumeTriggerType.API,
-                        trigger_data={"action": "confirm_tool_call"},
+                        payload={"action": "confirm_tool_call"},
                     ),
                 )
                 return