phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: do not overwrite agent_config

Browse source 
Summary:

Test Plan:
This commit is contained in:
Eric Huang 2025-02-21 16:37:20 -08:00
parent 45ffe87d7c
commit 8f221e4a33
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/client-sdk/agents/test_agents.py
View file

@ -332,8 +332,11 @@ def test_tool_choice(llama_stack_client, agent_config):
]
client_tool = TestClientTool()
for tool_choice, expected_tool in data:
agent_config["tool_config"] = {"tool_choice": tool_choice}
agent_config["client_tools"] = [client_tool.get_tool_definition()]
agent_config = {
**agent_config,
"tool_config": {"tool_choice": tool_choice},
"client_tools": [client_tool.get_tool_definition()],
}
agent = Agent(llama_stack_client, agent_config, client_tools=(client_tool,))
session_id = agent.create_session(f"test-session-{uuid4()}")