Skip to content

Commit 5c71ccb

Browse files
author
Valentina Bojan
committed
fix: fix lint
1 parent d35f2e6 commit 5c71ccb

2 files changed

Lines changed: 14 additions & 12 deletions

File tree

src/uipath_langchain/agent/guardrails/guardrail_nodes.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,9 @@ def _create_validation_command(
108108
if guardrail_result.result == GuardrailValidationResultType.PASSED:
109109
return Command(
110110
goto=success_node,
111-
update={"inner_state": {"guardrail_validation_result": guardrail_result.reason}},
111+
update={
112+
"inner_state": {"guardrail_validation_result": guardrail_result.reason}
113+
},
112114
)
113115

114116
if guardrail_result.result == GuardrailValidationResultType.VALIDATION_FAILED:

tests/agent/guardrails/test_guardrail_nodes.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)