From 5d196ff300a2ba3d34ff38db53d3150886e99a44 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 5 Apr 2024 08:56:05 -0700
Subject: [PATCH] test - azure/command-r-plus

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

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index b3c0f79f0..26dd6cc6e 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -53,6 +53,25 @@ def test_completion_custom_provider_model_name():
 # test_completion_custom_provider_model_name()
 
 
+def test_completion_azure_command_r():
+    try:
+        litellm.set_verbose = True
+
+        response = completion(
+            model="azure/command-r-plus",
+            model_id="Cohere-command-r-plus-gylpd",
+            api_base=os.getenv("AZURE_COHERE_API_BASE"),
+            api_key=os.getenv("AZURE_COHERE_API_KEY"),
+            messages=[{"role": "user", "content": "What is the meaning of life?"}],
+        )
+
+        print(response)
+    except litellm.Timeout as e:
+        pass
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
+
+
 def test_completion_claude():
     litellm.set_verbose = True
     litellm.cache = None