fix: tool confirmation for coded agents (create_agent)

Three fixes for coded agent tool confirmation:

1. hitl.py: @requires_approval now stamps REQUIRE_CONVERSATIONAL_CONFIRMATION
   metadata so the runtime can discover confirmation tools.

2. runtime.py: _get_tool_confirmation_info handles both graph shapes —
   UiPathToolNode (low-code, bound.tool) and LangGraph ToolNode
   (create_agent, bound.tools_by_name).

3. messages.py: seed chunk accumulator with an empty AIMessageChunk instead
   of the incoming message. Prevents self-merge on the first streaming chunk
   which doubled tool names (e.g. search_websearch_web) and broke the
   confirmation set lookup.

Author: JoshParkSJ

src/uipath_langchain/chat/hitl.py

@@ -221,6 +221,10 @@ def wrapper(**tool_args: Any) -> Any:
                 return_direct=return_direct,
             )
 
+        if result.metadata is None:
+            result.metadata = {}
+        result.metadata[REQUIRE_CONVERSATIONAL_CONFIRMATION] = True
+
         _created_tool.append(result)
         return result
 

src/uipath_langchain/runtime/messages.py

@@ -323,7 +323,7 @@ async def map_ai_message_chunk_to_events(
 
         # For every new message_id, start a new message
         if message.id not in self.seen_message_ids:
-            self.current_message = message
+            self.current_message = AIMessageChunk(content="", id=message.id)
             self.seen_message_ids.add(message.id)
             self._citation_stream_processor = CitationStreamProcessor()
             events.append(self.map_to_message_start_event(message.id))

src/uipath_langchain/runtime/runtime.py

@@ -494,20 +494,34 @@ def _get_tool_confirmation_info(self) -> tuple[set[str], dict[str, Any]]:
         """Single pass over graph nodes to collect confirmation tool names and schemas."""
         names: set[str] = set()
         schemas: dict[str, Any] = {}
+
+        def _record(tool: Any, fallback_name: str) -> None:
+            metadata = getattr(tool, "metadata", None) or {}
+            if not metadata.get(REQUIRE_CONVERSATIONAL_CONFIRMATION):
+                return
+            tool_name = getattr(tool, "name", fallback_name)
+            names.add(tool_name)
+            tool_call_schema = getattr(tool, "tool_call_schema", None)
+            if tool_call_schema is not None:
+                schemas[tool_name] = tool_call_schema.model_json_schema()
+            else:
+                schemas[tool_name] = {}
+
         for node_name, node_spec in self.graph.nodes.items():
-            # langgraph's processing node.bound -> runnable.tool -> baseTool (if tool node)
-            tool = getattr(getattr(node_spec, "bound", None), "tool", None)
-            if tool is None:
+            bound = getattr(node_spec, "bound", None)
+            if bound is None:
                 continue
-            metadata = getattr(tool, "metadata", None) or {}
-            if metadata.get(REQUIRE_CONVERSATIONAL_CONFIRMATION):
-                tool_name = getattr(tool, "name", node_name)
-                names.add(tool_name)
-                tool_call_schema = getattr(tool, "tool_call_schema", None)
-                if tool_call_schema is not None:
-                    schemas[tool_name] = tool_call_schema.model_json_schema()
-                else:
-                    schemas[tool_name] = {}
+            # Low-code: single-tool UiPathToolNode
+            tool = getattr(bound, "tool", None)
+            if tool is not None:
+                _record(tool, node_name)
+                continue
+            # Coded: multi-tool ToolNode (create_agent)
+            tools_by_name = getattr(bound, "tools_by_name", None)
+            if isinstance(tools_by_name, dict):
+                for fallback_name, tool in tools_by_name.items():
+                    _record(tool, fallback_name)
+
         return names, schemas
 
     def _is_middleware_node(self, node_name: str) -> bool: