phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-03 19:57:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: fix agents tests for non-ollama providers, provide max_tokens

Browse source
This commit is contained in:
Matthew Farrellee 2025-10-02 13:20:50 -04:00
parent 24ee577cb0
commit b4a1a15aa9
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/agents/test_agents.py
View file

@ -68,6 +68,7 @@ def agent_config(llama_stack_client, text_model_id):
"temperature": 0.0001, "temperature": 0.0001,
"top_p": 0.9, "top_p": 0.9,
}, },
"max_tokens": 512,
}, },
tools=[], tools=[],
input_shields=available_shields, input_shields=available_shields,
@ -88,6 +89,7 @@ def agent_config_without_safety(text_model_id):
"temperature": 0.0001, "temperature": 0.0001,
"top_p": 0.9, "top_p": 0.9,
}, },
"max_tokens": 512,
}, },
tools=[], tools=[],
enable_session_persistence=False, enable_session_persistence=False,
@ -198,6 +200,7 @@ def test_tool_config(agent_config):
"temperature": 1.0, "temperature": 1.0,
"top_p": 0.9, "top_p": 0.9,
}, },
"max_tokens": 512,
}, },
toolgroups=[], toolgroups=[],
enable_session_persistence=False, enable_session_persistence=False,