Fix precommit

Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2024-10-10 21:10:50 -04:00
parent 98ed50bb0e
commit c18ca95619
No known key found for this signature in database

View file

@ -170,7 +170,7 @@ class LlamaGuardShield(ShieldBase):
for i in range(1, len(messages)): for i in range(1, len(messages)):
if messages[i].role == messages[i - 1].role: if messages[i].role == messages[i - 1].role:
raise ValueError( raise ValueError(
f"Messages must alternate between user and assistant. Message {i} has the same role as message {i-1}" f"Messages must alternate between user and assistant. Message {i} has the same role as message {i - 1}"
) )
return messages return messages