phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-17 15:13:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

yet another refactor to make this more general

Browse source 
now it accepts --inference-model, --safety-model options also
This commit is contained in:
Ashwin Bharambe 2024-11-04 14:16:35 -08:00 • committed by Ashwin Bharambe
parent 2ed0267fbb
commit 60800bc09b
13 changed files with 127 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/tests/agents/test_agents.py
View file

@ -75,12 +75,6 @@ async def create_agent_session(agents_impl, agent_config):
return agent_id, session_id
@pytest.mark.parametrize(
"inference_model", [pytest.param("Llama3.1-8B-Instruct", id="")], indirect=True
)
@pytest.mark.parametrize(
"safety_model", [pytest.param("Llama-Guard-3-8B", id="")], indirect=True
)
class TestAgents:
@pytest.mark.asyncio
async def test_agent_turns_with_safety(self, agents_stack, common_params):