From 3132e5962d875b40be00bb9008c1561e463423b4 Mon Sep 17 00:00:00 2001
From: Yulong Liu <lyl890808@gmail.com>
Date: Mon, 8 Jul 2024 16:32:58 -0700
Subject: [PATCH] add tests

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

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index c69fc1017a0..32ddeffedeb 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -218,6 +218,34 @@ def test_completion_claude():
 # test_completion_claude()
 
 
+def test_completion_empower():
+    litellm.set_verbose = True
+    messages = [
+        {
+            "role": "user",
+            "content": "\nWhat is the query for `console.log` => `console.error`\n",
+        },
+        {
+            "role": "assistant",
+            "content": "\nThis is the GritQL query for the given before/after examples:\n<gritql>\n`console.log` => `console.error`\n</gritql>\n",
+        },
+        {
+            "role": "user",
+            "content": "\nWhat is the query for `console.info` => `consdole.heaven`\n",
+        },
+    ]
+    try:
+        # test without max tokens
+        response = completion(
+            model="empower/empower-functions-small",
+            messages=messages,
+        )
+        # Add any assertions, here to check response args
+        print(response)
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
+
+
 def test_completion_claude_3_empty_response():
     litellm.set_verbose = True