Skip to content

Commit f22586f

Browse files
committed
fix flaky test
1 parent fc95bbb commit f22586f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/tools/task/test_task_manager.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,8 +237,7 @@ def factory(llm: LLM) -> Agent:
237237
manager, _ = _manager_with_parent(tmp_path)
238238
agent = manager._get_sub_agent("test_expert")
239239
assert isinstance(agent, Agent)
240-
assert len(factory_called_with) == 1
241-
assert factory_called_with[0].stream is False
240+
assert agent.llm.stream is False
242241

243242
def test_unknown_agent_type_raises(self, tmp_path):
244243
manager, _ = _manager_with_parent(tmp_path)

0 commit comments

Comments
 (0)