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

fix: test_code_execution_flag_adds_code_tool_upon_kickoff

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-07-04 20:49:34 -03:00
parent 72af5717ab
commit a69133e5d5
2 changed files with 7 additions and 1888 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1885
tests/cassettes/test_code_execution_flag_adds_code_tool_upon_kickoff.yaml
View File

File diff suppressed because it is too large Load Diff

10
tests/crew_test.py
View File

@@ -936,6 +936,7 @@ def test_task_with_no_arguments():
assert result == "75"
@pytest.mark.vcr(filter_headers=["authorization"])
def test_code_execution_flag_adds_code_tool_upon_kickoff():
from crewai_tools import CodeInterpreterTool
@@ -954,9 +955,12 @@ def test_code_execution_flag_adds_code_tool_upon_kickoff():
)
crew = Crew(agents=[programmer], tasks=[task])
crew.kickoff()
assert len(programmer.tools) == 1
assert programmer.tools[0].__class__ == CodeInterpreterTool
with patch.object(Agent, "execute_task") as executor:
executor.return_value = "ok"
crew.kickoff()
assert len(programmer.tools) == 1
assert programmer.tools[0].__class__ == CodeInterpreterTool
@pytest.mark.vcr(filter_headers=["authorization"])