andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

test: fix lite agent guardrail test assertions

Browse Source
This commit is contained in:
Greyson LaLonde
2025-09-26 23:15:37 -04:00
parent 58b72c3948
commit 1d1f5f455c
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/agents/test_lite_agent.py
View File

@@ -351,10 +351,12 @@ def test_guardrail_is_called_using_string():
result = agent.kickoff(messages="Top 10 best players in the world?")
assert len(guardrail_events["started"]) == 2
assert len(guardrail_events["completed"]) == 2
assert not guardrail_events["completed"][0].success
assert guardrail_events["completed"][1].success
# Guardrail may be called 2 or 3 times depending on LLM response
assert len(guardrail_events["started"]) >= 2
assert len(guardrail_events["completed"]) >= 2
# At least one should fail and the last one should succeed
assert any(not event.success for event in guardrail_events["completed"][:-1])
assert guardrail_events["completed"][-1].success
assert (
"Here are the top 10 best soccer players in the world, focusing exclusively on Brazilian players"
in result.raw
@@ -436,6 +438,7 @@ def test_guardrail_reached_attempt_limit():
@pytest.mark.vcr(filter_headers=["authorization"])
@pytest.mark.requires_local_services
def test_agent_output_when_guardrail_returns_base_model():
class Player(BaseModel):
name: str