phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 10:42:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix llama guard

Browse source
This commit is contained in:
Xi Yan 2025-01-15 11:11:31 -08:00
parent 32d3abe964
commit 6cec64c2e4
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_stack/providers/inline/safety/llama_guard/llama_guard.py
View file

@ -263,9 +263,12 @@ class LlamaGuardShield:
stream=True,
):
event = chunk.event
if event.event_type == ChatCompletionResponseEventType.progress:
assert isinstance(event.delta, str)
content += event.delta
if (
event.event_type == ChatCompletionResponseEventType.progress
and event.delta.type == "text"
):
assert isinstance(event.delta.text, str)
content += event.delta.text
content = content.strip()
return self.get_shield_response(content)