From a604ca577b6c10b053145f5fc638aa9d5a15d2a6 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 28 Sep 2023 17:00:42 -0700
Subject: [PATCH] remove junk hf stream print statement

---
 litellm/tests/test_completion.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index edcfa6d366..4cc36beb7e 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -73,7 +73,7 @@ def test_completion_claude_max_tokens():
         litellm.set_verbose = False
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
-test_completion_claude_max_tokens()
+# test_completion_claude_max_tokens()
 
 # def test_completion_oobabooga():
 #     try:
@@ -209,6 +209,22 @@ def test_get_hf_task_for_model():
 #         pytest.fail(f"Error occurred: {e}")
 # hf_test_completion_tgi()
 
+# def hf_test_completion_tgi_stream():
+#     try:
+#         response = litellm.completion(
+#             model="huggingface/glaiveai/glaive-coder-7b",
+#             messages=[{ "content": "Hello, how are you?","role": "user"}],
+#             api_base="https://wjiegasee9bmqke2.us-east-1.aws.endpoints.huggingface.cloud",
+#             stream=True
+#         )
+#         # Add any assertions here to check the response
+#         print(response)
+#         for chunk in response:
+#             print(chunk)
+#     except Exception as e:
+#         pytest.fail(f"Error occurred: {e}")
+# hf_test_completion_tgi_stream()
+
 # ################### Hugging Face Conversational models ########################
 # def hf_test_completion_conv():
 #     try: