tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-08 22:25:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(factory.py): raise exception on invalid message being received

Browse Source
This commit is contained in:
Krrish Dholakia
2024-03-15 07:55:29 -07:00
parent 4c834714ab
commit 5a2e024576
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/prompt_templates/factory.py
+10 -5
View File
@@ -627,11 +627,16 @@ def anthropic_messages_pt(messages: list):
new_messages.append(messages[-1])
if last_assistant_message_idx is not None:
new_messages[last_assistant_message_idx]["content"] = new_messages[
last_assistant_message_idx
][
"content"
].strip() # no trailing whitespace for final assistant message
try:
new_messages[last_assistant_message_idx]["content"] = new_messages[
last_assistant_message_idx
][
"content"
].strip() # no trailing whitespace for final assistant message
except Exception as e:
raise ValueError(
f"LiteLLMException: Invalid final assistant message passed in. Messages={messages}"
)
return new_messages