chore: handle possible edge case

Author: norman-le

packages/uipath/src/uipath/_cli/_chat/_bridge.py

@@ -498,10 +498,13 @@ def _resolve_or_store_resume(
             if not future.done():
                 future.set_result(item)
             else:
+                # Future was cancelled or already resolved — store the payload
+                # so a subsequent wait_for_resume() can still find it.
                 logger.warning(
-                    f"Duplicate resume for tool_call_id={tool_call_id} — "
-                    "future already resolved, ignoring."
+                    f"Resume for tool_call_id={tool_call_id} — "
+                    "future already done, storing as fallback."
                 )
+                self._tool_resume_results[tool_call_id] = item
         else:
             if tool_call_id in self._tool_resume_results:
                 logger.warning(

packages/uipath/tests/cli/chat/test_bridge.py

@@ -683,7 +683,9 @@ def _make_end_event(self, tool_call_id: str, output: Any = None) -> dict[str, An
                     "toolCall": {
                         "toolCallId": tool_call_id,
                         "endToolCall": {
-                            "output": output or {"result": "ok"},
+                            "output": output
+                            if output is not None
+                            else {"result": "ok"},
                             "isError": False,
                         },
                     },
@@ -915,7 +917,9 @@ def _make_end_event(self, tool_call_id: str, output: Any = None) -> dict[str, An
                     "toolCall": {
                         "toolCallId": tool_call_id,
                         "endToolCall": {
-                            "output": output or {"result": "ok"},
+                            "output": output
+                            if output is not None
+                            else {"result": "ok"},
                             "isError": False,
                         },
                     },