mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-17 00:26:01 +00:00
fix(factory.py): fix anthropic prompt template
This commit is contained in:
@@ -575,6 +575,7 @@ def anthropic_messages_pt(messages: list):
|
||||
if messages[i]["role"] == "assistant":
|
||||
last_assistant_message_idx = i
|
||||
|
||||
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
|
||||
|
||||
@@ -82,6 +82,23 @@ def test_completion_claude():
|
||||
# test_completion_claude()
|
||||
|
||||
|
||||
def test_completion_claude_3_empty_response():
|
||||
messages = [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are 2twNLGfqk4GMOn3ffp4p.",
|
||||
},
|
||||
{"role": "user", "content": "Hi gm!"},
|
||||
{"role": "assistant", "content": "Good morning! How are you doing today?"},
|
||||
{
|
||||
"role": "user",
|
||||
"content": "I was hoping we could chat a bit",
|
||||
},
|
||||
]
|
||||
response = litellm.completion(model="claude-3-opus-20240229", messages=messages)
|
||||
print(response)
|
||||
|
||||
|
||||
def test_completion_claude_3():
|
||||
litellm.set_verbose = True
|
||||
messages = [{"role": "user", "content": "Hello, world"}]
|
||||
|
||||
Reference in New Issue
Block a user