From 3ebb2c667c34792eec59f7c8b4737498e26afef1 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 26 Sep 2023 10:02:12 -0700
Subject: [PATCH] add palm test

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

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index e9431b100c..912b48fa58 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -882,6 +882,21 @@ def test_completion_deep_infra():
         print(response.response_ms)
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
+
+
+# Palm tests
+def test_completion_palm():
+    # litellm.set_verbose = True
+    model_name = "palm/chat-bison"
+    try:
+        response = completion(model=model_name, messages=messages)
+        # Add any assertions here to check the response
+        print(response)
+        print(response.response_ms)
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
+# test_completion_palm()
+
 # test_completion_deep_infra()
 # test_completion_ai21()
 # test config file with completion #