From d0d568c5ba220a6e34a3a0c48c68bf7ddaf62e43 Mon Sep 17 00:00:00 2001 From: ehhuang Date: Thu, 6 Feb 2025 20:19:38 -0800 Subject: [PATCH] test: fix flaky agent test (#1002) Summary: Test Plan: LLAMA_STACK_CONFIG=fireworks pytest -s -v tests/client-sdk/ --safety-shield meta-llama/Llama-Guard-3-8 all tests passed --- tests/client-sdk/agents/test_agents.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/client-sdk/agents/test_agents.py b/tests/client-sdk/agents/test_agents.py index 2b1db7df0..302c1c6e7 100644 --- a/tests/client-sdk/agents/test_agents.py +++ b/tests/client-sdk/agents/test_agents.py @@ -269,6 +269,7 @@ def test_override_system_message_behavior(llama_stack_client, agent_config): **agent_config, "instructions": "You are a pirate", "client_tools": [client_tool.get_tool_definition()], + "model": "meta-llama/Llama-3.2-3B-Instruct", } agent = Agent(llama_stack_client, agent_config, client_tools=(client_tool,))