phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-08 00:11:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make test easier

Browse source
This commit is contained in:
Xi Yan 2025-02-15 15:27:34 -08:00
parent 450bd60517
commit 081f245772
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -451,12 +451,12 @@ def test_rag_agent(llama_stack_client, agent_config):
session_id = rag_agent.create_session(f"test-session-{uuid4()}")
user_prompts = [
(
"What is main changes between Llama2-7B and Llama3-8B models on how attention is used?",
"grouped",
"What term is used to refer to the underlying data type used to represent the model and optimizer parameters.",
"precision",
),
(
"What `tune` command to use for getting access to Llama3-8B-Instruct model?",
"download",
"What models is LoRA is most commonly applied to?",
"transformer",
),
]
for prompt, expected_kw in user_prompts: