Fix async usage (#167)

Co-authored-by: Jason Liu <jxnl@users.noreply.github.com>
Co-authored-by: Jason Liu <jason@jxnl.co>

tests/test_patch.py

@@ -1,14 +1,17 @@
+import functools
 import pytest
 import instructor
 
-from pydantic import BaseModel, ValidationError, BeforeValidator
+from pydantic import BaseModel, Field, ValidationError, BeforeValidator
 from openai import OpenAI, AsyncOpenAI
 from instructor import llm_validator
 from typing_extensions import Annotated
 
 
+from instructor.patch import is_async, wrap_chatcompletion
+
 client = instructor.patch(OpenAI())
-aclient = instructor.apatch(AsyncOpenAI())
+aclient = instructor.patch(AsyncOpenAI())
 
 
 @pytest.mark.asyncio
@@ -78,6 +81,61 @@ def test_runmodel_validator():
         model, "_raw_response"
     ), "The raw response should be available from OpenAI"
 
+
+def test_patch_completes_successfully():
+    instructor.patch(OpenAI())
+
+
+def test_apatch_completes_successfully():
+    instructor.apatch(AsyncOpenAI())
+
+
+@pytest.mark.asyncio
+async def test_wrap_chatcompletion_wraps_async_input_function():
+    async def input_function(*args, **kwargs):
+        return "Hello, World!"
+
+    wrapped_function = wrap_chatcompletion(input_function)
+    result = await wrapped_function()
+
+    assert result == "Hello, World!"
+
+
+def test_wrap_chatcompletion_wraps_input_function():
+    def input_function(*args, **kwargs):
+        return "Hello, World!"
+
+    wrapped_function = wrap_chatcompletion(input_function)
+    result = wrapped_function()
+
+    assert result == "Hello, World!"
+
+
+def test_is_async_returns_true_if_function_is_async():
+    async def async_function():
+        pass
+
+    assert is_async(async_function) is True
+
+
+def test_is_async_returns_false_if_function_is_not_async():
+    def sync_function():
+        pass
+
+    assert is_async(sync_function) is False
+
+
+def test_is_async_returns_true_if_wrapped_function_is_async():
+    async def async_function():
+        pass
+
+    @functools.wraps(async_function)
+    def wrapped_function():
+        pass
+
+    assert is_async(wrapped_function) is True
+
+
 @pytest.mark.asyncio
 async def test_async_runmodel_validator():
     aclient = instructor.apatch(AsyncOpenAI())