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

final pass

Browse Source
This commit is contained in:
João Moura
2024-09-16 06:16:48 -03:00
parent 8ca0821718
commit e5ff0363d9
95 changed files with 13936 additions and 17097 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/task_test.py
View File

@@ -236,7 +236,7 @@ def test_output_pydantic_sequential():
crew = Crew(agents=[scorer], tasks=[task], process=Process.sequential)
result = crew.kickoff()
assert isinstance(result.pydantic, ScoreOutput)
assert result.to_dict() == {"score": 5}
assert result.to_dict() == {"score": 4}
@pytest.mark.vcr(filter_headers=["authorization"])
@@ -404,8 +404,8 @@ def test_output_json_dict_hierarchical():
manager_llm="gpt-4o",
)
result = crew.kickoff()
assert {"score": 4} == result.json_dict
assert result.to_dict() == {"score": 4}
assert {"score": 5} == result.json_dict
assert result.to_dict() == {"score": 5}
@pytest.mark.vcr(filter_headers=["authorization"])
@@ -443,7 +443,7 @@ def test_output_pydantic_to_another_task():
assert isinstance(
pydantic_result, ScoreOutput
), "Expected pydantic result to be of type ScoreOutput"
assert 4 == pydantic_result.score
assert 3 == pydantic_result.score
@pytest.mark.vcr(filter_headers=["authorization"])
@@ -474,7 +474,7 @@ def test_output_json_to_another_task():
crew = Crew(agents=[scorer], tasks=[task1, task2])
result = crew.kickoff()
assert '{"score": 5}' == result.json
assert '{"score": 4}' == result.json
@pytest.mark.vcr(filter_headers=["authorization"])