From 6e48d92891586ddf2e3e50ee6209fee9cc9fc375 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 29 Sep 2023 11:57:37 -0700
Subject: [PATCH] hf embedding tests

---
 litellm/tests/test_embedding.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py
index eea95e395..e27f930be 100644
--- a/litellm/tests/test_embedding.py
+++ b/litellm/tests/test_embedding.py
@@ -33,3 +33,17 @@ def test_cohere_embedding():
 
 # test_cohere_embedding()
 
+# comment out hf tests - since hf endpoints are unstable
+# def test_hf_embedding():
+#     try:
+#         # huggingface/microsoft/codebert-base
+#         # huggingface/facebook/bart-large
+#         response = embedding(
+#             model="huggingface/BAAI/bge-large-zh", input=["good morning from litellm", "this is another item"]
+#         )
+#         print(f"response:", response)
+#     except Exception as e:
+#         pytest.fail(f"Error occurred: {e}")
+# test_hf_embedding()
+
+