fixing RPM controlelr being set unencessarily

This commit is contained in:
João Moura
2024-02-07 23:09:36 -08:00
parent 98133a4eb6
commit a45c82c5f7
3 changed files with 26 additions and 3 deletions

View File

@@ -356,6 +356,25 @@ def test_api_calls_throttling(capsys):
moveon.assert_called()
def test_agents_rpm_is_never_set_if_crew_max_RPM_is_not_set():
agent = Agent(
role="test role",
goal="test goal",
backstory="test backstory",
allow_delegation=False,
verbose=True,
)
task = Task(
description="just say hi!",
agent=agent,
)
Crew(agents=[agent], tasks=[task], verbose=2)
assert agent._rpm_controller is None
def test_async_task_execution():
import threading
from unittest.mock import patch