From 34751d8562d85e3e5b12bc99b2b5c6c417a66fc9 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 3 Nov 2023 18:02:50 -0700 Subject: [PATCH] (test) add text_completion with prompt list --- litellm/tests/test_text_completion.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_text_completion.py b/litellm/tests/test_text_completion.py index 2548ac356eb..45a71199320 100644 --- a/litellm/tests/test_text_completion.py +++ b/litellm/tests/test_text_completion.py @@ -69,4 +69,20 @@ def test_completion_hf_prompt_array(): # response_str = response["choices"][0]["text"] except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_hf_prompt_array() \ No newline at end of file +# test_completion_hf_prompt_array() + + +def test_completion_text_003_prompt_array(): + try: + litellm.set_verbose=False + response = text_completion( + model="text-davinci-003", + prompt=token_prompt, # token prompt is a 2d list + ) + print("\n\n response") + + print(response) + # response_str = response["choices"][0]["text"] + except Exception as e: + pytest.fail(f"Error occurred: {e}") +# test_completion_text_003_prompt_array() \ No newline at end of file