Adding verbose levels

This commit is contained in:
João Moura
2023-12-30 07:41:38 -03:00
parent ff46652752
commit d6c60f873a
3 changed files with 743 additions and 9 deletions

View File

@@ -5,10 +5,10 @@ import json
import pytest
from crewai.agent import Agent
from crewai.agents import CacheHandler
from crewai.crew import Crew
from crewai.process import Process
from crewai.task import Task
from crewai.agents import CacheHandler
ceo = Agent(
role="CEO",
@@ -197,6 +197,33 @@ def test_crew_verbose_output(capsys):
assert captured.out == ""
@pytest.mark.vcr(filter_headers=["authorization"])
def test_crew_verbose_levels_output(capsys):
tasks = [Task(description="Write about AI advancements.", agent=researcher)]
crew = Crew(agents=[researcher], tasks=tasks, process=Process.sequential, verbose=1)
crew.kickoff()
captured = capsys.readouterr()
expected_strings = ["Working Agent: Researcher", "Task output:"]
for expected_string in expected_strings:
assert expected_string in captured.out
# Now test with verbose set to 2
crew.verbose = 2
crew.kickoff()
captured = capsys.readouterr()
expected_strings = [
"Working Agent: Researcher",
"Starting Task: Write about AI advancements. ...",
"Task output:",
]
for expected_string in expected_strings:
assert expected_string in captured.out
@pytest.mark.vcr(filter_headers=["authorization"])
def test_cache_hitting_between_agents():
from unittest.mock import patch