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

test: add assertions for have_forced_answer state (#1815)

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2024-12-29 15:52:48 +00:00
parent 625a807d29
commit 0d8f7114ba
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/agent_test.py
View File

@@ -466,6 +466,8 @@ def test_agent_custom_max_iterations():
tools=[get_final_answer],
)
assert private_mock.call_count == 2
# Verify that have_forced_answer was set to True after max iterations
assert agent.agent_executor.have_forced_answer == True, "have_forced_answer should be True after max iterations"
@pytest.mark.vcr(filter_headers=["authorization"])
@@ -566,6 +568,8 @@ def test_agent_moved_on_after_max_iterations():
tools=[get_final_answer],
)
assert output == "The final answer is 42."
# Verify that have_forced_answer was set to True after max iterations
assert agent.agent_executor.have_forced_answer == True, "have_forced_answer should be True after max iterations"
@pytest.mark.vcr(filter_headers=["authorization"])