From e29b2e8ce4e4f80e422d1b0b075f163b57d6624d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 3 Nov 2023 16:39:06 -0700 Subject: [PATCH] (fix) testing text_completion --- litellm/tests/test_text_completion.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_text_completion.py b/litellm/tests/test_text_completion.py index cd6ac6450..2548ac356 100644 --- a/litellm/tests/test_text_completion.py +++ b/litellm/tests/test_text_completion.py @@ -24,9 +24,11 @@ def test_completion_openai_prompt(): response = text_completion( model="gpt-3.5-turbo", prompt="What's the weather in SF?" ) + print(response) response_str = response["choices"][0]["text"] except Exception as e: pytest.fail(f"Error occurred: {e}") +# test_completion_openai_prompt() def test_completion_openai_prompt_array(): @@ -61,10 +63,10 @@ def test_completion_hf_prompt_array(): model="huggingface/mistralai/Mistral-7B-v0.1", prompt=token_prompt, # token prompt is a 2d list ) - print("\n\n response\n\n") + print("\n\n response") print(response) # 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() \ No newline at end of file