From 2a9fd4c28dc2a9a868d6df604b4febe7f52be21f Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 23 Mar 2024 14:34:42 -0700 Subject: [PATCH] test(test_completion.py): make default claude 3 test message multi-turn --- litellm/tests/test_completion.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index d0020fd61..215bdef16 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -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\n`console.log` => `console.error`\n\n", + }, + { + "role": "user", + "content": "\nWhat is the query for `console.info` => `consdole.heaven`\n", + }, + ] try: # test without max tokens response = completion(