feat: address feedback [AL-228, AL-241]

Author: Valentina Bojan

src/uipath_langchain/agent/guardrails/actions/escalate_action.py

@@ -16,7 +16,7 @@
 from ...exceptions import AgentTerminationException
 from ...react.types import AgentGuardrailsGraphState
 from ..types import ExecutionStage
-from ..utils import get_message_text
+from ..utils import get_message_content
 from .base_action import GuardrailAction, GuardrailActionNode
 
 
@@ -378,7 +378,7 @@ def _extract_llm_escalation_content(
         if isinstance(last_message, ToolMessage):
             return last_message.content
 
-        content = get_message_text(last_message)
+        content = get_message_content(last_message)
         return json.dumps(content) if content else ""
 
     # For AI messages, process tool calls if present
@@ -396,14 +396,14 @@ def _extract_llm_escalation_content(
                 ):
                     content_list.append(json.dumps(args["content"]))
 
-        message_content = get_message_text(last_message)
+        message_content = get_message_content(last_message)
         if message_content:
             content_list.append(message_content)
 
         return json.dumps(content_list)
 
     # Fallback for other message types
-    return get_message_text(last_message)
+    return get_message_content(last_message)
 
 
 def _extract_agent_escalation_content(

src/uipath_langchain/agent/guardrails/guardrail_nodes.py

@@ -12,7 +12,7 @@
 )
 
 from uipath_langchain.agent.guardrails.types import ExecutionStage
-from uipath_langchain.agent.guardrails.utils import get_message_text
+from uipath_langchain.agent.guardrails.utils import get_message_content
 from uipath_langchain.agent.react.types import AgentGuardrailsGraphState
 
 logger = logging.getLogger(__name__)
@@ -64,7 +64,7 @@ def create_llm_guardrail_node(
     def _payload_generator(state: AgentGuardrailsGraphState) -> str:
         if not state.messages:
             return ""
-        return get_message_text(state.messages[-1])
+        return get_message_content(state.messages[-1])
 
     return _create_guardrail_node(
         guardrail,
@@ -85,7 +85,7 @@ def create_agent_init_guardrail_node(
     def _payload_generator(state: AgentGuardrailsGraphState) -> str:
         if not state.messages:
             return ""
-        return get_message_text(state.messages[-1])
+        return get_message_content(state.messages[-1])
 
     return _create_guardrail_node(
         guardrail,
@@ -173,7 +173,7 @@ def _payload_generator(state: AgentGuardrailsGraphState) -> str:
                     if args is not None:
                         return json.dumps(args)
 
-        return get_message_text(state.messages[-1])
+        return get_message_content(state.messages[-1])
 
     return _create_guardrail_node(
         guardrail,

src/uipath_langchain/agent/guardrails/utils.py

@@ -1,7 +1,7 @@
 from langchain_core.messages import AnyMessage, HumanMessage, SystemMessage
 
 
-def get_message_text(msg: AnyMessage) -> str:
+def get_message_content(msg: AnyMessage) -> str:
     if isinstance(msg, (HumanMessage, SystemMessage)):
         return msg.content if isinstance(msg.content, str) else str(msg.content)
     return str(getattr(msg, "content", "")) if hasattr(msg, "content") else ""

src/uipath_langchain/agent/react/agent.py

@@ -14,7 +14,8 @@
 from .guardrails.guardrails_subgraph import (
     create_agent_init_guardrails_subgraph,
     create_agent_terminate_guardrails_subgraph,
-    create_tools_guardrails_subgraph, create_llm_guardrails_subgraph,
+    create_llm_guardrails_subgraph,
+    create_tools_guardrails_subgraph,
 )
 from .init_node import (
     create_init_node,
@@ -99,7 +100,7 @@ def create_agent(
         builder.add_node(tool_name, tool_node)
 
     terminate_with_guardrails_subgraph = create_agent_terminate_guardrails_subgraph(
-        (AgentGraphNode.TERMINATE, terminate_node),
+        (AgentGraphNode.GUARDED_TERMINATE, terminate_node),
         guardrails,
     )
     builder.add_node(AgentGraphNode.TERMINATE, terminate_with_guardrails_subgraph)