fix: match guardrail tools by display_name in addition to tool name (#748)

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: ctiliescuuipath

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.9.18"
+version = "0.9.19"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath_langchain/agent/guardrails/guardrails_factory.py

@@ -315,8 +315,18 @@ def _compute_field_sources_for_guardrail(
 
     # Deterministic guardrails have one single tool
     if guardrail.selector.match_names and len(guardrail.selector.match_names) > 0:
+        match_name = guardrail.selector.match_names[0]
         matching_tool = next(
-            (t for t in tools if t.name == guardrail.selector.match_names[0]), None
+            (
+                t
+                for t in tools
+                if t.name == match_name
+                or (
+                    isinstance(t.metadata, dict)
+                    and t.metadata.get("display_name") == match_name
+                )
+            ),
+            None,
         )
 
         if matching_tool:

tests/agent/guardrails/test_guardrails_factory.py

@@ -1179,6 +1179,7 @@ def test_convert_with_nonexistent_tool_raises_error(self) -> None:
 
         mock_tool = Mock(spec=BaseTool)
         mock_tool.name = "different_tool"
+        mock_tool.metadata = None
 
         agent_guardrail = AgentCustomGuardrail.model_validate(
             {