From 85692398c1d955cc78021d2dc5d9f78bdf2edbbf Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 31 Oct 2023 15:33:45 -0700 Subject: [PATCH] (test) add support for returning logprobs --- litellm/tests/test_completion.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 14058404d4..e1c514616d 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -56,7 +56,7 @@ def test_completion_claude(): except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_claude() +test_completion_claude() # def test_completion_oobabooga(): # try: @@ -400,9 +400,9 @@ def test_completion_openai_prompt(): def test_completion_openai_prompt_array(): try: - litellm.set_verbose=True + litellm.set_verbose=False response = text_completion( - model="text-davinci-003", prompt=[[3152, 833]], max_tokens=10 + model="text-davinci-003", prompt=[[3152, 833]], max_tokens=10, logprobs=10 ) print(response) response_str = response["choices"][0]["text"]