From f016234cbf11aaf4dbfce7ff5cee4eec3bb979fa Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 31 Oct 2023 18:20:58 -0700 Subject: [PATCH] (test) text_completion add echo for hf --- litellm/tests/test_completion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 85b5f7a429..db9e254206 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -402,7 +402,7 @@ def test_completion_openai_prompt_array(): try: litellm.set_verbose=False response = text_completion( - model="text-davinci-003", prompt=[[3152, 833]], max_tokens=10, logprobs=10, echo=True + model="text-davinci-003", prompt="good morning", max_tokens=10, logprobs=10, echo=True ) print(response) response_str = response["choices"][0]["text"] @@ -877,7 +877,7 @@ def test_completion_bedrock_claude(): pass except Exception as e: pytest.fail(f"Error occurred: {e}") -test_completion_bedrock_claude() +# test_completion_bedrock_claude() def test_completion_bedrock_claude_completion_auth():