diff --git a/test/agentchat/contrib/test_agent_builder.py b/test/agentchat/contrib/test_agent_builder.py index c945f127148d..3f9d69c32ea6 100644 --- a/test/agentchat/contrib/test_agent_builder.py +++ b/test/agentchat/contrib/test_agent_builder.py @@ -53,10 +53,6 @@ def test_build(): # check number of agents assert len(agent_config["agent_configs"]) <= builder.max_agents - # check system message - for cfg in agent_config["agent_configs"]: - assert "TERMINATE" in cfg["system_message"] - @pytest.mark.skipif( skip, @@ -87,10 +83,6 @@ def test_build_from_library(): # check number of agents assert len(agent_config["agent_configs"]) <= builder.max_agents - # check system message - for cfg in agent_config["agent_configs"]: - assert "TERMINATE" in cfg["system_message"] - builder.clear_all_agents() # test embedding similarity selection @@ -111,12 +103,6 @@ def test_build_from_library(): # check number of agents assert len(agent_config["agent_configs"]) <= builder.max_agents - # Disabling the assertion below to avoid test failure - # TODO: check whether the assertion is necessary - # check system message - # for cfg in agent_config["agent_configs"]: - # assert "TERMINATE" in cfg["system_message"] - @pytest.mark.skipif( skip,