forked from phoenix-oss/llama-stack-mirror
Fix incorrect completion() signature for Databricks provider (#236)
This commit is contained in:
parent
9fbe8852aa
commit
2128e61da2
6 changed files with 19 additions and 6 deletions
|
@ -170,7 +170,7 @@ class LlamaGuardShield(ShieldBase):
|
|||
for i in range(1, len(messages)):
|
||||
if messages[i].role == messages[i - 1].role:
|
||||
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
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue