docs

docs/my-website/docs/tutorials/mock_completion.md

@@ -3,7 +3,7 @@
 Trying to test making LLM Completion calls without calling the LLM APIs ? 
 Pass `mock_response` to `litellm.completion` and litellm will directly return the response without neededing the call the LLM API and spend $$ 
 
-## Using `mock_response`
+## Using `completion()` with `mock_response`
 
 ```python
 from litellm import completion 
@@ -12,4 +12,24 @@ model = "gpt-3.5-turbo"
 messages = [{"role":"user", "content":"Why is LiteLLM amazing?"}]
 
 completion(model=model, messages=messages, mock_response="It's simple to use and easy to get started")
-```
+```
+
+## Building a pytest function using `completion`
+
+```python
+from litellm import completion
+import pytest
+
+def test_completion_openai():
+    try:
+        response = completion(
+            model="gpt-3.5-turbo",
+            messages=[{"role":"user", "content":"Why is LiteLLM amazing?"}],
+            mock_response="LiteLLM is awesome"
+        )
+        # Add any assertions here to check the response
+        print(response)
+        print(response['choices'][0]['finish_reason'])
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")
+```