diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index a1e5b4bf13..6cd1e83053 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -49,15 +49,15 @@ def test_completion_hf_api(): 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 5aea23ba69..247d76ec64 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.395" +version = "0.1.396" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"