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
This commit is contained in:
ehhuang 2025-02-06 20:19:38 -08:00 committed by GitHub
parent af15426ad7
commit d0d568c5ba
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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,))