From a9fdae0d237e7f307d1e138764e897a5bff04205 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 29 Nov 2023 19:58:00 -0800 Subject: [PATCH] test(test_streaming.py): refactor testing --- litellm/tests/test_streaming.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index 1f51e54bb..38af89631 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -760,17 +760,6 @@ def hf_test_completion_tgi_stream(): pytest.fail(f"Error occurred: {e}") # hf_test_completion_tgi_stream() -def test_petals(): - print(f"making petals call") - response = completion( - model="petals/meta-llama/Llama-2-70b-chat-hf", - messages=[{ "content": "Hello, how are you?","role": "user"}], - # stream=True, - base_url="https://chat.petals.dev/api/v1/generate" - ) - - print(f"response: {response}") -test_petals() # def test_completion_aleph_alpha(): # try: # response = completion(