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

fix the code execution test in sdk tests

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-16 16:39:43 -08:00
parent 35bf6ea75a
commit c862b5e92a
1 changed files with 5 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -206,18 +206,13 @@ def test_builtin_tool_code_execution(llama_stack_client, agent_config):
assert "Tool:code_interpreter Response" in logs_str
def test_code_execution(llama_stack_client):
agent_config = AgentConfig(
model="meta-llama/Llama-3.1-8B-Instruct",
instructions="You are a helpful assistant",
toolgroups=[
def test_code_execution(llama_stack_client, agent_config):
agent_config = {
**agent_config,
"toolgroups": [
"builtin::code_interpreter",
],
tool_choice="required",
input_shields=[],
output_shields=[],
enable_session_persistence=False,
)
}
codex_agent = Agent(llama_stack_client, agent_config)
session_id = codex_agent.create_session("test-session")