Simplify mock_patch_acompletion

litellm/tests/test_proxy_server.py

@@ -37,11 +37,7 @@ token = "sk-1234"
 
 headers = {"Authorization": f"Bearer {token}"}
 
-
+example_completion_result = {
-@contextlib.contextmanager
-def mock_patch_acompletion():
-    async def side_effect(*args, **kwargs):
-        return {
     "choices": [
         {
             "message": {
@@ -52,9 +48,12 @@ def mock_patch_acompletion():
     ],
 }
 
+
+@contextlib.contextmanager
+def mock_patch_acompletion():
     with mock.patch(
         "litellm.proxy.proxy_server.llm_router.acompletion",
-        side_effect=side_effect,
+        return_value=example_completion_result,
     ):
         yield