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