From 9c37619e5c3df3fa6df96646169f606db1e5a98e Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 19 Sep 2023 09:24:24 -0700
Subject: [PATCH] test petals

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

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 0010a9a326..3bfd3956db 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -688,6 +688,18 @@ def test_completion_with_fallbacks():
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
 
+# def test_petals():
+#     try:
+#         response = completion(model="petals-team/StableBeluga2", messages=messages)
+#         # Add any assertions here to check the response
+#         print(response)
+
+#         response = completion(model="petals-team/StableBeluga2", messages=messages)
+#         # Add any assertions here to check the response
+#         print(response)
+#     except Exception as e:
+#         pytest.fail(f"Error occurred: {e}")
+
 # def test_baseten():
 #     try: