phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

test: do not overwrite agent_config (#1216)

Browse source 
Summary:

Test Plan:
This commit is contained in:
ehhuang 2025-02-21 16:38:56 -08:00 • committed by GitHub
parent 187524d4ae
commit c9e08cc0a8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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()}")