(test) add gpt-4-turbo

2023-11-06 13:32:53 -08:00 · 2023-11-06 13:32:53 -08:00 · 8d9e2574cf
commit 8d9e2574cf
parent f7990525ff
1 changed files with 12 additions and 0 deletions
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@ -110,6 +110,18 @@ def test_completion_with_litellm_call_id():
        pytest.fail(f"Error occurred: {e}")
 # test_completion_with_litellm_call_id()

+def test_completion_gpt4_turbo():
+    try:
+        response = completion(
+            model="gpt-4-1106-preview", 
+            messages=messages,
+            max_tokens=10,
+        )
+        print(response)
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
+# test_completion_gpt4_turbo()
+
 def test_completion_perplexity_api():
    try:
        # litellm.set_verbose=True