mirascope 1.23.2__tar.gz → 1.23.3__tar.gz

{mirascope-1.23.2 → mirascope-1.23.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mirascope
-Version: 1.23.2
+Version: 1.23.3
 Summary: LLM abstractions that aren't obstructions
 Project-URL: Homepage, https://mirascope.com
 Project-URL: Documentation, https://mirascope.com/WELCOME

{mirascope-1.23.2 → mirascope-1.23.3}/mirascope/core/base/_utils/_fn_is_async.py

@@ -12,4 +12,13 @@ _R = TypeVar("_R")
 def fn_is_async(
     fn: Callable[..., Awaitable[_R] | Coroutine[Any, Any, _R]] | Callable[..., _R],
 ) -> TypeIs[Callable[..., Awaitable[_R]]]:
-    return inspect.iscoroutinefunction(fn)
+    if inspect.iscoroutinefunction(fn):
+        return True
+
+    # Check if it's a wrapper around an `async def` function (using functools.wraps).
+    _fn = fn
+    while hasattr(_fn, "__wrapped__"):
+        _fn = _fn.__wrapped__  # pyright: ignore[reportFunctionMemberAccess]
+        if inspect.iscoroutinefunction(_fn):
+            return True
+    return False

{mirascope-1.23.2 → mirascope-1.23.3}/mirascope/integrations/_middleware_factory.py

@@ -1,6 +1,5 @@
 """The `middleware_factory` method for handling the call response."""
 
-import inspect
 from collections.abc import AsyncGenerator, Awaitable, Callable, Generator
 from contextlib import AbstractContextManager, contextmanager
 from functools import wraps
@@ -16,6 +15,7 @@ from pydantic import BaseModel
 
 from mirascope.core.base._utils._base_type import BaseType
 
+from ..core.base._utils import fn_is_async
 from ..core.base.call_response import BaseCallResponse
 from ..core.base.stream import BaseStream
 from ..core.base.structured_stream import BaseStructuredStream
@@ -136,14 +136,14 @@ def middleware_factory(
     def decorator(
         fn: Callable[_P, _R | Awaitable[_R]],
     ) -> Callable[_P, _R | Awaitable[_R]]:
-        if inspect.iscoroutinefunction(fn):
+        if fn_is_async(fn):
 
             @wraps(fn)
             async def wrapper_async(*args: _P.args, **kwargs: _P.kwargs) -> _R:
                 context_manager = custom_context_manager(fn)
                 context = context_manager.__enter__()
                 try:
-                    result = await fn(*args, **kwargs)
+                    result = await fn(*args, **kwargs)  # pyright: ignore [reportGeneralTypeIssues]
                 except Exception as e:
                     if handle_error_async:
                         try:

{mirascope-1.23.2 → mirascope-1.23.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mirascope"
-version = "1.23.2"
+version = "1.23.3"
 description = "LLM abstractions that aren't obstructions"
 readme = "README.md"
 license = { file = "LICENSE" }

{mirascope-1.23.2 → mirascope-1.23.3}/uv.lock

@@ -3170,7 +3170,7 @@ wheels = [
 
 [[package]]
 name = "mirascope"
-version = "1.23.1"
+version = "1.23.2"
 source = { editable = "." }
 dependencies = [
     { name = "docstring-parser" },