From 0fb4b7de6f80ea99fc41b69d937fe4d35e004a98 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Tue, 17 Dec 2024 17:11:21 -0800 Subject: [PATCH] Add more debugging logs to when llama guard fails --- llama_stack/providers/inline/safety/llama_guard/llama_guard.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/llama_stack/providers/inline/safety/llama_guard/llama_guard.py b/llama_stack/providers/inline/safety/llama_guard/llama_guard.py index c243427d3..bbdd5c3df 100644 --- a/llama_stack/providers/inline/safety/llama_guard/llama_guard.py +++ b/llama_stack/providers/inline/safety/llama_guard/llama_guard.py @@ -226,6 +226,8 @@ class LlamaGuardShield: for i in range(1, len(messages)): if messages[i].role == messages[i - 1].role: + for i, m in enumerate(messages): + print(f"{i}: {m.role}: {m.content}") raise ValueError( f"Messages must alternate between user and assistant. Message {i} has the same role as message {i - 1}" )