test(test_completion.py): make default claude 3 test message multi-turn

This commit is contained in:
Krrish Dholakia 2024-03-23 14:34:42 -07:00
parent 05dfc9e6ac
commit 2a9fd4c28d

View file

@ -102,7 +102,20 @@ def test_completion_claude_3_empty_response():
def test_completion_claude_3():
litellm.set_verbose = True
messages = [{"role": "user", "content": "Hello, world"}]
messages = [
{
"role": "user",
"content": "\nWhat is the query for `console.log` => `console.error`\n",
},
{
"role": "assistant",
"content": "\nThis is the GritQL query for the given before/after examples:\n<gritql>\n`console.log` => `console.error`\n</gritql>\n",
},
{
"role": "user",
"content": "\nWhat is the query for `console.info` => `consdole.heaven`\n",
},
]
try:
# test without max tokens
response = completion(