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