Add more debugging logs to when llama guard fails

This commit is contained in:
Ashwin Bharambe 2024-12-17 17:11:21 -08:00
parent eea478618d
commit 0fb4b7de6f

View file

@ -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}"
)