diff --git a/dist/litellm-1.3.3.dev1-py3-none-any.whl b/dist/litellm-1.3.3.dev1-py3-none-any.whl new file mode 100644 index 000000000..6c189e365 Binary files /dev/null and b/dist/litellm-1.3.3.dev1-py3-none-any.whl differ diff --git a/dist/litellm-1.3.3.dev1.tar.gz b/dist/litellm-1.3.3.dev1.tar.gz new file mode 100644 index 000000000..ac249b71c Binary files /dev/null and b/dist/litellm-1.3.3.dev1.tar.gz differ diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py index 501667d07..122ac63a9 100644 --- a/litellm/llms/huggingface_restapi.py +++ b/litellm/llms/huggingface_restapi.py @@ -406,7 +406,7 @@ class Huggingface(BaseLLM): try: completion_response = response.json() if isinstance(completion_response, dict): - completion_response: List[Dict[str, Any]] = [{"generated_text": content}] + completion_response: List[Dict[str, Any]] = [{"generated_text": completion_response}] except: import traceback raise HuggingfaceError( diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index b5a2b72e7..47cea9fd1 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -159,7 +159,7 @@ def test_completion_gpt4_vision(): pass except Exception as e: pytest.fail(f"Error occurred: {e}") -test_completion_gpt4_vision() +# test_completion_gpt4_vision() def test_completion_perplexity_api(): try: @@ -283,7 +283,7 @@ def hf_test_completion_tgi(): print(response) except Exception as e: pytest.fail(f"Error occurred: {e}") -# hf_test_completion_tgi() +hf_test_completion_tgi() def hf_test_completion_tgi_stream(): try: