@@ -91,7 +91,7 @@ async def test_llm_success_pre_and_post(
9191 guardrail = MagicMock (spec = BuiltInValidatorGuardrail )
9292 guardrail .name = "Example"
9393 _patch_uipath (
94- monkeypatch , result = GuardrailValidationResultType .PASSED , reason = ""
94+ monkeypatch , result = GuardrailValidationResultType .PASSED , reason = "validation passed "
9595 )
9696 node_name , node = create_llm_guardrail_node (
9797 guardrail = guardrail ,
@@ -103,7 +103,7 @@ async def test_llm_success_pre_and_post(
103103 state = AgentGuardrailsGraphState (messages = [HumanMessage ("payload" )])
104104 cmd = await node (state )
105105 assert cmd .goto == "ok"
106- assert cmd .update == {"inner_state" : {"guardrail_validation_result" : None }}
106+ assert cmd .update == {"inner_state" : {"guardrail_validation_result" : "validation passed" }}
107107
108108 @pytest .mark .asyncio
109109 @pytest .mark .parametrize (
@@ -162,7 +162,7 @@ async def test_agent_init_success_pre_and_post(
162162 guardrail = MagicMock (spec = BuiltInValidatorGuardrail )
163163 guardrail .name = "Example"
164164 fake = _patch_uipath (
165- monkeypatch , result = GuardrailValidationResultType .PASSED , reason = ""
165+ monkeypatch , result = GuardrailValidationResultType .PASSED , reason = "validation passed "
166166 )
167167
168168 node_name , node = create_agent_init_guardrail_node (
@@ -176,7 +176,7 @@ async def test_agent_init_success_pre_and_post(
176176 state = AgentGuardrailsGraphState (messages = [HumanMessage ("payload" )])
177177 cmd = await node (state )
178178 assert cmd .goto == "ok"
179- assert cmd .update == {"inner_state" : {"guardrail_validation_result" : None }}
179+ assert cmd .update == {"inner_state" : {"guardrail_validation_result" : "validation passed" }}
180180 assert fake .guardrails .last_text == "payload"
181181
182182 @pytest .mark .asyncio
@@ -239,7 +239,7 @@ async def test_agent_terminate_success_pre_and_post(
239239 guardrail = MagicMock (spec = BuiltInValidatorGuardrail )
240240 guardrail .name = "Example"
241241 fake = _patch_uipath (
242- monkeypatch , result = GuardrailValidationResultType .PASSED , reason = ""
242+ monkeypatch , result = GuardrailValidationResultType .PASSED , reason = "validation passed "
243243 )
244244
245245 node_name , node = create_agent_terminate_guardrail_node (
@@ -257,7 +257,7 @@ async def test_agent_terminate_success_pre_and_post(
257257 )
258258 cmd = await node (state )
259259 assert cmd .goto == "ok"
260- assert cmd .update == {"inner_state" : {"guardrail_validation_result" : None }}
260+ assert cmd .update == {"inner_state" : {"guardrail_validation_result" : "validation passed" }}
261261 assert fake .guardrails .last_text == str (agent_result )
262262
263263 @pytest .mark .asyncio
@@ -348,15 +348,15 @@ async def test_tool_success_pre_and_post(
348348 )
349349 cmd = await node (state )
350350 assert cmd .goto == "ok"
351- assert cmd .update == {"inner_state" : {"guardrail_validation_result" : None }}
351+ assert cmd .update == {"inner_state" : {"guardrail_validation_result" : "" }}
352352 assert json .loads (fake .guardrails .last_text or "{}" ) == {"x" : 1 }
353353 else :
354354 state = AgentGuardrailsGraphState (
355355 messages = [ToolMessage (content = "tool output" , tool_call_id = "call_1" )]
356356 )
357357 cmd = await node (state )
358358 assert cmd .goto == "ok"
359- assert cmd .update == {"inner_state" : {"guardrail_validation_result" : None }}
359+ assert cmd .update == {"inner_state" : {"guardrail_validation_result" : "" }}
360360 assert fake .guardrails .last_text == "tool output"
361361
362362 @pytest .mark .asyncio
@@ -509,7 +509,7 @@ async def test_evaluate_builtin_guardrail(self, monkeypatch):
509509 )
510510
511511 fake = _patch_uipath (
512- monkeypatch , result = GuardrailValidationResultType .PASSED , reason = ""
512+ monkeypatch , result = GuardrailValidationResultType .PASSED , reason = "validation passed "
513513 )
514514
515515 guardrail = MagicMock (spec = BuiltInValidatorGuardrail )
@@ -532,12 +532,12 @@ def test_create_validation_command_success(self):
532532
533533 result = GuardrailValidationResult (
534534 result = GuardrailValidationResultType .PASSED ,
535- reason = "" ,
535+ reason = "validation passed " ,
536536 )
537537 command = _create_validation_command (result , "success_node" , "failure_node" )
538538
539539 assert command .goto == "success_node"
540- assert command .update == {"inner_state" : {"guardrail_validation_result" : None }}
540+ assert command .update == {"inner_state" : {"guardrail_validation_result" : "validation passed" }}
541541
542542 def test_create_validation_command_failure (self ):
543543 """Test validation command creation for failed validation."""
0 commit comments