phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 05:32:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix agents tests

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-11 22:31:09 -08:00
parent 38257a9cbe
commit abe1cc6303
7 changed files with 51 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/tests/safety/fixtures.py
View file

@ -99,7 +99,7 @@ async def safety_stack(inference_model, safety_model, request):
provider_data.update(safety_fixture.provider_data)
shield_provider_type = safety_fixture.providers[0].provider_type
shield = get_shield(
shield = get_shield_to_register(
shield_provider_type, safety_fixture.providers[0].provider_id, safety_model
)
@ -120,7 +120,7 @@ async def safety_stack(inference_model, safety_model, request):
return impls[Api.safety], impls[Api.shields], shield
def get_shield(provider_type: str, provider_id: str, safety_model: str):
def get_shield_to_register(provider_type: str, provider_id: str, safety_model: str):
shield_config = {}
shield_type = ShieldType.llama_guard
identifier = "llama_guard"