diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 200f13b1df..ca066db245 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -79,15 +79,15 @@ def test_completion_claude_stream(): # except Exception as e: # pytest.fail(f"Error occurred: {e}") -def test_completion_hf_deployed_api(): - try: - user_message = "There's a llama in my garden 😱 What should I do?" - messages = [{ "content": user_message,"role": "user"}] - response = completion(model="meta-llama/Llama-2-7b-chat-hf", messages=messages, custom_llm_provider="huggingface", custom_api_base="https://s7c7gytn18vnu4tw.us-east-1.aws.endpoints.huggingface.cloud", logger_fn=logger_fn) - # Add any assertions here to check the response - print(response) - except Exception as e: - pytest.fail(f"Error occurred: {e}") +# def test_completion_hf_deployed_api(): +# try: +# user_message = "There's a llama in my garden 😱 What should I do?" +# messages = [{ "content": user_message,"role": "user"}] +# response = completion(model="meta-llama/Llama-2-7b-chat-hf", messages=messages, custom_llm_provider="huggingface", custom_api_base="https://s7c7gytn18vnu4tw.us-east-1.aws.endpoints.huggingface.cloud", logger_fn=logger_fn) +# # Add any assertions here to check the response +# print(response) +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") def test_completion_cohere(): diff --git a/pyproject.toml b/pyproject.toml index b6e6af52bf..a47a36b5fd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.499" +version = "0.1.500" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"