diff --git a/tests/functional/agent/agent_functional_test.py b/tests/functional/agent/agent_functional_test.py index b1a9d453..5df56d13 100644 --- a/tests/functional/agent/agent_functional_test.py +++ b/tests/functional/agent/agent_functional_test.py @@ -300,7 +300,8 @@ def test_update_tools_of_agent(run_input_map, delete_agents_and_team_agents, Age }, ], ) -def test_specific_model_parameters_e2e(tool_config): +def test_specific_model_parameters_e2e(tool_config, delete_agents_and_team_agents): + assert delete_agents_and_team_agents """Test end-to-end agent execution with specific model parameters""" # Create tool based on config if tool_config["type"] == "search": diff --git a/tests/functional/team_agent/team_agent_functional_test.py b/tests/functional/team_agent/team_agent_functional_test.py index 3cabad8c..cb5f80a9 100644 --- a/tests/functional/team_agent/team_agent_functional_test.py +++ b/tests/functional/team_agent/team_agent_functional_test.py @@ -301,6 +301,7 @@ def test_team_agent_with_parameterized_agents(run_input_map, delete_agents_and_t llm_id=run_input_map["llm_id"], tools=[search_tool], ) + search_agent.deploy() # Create second agent with translation tool translation_function = Function.TRANSLATION @@ -318,7 +319,7 @@ def test_team_agent_with_parameterized_agents(run_input_map, delete_agents_and_t llm_id=run_input_map["llm_id"], tools=[translation_tool], ) - + translation_agent.deploy() team_agent = create_team_agent( TeamAgentFactory, [search_agent, translation_agent], run_input_map, use_mentalist=True, use_inspector=True )