fix: tests

Author: JoshParkSJ

tests/agent/tools/test_tool_node.py

@@ -1,5 +1,6 @@
 """Tests for tool_node.py module."""
 
+import json
 from typing import Any, Dict
 from unittest.mock import patch
 
@@ -533,7 +534,7 @@ def test_cancelled_returns_cancelled_message(
         assert isinstance(result, dict)
         msg = result["messages"][0]
         assert isinstance(msg, ToolMessage)
-        assert msg.content == CANCELLED_MESSAGE
+        assert msg.content == json.dumps({"meta": CANCELLED_MESSAGE})
 
     @patch(
         "uipath_langchain.chat.hitl.request_approval",
@@ -568,7 +569,6 @@ def test_approved_modified_args_injects_meta(
         assert result is not None
         assert isinstance(result, dict)
         msg = result["messages"][0]
-        import json
 
         assert isinstance(msg.content, str)
         wrapped = json.loads(msg.content)
@@ -588,7 +588,7 @@ async def test_async_cancelled(
         assert result is not None
         assert isinstance(result, dict)
         msg = result["messages"][0]
-        assert msg.content == CANCELLED_MESSAGE
+        assert msg.content == json.dumps({"meta": CANCELLED_MESSAGE})
 
     @patch(
         "uipath_langchain.chat.hitl.request_approval",
@@ -605,7 +605,6 @@ async def test_async_approved_modified_args(
         assert result is not None
         assert isinstance(result, dict)
         msg = result["messages"][0]
-        import json
 
         assert isinstance(msg.content, str)
         wrapped = json.loads(msg.content)

tests/chat/test_hitl.py

@@ -1,5 +1,6 @@
 """Tests for hitl.py module."""
 
+import json
 from typing import Any
 from unittest.mock import patch
 
@@ -60,7 +61,7 @@ def test_cancelled_returns_tool_message(self, mock_approval):
         assert isinstance(result, ConfirmationResult)
         assert result.cancelled is not None
         assert isinstance(result.cancelled, ToolMessage)
-        assert result.cancelled.content == CANCELLED_MESSAGE
+        assert result.cancelled.content == json.dumps({"meta": CANCELLED_MESSAGE})
         assert result.cancelled.name == "mock_tool"
         assert result.cancelled.tool_call_id == "call_1"
         assert result.args_modified is False