phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Add a test runner and 2 very simple tests for agents

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-19 12:05:36 -07:00
parent 543222ac39
commit abb43936ab
9 changed files with 243 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/impls/meta_reference/agents/safety.py
View file

@ -43,12 +43,10 @@ class ShieldRunnerMixin:
if len(messages) > 0 and messages[0].role != Role.user.value:
messages[0] = UserMessage(content=messages[0].content)
res = await self.safety_api.run_shields(
results = await self.safety_api.run_shields(
messages=messages,
shields=shields,
)
results = res.responses
for shield, r in zip(shields, results):
if r.is_violation:
if shield.on_violation_action == OnViolationAction.RAISE: