phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 23:03: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/fixtures.py
View file

@ -22,7 +22,7 @@ from ..conftest import ProviderFixture
@pytest.fixture(scope="session")
def agents_meta_reference(inference_model, safety_model) -> ProviderFixture:
def agents_meta_reference() -> ProviderFixture:
sqlite_file = tempfile.NamedTemporaryFile(delete=False, suffix=".db")
return ProviderFixture(
provider=Provider(
@ -42,12 +42,12 @@ AGENTS_FIXTURES = ["meta_reference"]
@pytest_asyncio.fixture(scope="session")
async def agents_stack(inference_model, safety_model, request):
async def agents_stack(request):
fixture_dict = request.param
providers = {}
provider_data = {}
for key in ["agents", "inference", "safety", "memory"]:
for key in ["inference", "safety", "memory", "agents"]:
fixture = request.getfixturevalue(f"{key}_{fixture_dict[key]}")
providers[key] = [fixture.provider]
if fixture.provider_data: