andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-27 17:18:13 +00:00
Code Issues Actions 5 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?") result = agent.kickoff(messages="Top 10 best players in the world?")
assert len(guardrail_events["started"]) == 2 # Guardrail may be called 2 or 3 times depending on LLM response
assert len(guardrail_events["completed"]) == 2 assert len(guardrail_events["started"]) >= 2
assert not guardrail_events["completed"][0].success assert len(guardrail_events["completed"]) >= 2
assert guardrail_events["completed"][1].success # 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 ( assert (
"Here are the top 10 best soccer players in the world, focusing exclusively on Brazilian players" "Here are the top 10 best soccer players in the world, focusing exclusively on Brazilian players"
in result.raw in result.raw
@@ -436,6 +438,7 @@ def test_guardrail_reached_attempt_limit():
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])
@pytest.mark.requires_local_services
def test_agent_output_when_guardrail_returns_base_model(): def test_agent_output_when_guardrail_returns_base_model():
class Player(BaseModel): class Player(BaseModel):
name: str name: str