forked from phoenix/litellm-mirror
test(test_completion.py): make default claude 3 test message multi-turn
This commit is contained in:
parent
05dfc9e6ac
commit
2a9fd4c28d
1 changed files with 14 additions and 1 deletions
|
@ -102,7 +102,20 @@ def test_completion_claude_3_empty_response():
|
||||||
|
|
||||||
def test_completion_claude_3():
|
def test_completion_claude_3():
|
||||||
litellm.set_verbose = True
|
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:
|
try:
|
||||||
# test without max tokens
|
# test without max tokens
|
||||||
response = completion(
|
response = completion(
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue