fix: asyncify on non-asyncio runtimes (#696)

Author: stainless-app[bot]

src/lithic/_utils/_sync.py

@@ -7,16 +7,20 @@
 from typing import Any, TypeVar, Callable, Awaitable
 from typing_extensions import ParamSpec
 
+import anyio
+import sniffio
+import anyio.to_thread
+
 T_Retval = TypeVar("T_Retval")
 T_ParamSpec = ParamSpec("T_ParamSpec")
 
 
 if sys.version_info >= (3, 9):
-    to_thread = asyncio.to_thread
+    _asyncio_to_thread = asyncio.to_thread
 else:
     # backport of https://docs.python.org/3/library/asyncio-task.html#asyncio.to_thread
     # for Python 3.8 support
-    async def to_thread(
+    async def _asyncio_to_thread(
         func: Callable[T_ParamSpec, T_Retval], /, *args: T_ParamSpec.args, **kwargs: T_ParamSpec.kwargs
     ) -> Any:
         """Asynchronously run function *func* in a separate thread.
@@ -34,6 +38,17 @@ async def to_thread(
         return await loop.run_in_executor(None, func_call)
 
 
+async def to_thread(
+    func: Callable[T_ParamSpec, T_Retval], /, *args: T_ParamSpec.args, **kwargs: T_ParamSpec.kwargs
+) -> T_Retval:
+    if sniffio.current_async_library() == "asyncio":
+        return await _asyncio_to_thread(func, *args, **kwargs)
+
+    return await anyio.to_thread.run_sync(
+        functools.partial(func, *args, **kwargs),
+    )
+
+
 # inspired by `asyncer`, https://github.com/tiangolo/asyncer
 def asyncify(function: Callable[T_ParamSpec, T_Retval]) -> Callable[T_ParamSpec, Awaitable[T_Retval]]:
     """

tests/api_resources/test_account_holders.py

@@ -720,15 +720,15 @@ def test_path_params_retrieve_document(self, client: Lithic) -> None:
     @parametrize
     def test_method_simulate_enrollment_document_review(self, client: Lithic) -> None:
         account_holder = client.account_holders.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
         )
         assert_matches_type(Document, account_holder, path=["response"])
 
     @parametrize
     def test_method_simulate_enrollment_document_review_with_all_params(self, client: Lithic) -> None:
         account_holder = client.account_holders.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
             accepted_entity_status_reasons=["string"],
             status_reason="DOCUMENT_MISSING_REQUIRED_DATA",
@@ -738,7 +738,7 @@ def test_method_simulate_enrollment_document_review_with_all_params(self, client
     @parametrize
     def test_raw_response_simulate_enrollment_document_review(self, client: Lithic) -> None:
         response = client.account_holders.with_raw_response.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
         )
 
@@ -750,7 +750,7 @@ def test_raw_response_simulate_enrollment_document_review(self, client: Lithic)
     @parametrize
     def test_streaming_response_simulate_enrollment_document_review(self, client: Lithic) -> None:
         with client.account_holders.with_streaming_response.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
         ) as response:
             assert not response.is_closed
@@ -1539,15 +1539,15 @@ async def test_path_params_retrieve_document(self, async_client: AsyncLithic) ->
     @parametrize
     async def test_method_simulate_enrollment_document_review(self, async_client: AsyncLithic) -> None:
         account_holder = await async_client.account_holders.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
         )
         assert_matches_type(Document, account_holder, path=["response"])
 
     @parametrize
     async def test_method_simulate_enrollment_document_review_with_all_params(self, async_client: AsyncLithic) -> None:
         account_holder = await async_client.account_holders.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
             accepted_entity_status_reasons=["string"],
             status_reason="DOCUMENT_MISSING_REQUIRED_DATA",
@@ -1557,7 +1557,7 @@ async def test_method_simulate_enrollment_document_review_with_all_params(self,
     @parametrize
     async def test_raw_response_simulate_enrollment_document_review(self, async_client: AsyncLithic) -> None:
         response = await async_client.account_holders.with_raw_response.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
         )
 
@@ -1569,7 +1569,7 @@ async def test_raw_response_simulate_enrollment_document_review(self, async_clie
     @parametrize
     async def test_streaming_response_simulate_enrollment_document_review(self, async_client: AsyncLithic) -> None:
         async with async_client.account_holders.with_streaming_response.simulate_enrollment_document_review(
-            document_upload_token="b11cd67b-0a52-4180-8365-314f3def5426",
+            document_upload_token="document_upload_token",
             status="UPLOADED",
         ) as response:
             assert not response.is_closed

tests/test_client.py

@@ -24,10 +24,12 @@
 
 from lithic import Lithic, AsyncLithic, APIResponseValidationError
 from lithic._types import Omit
+from lithic._utils import maybe_transform
 from lithic._models import BaseModel, FinalRequestOptions
 from lithic._constants import RAW_RESPONSE_HEADER
 from lithic._exceptions import LithicError, APIStatusError, APITimeoutError, APIResponseValidationError
 from lithic._base_client import DEFAULT_TIMEOUT, HTTPX_DEFAULT_TIMEOUT, BaseClient, make_request_options
+from lithic.types.card_create_params import CardCreateParams
 
 from .utils import update_env
 
@@ -822,7 +824,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
         with pytest.raises(APITimeoutError):
             self.client.post(
                 "/v1/cards",
-                body=cast(object, dict(type="SINGLE_USE")),
+                body=cast(object, maybe_transform(dict(type="SINGLE_USE"), CardCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )
@@ -837,7 +839,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> Non
         with pytest.raises(APIStatusError):
             self.client.post(
                 "/v1/cards",
-                body=cast(object, dict(type="SINGLE_USE")),
+                body=cast(object, maybe_transform(dict(type="SINGLE_USE"), CardCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )
@@ -1740,7 +1742,7 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
         with pytest.raises(APITimeoutError):
             await self.client.post(
                 "/v1/cards",
-                body=cast(object, dict(type="SINGLE_USE")),
+                body=cast(object, maybe_transform(dict(type="SINGLE_USE"), CardCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )
@@ -1755,7 +1757,7 @@ async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter)
         with pytest.raises(APIStatusError):
             await self.client.post(
                 "/v1/cards",
-                body=cast(object, dict(type="SINGLE_USE")),
+                body=cast(object, maybe_transform(dict(type="SINGLE_USE"), CardCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )