fix: name change nits

Author: JoshParkSJ

src/uipath_langchain/agent/tools/tool_node.py

@@ -291,7 +291,7 @@ async def _afunc(state: AgentGraphState) -> OutputType:
 
 
 class ConversationalToolRunnableCallable(RunnableCallable):
-    """Preserves a reference to the underlying BaseTool for conversational HITL confirmation."""
+    """Preserves a reference to the underlying BaseTool for conversational tool confirmation."""
 
     def __init__(self, *, func: Any, afunc: Any, name: str, tool: BaseTool):
         super().__init__(func=func, afunc=afunc, name=name)

src/uipath_langchain/runtime/messages.py

@@ -59,7 +59,7 @@ def __init__(self, runtime_id: str, storage: UiPathRuntimeStorageProtocol | None
         self.runtime_id = runtime_id
         self.storage = storage
         self.current_message: AIMessageChunk | AIMessage
-        self.tool_confirmation_schemas: dict[str, Any] = {}
+        self.tools_requiring_confirmation: dict[str, Any] = {}
         self.seen_message_ids: set[str] = set()
         self._storage_lock = asyncio.Lock()
         self._citation_stream_processor = CitationStreamProcessor()
@@ -320,6 +320,7 @@ async def map_ai_message_chunk_to_events(
 
         events: list[UiPathConversationMessageEvent] = []
 
+        # For every new message_id, start a new message
         if message.id not in self.seen_message_ids:
             self.current_message = message
             self.seen_message_ids.add(message.id)
@@ -338,6 +339,7 @@ async def map_ai_message_chunk_to_events(
                                 self._chunk_to_message_event(message.id, chunk)
                             )
                     case "tool_call_chunk":
+                        # Accumulate the message chunk. Note that we assume no interweaving of AIMessage and AIMessageChunks for a given message.
                         # Skip the first chunk — it's already assigned as current_message above,
                         # so accumulating it with itself would duplicate fields via string concat
                         # (e.g. tool name "search_web" becomes "search_websearch_web").
@@ -431,9 +433,9 @@ async def map_current_message_to_start_tool_call_events(self):
 
                         tool_name = tool_call["name"]
                         require_confirmation = (
-                            tool_name in self.tool_confirmation_schemas
+                            tool_name in self.tools_requiring_confirmation
                         )
-                        input_schema = self.tool_confirmation_schemas.get(tool_name)
+                        input_schema = self.tools_requiring_confirmation.get(tool_name)
                         events.append(
                             self.map_tool_call_to_tool_call_start_event(
                                 self.current_message.id,

src/uipath_langchain/runtime/runtime.py

@@ -65,7 +65,7 @@ def __init__(
         self.entrypoint: str | None = entrypoint
         self.callbacks: list[BaseCallbackHandler] = callbacks or []
         self.chat = UiPathChatMessagesMapper(self.runtime_id, storage)
-        self.chat.tool_confirmation_schemas = self._get_tool_confirmation_info()
+        self.chat.tools_requiring_confirmation = self._get_tool_confirmation_info()
         self._middleware_node_names: set[str] = self._detect_middleware_nodes()
 
     async def execute(

tests/runtime/test_chat_message_mapper.py

@@ -2019,7 +2019,7 @@ async def test_confirmation_tool_has_requires_confirmation_metadata(self):
         storage = create_mock_storage()
         storage.get_value.return_value = {}
         mapper = UiPathChatMessagesMapper("test-runtime", storage)
-        mapper.tool_confirmation_schemas = {"confirm_tool": {}}
+        mapper.tools_requiring_confirmation = {"confirm_tool": {}}
 
         first_chunk = AIMessageChunk(
             content="",
@@ -2053,7 +2053,7 @@ async def test_normal_tool_has_no_confirmation_metadata(self):
         storage = create_mock_storage()
         storage.get_value.return_value = {}
         mapper = UiPathChatMessagesMapper("test-runtime", storage)
-        mapper.tool_confirmation_schemas = {"other_tool": {}}
+        mapper.tools_requiring_confirmation = {"other_tool": {}}
 
         first_chunk = AIMessageChunk(
             content="",
@@ -2086,7 +2086,7 @@ async def test_mixed_tools_only_confirmation_has_metadata(self):
         storage = create_mock_storage()
         storage.get_value.return_value = {}
         mapper = UiPathChatMessagesMapper("test-runtime", storage)
-        mapper.tool_confirmation_schemas = {"confirm_tool": {}}
+        mapper.tools_requiring_confirmation = {"confirm_tool": {}}
 
         first_chunk = AIMessageChunk(
             content="",

tests/runtime/test_tool_confirmation_discovery.py

@@ -70,20 +70,20 @@ def test_discovers_confirmation_tool_through_wrapper(self):
         graph = _compile_graph_with_wrapped_tools([_ConfirmableTool(), _NormalTool()])
         runtime = UiPathLangGraphRuntime(graph)
 
-        schemas = runtime.chat.tool_confirmation_schemas
+        schemas = runtime.chat.tools_requiring_confirmation
         assert "needs_confirmation" in schemas
         assert "no_confirmation" not in schemas
 
     def test_schema_contains_input_properties(self):
         graph = _compile_graph_with_wrapped_tools([_ConfirmableTool()])
         runtime = UiPathLangGraphRuntime(graph)
 
-        schema = runtime.chat.tool_confirmation_schemas["needs_confirmation"]
+        schema = runtime.chat.tools_requiring_confirmation["needs_confirmation"]
         assert "properties" in schema
         assert "query" in schema["properties"]
 
     def test_empty_when_no_confirmation_tools(self):
         graph = _compile_graph_with_wrapped_tools([_NormalTool()])
         runtime = UiPathLangGraphRuntime(graph)
 
-        assert runtime.chat.tool_confirmation_schemas == {}
+        assert runtime.chat.tools_requiring_confirmation == {}