From e5ceed4334cf795f96501f02545177c56ccbf4dd Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 27 Sep 2023 10:53:45 -0700 Subject: [PATCH] update hf testing --- litellm/tests/test_completion.py | 17 ----------------- pyproject.toml | 2 +- 2 files changed, 1 insertion(+), 18 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 79273a7a30a..f96a46b936d 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -200,23 +200,6 @@ def hf_test_completion(): # hf_test_completion() -def hf_test_completion_task_none(): - try: - litellm.set_verbose=True - user_message = "My name is Merve and my favorite" - messages = [{ "content": user_message,"role": "user"}] - model = "huggingface/cerebras/btlm-3b-8k-base" - response = completion( - model=model, - messages=messages, - task=None, - ) - # Add any assertions here to check the response - print(response) - except Exception as e: - pytest.fail(f"Error occurred: {e}") -hf_test_completion_task_none() - # this should throw an exception, to trigger https://logs.litellm.ai/ # def hf_test_error_logs(): # try: diff --git a/pyproject.toml b/pyproject.toml index 698dba3479b..db07be1edce 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.783" +version = "0.1.784" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"