mirascope 1.23.2__py3-none-any.whl → 1.23.3__py3-none-any.whl

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/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.dist-info → mirascope-1.23.3.dist-info}/METADATA

@@ -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.dist-info → mirascope-1.23.3.dist-info}/RECORD

@@ -114,7 +114,7 @@ mirascope/core/base/_utils/_convert_messages_to_message_params.py,sha256=symOUsX
 mirascope/core/base/_utils/_convert_provider_finish_reason_to_finish_reason.py,sha256=Mki5mYbYX8vUW-oosC4PaRNUHW_T5xAQWti3_1ndtTk,611
 mirascope/core/base/_utils/_default_tool_docstring.py,sha256=JLyryjGDaHMU-P7gUpnjkPyELCQsQgi8AP4Dp_yXPOM,277
 mirascope/core/base/_utils/_extract_tool_return.py,sha256=ZDBZJ4cacFd8nijSWZEhib7B58ZnSFD_rK1FiGNTYU0,1553
-mirascope/core/base/_utils/_fn_is_async.py,sha256=17uagVPZpo4-MkfD6fylrX1xGeGypvcGC7mibIwRbfU,409
+mirascope/core/base/_utils/_fn_is_async.py,sha256=dMrvkwqyyNhuaQUHtNMxpMoBoHqZuHPbh4nqfK-KKi0,728
 mirascope/core/base/_utils/_format_template.py,sha256=3l0Iq1LxQhocR588bzUDtfLalfiq_Rw_tHrVdgoZHiA,1061
 mirascope/core/base/_utils/_get_audio_type.py,sha256=PN8MlgKgl5OIDANgSUlI4UarLkfBHfWmWWls9shPM8M,683
 mirascope/core/base/_utils/_get_common_usage.py,sha256=s9Gvq0wl4ZM1V3zmI0LDEIG8wIxWEOXkjuDBjtyFx9s,675
@@ -321,7 +321,7 @@ mirascope/core/xai/_utils/_setup_call.py,sha256=0pCy8SqEQQEFUrCYTe6EEn_z7GuTY7-G
 mirascope/experimental/graphs/__init__.py,sha256=xwvU6T5lqfMXeHZ0IN8cepzkFkSURpeeV7ozokjzaBY,162
 mirascope/experimental/graphs/finite_state_machine.py,sha256=xsiguJCW7nyMuITezYKHevM1R2LIs1rdMeqw5kmJITs,26470
 mirascope/integrations/__init__.py,sha256=ieLWknpbkO_gABIVl9790YTTCCRO9ISQ35-1SeOSU0Q,392
-mirascope/integrations/_middleware_factory.py,sha256=rv9MaNgn7AL9Z7qADMRxxD9A_abePqj80Y0SReU9LrE,17305
+mirascope/integrations/_middleware_factory.py,sha256=v-S-hVU5S7P9Lu8PiKyUhmtp_rbQaIJ1droC8BcKBAE,17362
 mirascope/integrations/tenacity.py,sha256=jk64MBncCMbgoQMaXQgjxg9Y9UstRqTt2RCeA86pdCU,326
 mirascope/integrations/langfuse/__init__.py,sha256=wG3eBXwGPbFedB28L1K_q1iCf_dERjVmTDCWK4nHZyM,71
 mirascope/integrations/langfuse/_utils.py,sha256=SsZfQkp-_Ytxzoq03XwEy3a-PuHB6oVs3uy3fAZKMOg,3457
@@ -371,7 +371,7 @@ mirascope/v0/base/ops_utils.py,sha256=1Qq-VIwgHBaYutiZsS2MUQ4OgPC3APyywI5bTiTAmA
 mirascope/v0/base/prompts.py,sha256=FM2Yz98cSnDceYogiwPrp4BALf3_F3d4fIOCGAkd-SE,1298
 mirascope/v0/base/types.py,sha256=ZfatJoX0Yl0e3jhv0D_MhiSVHLYUeJsdN3um3iE10zY,352
 mirascope/v0/base/utils.py,sha256=XREPENRQTu8gpMhHU8RC8qH_am3FfGUvY-dJ6x8i-mw,681
-mirascope-1.23.2.dist-info/METADATA,sha256=n0lFKABsKJ5IhQy56-SCTdYMubOi9KlUUBePMfSzm3Y,8783
-mirascope-1.23.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mirascope-1.23.2.dist-info/licenses/LICENSE,sha256=LAs5Q8mdawTsVdONpDGukwsoc4KEUBmmonDEL39b23Y,1072
-mirascope-1.23.2.dist-info/RECORD,,
+mirascope-1.23.3.dist-info/METADATA,sha256=22W90rT7-eiVW-Iz5P-qoVDvwoQii6qiW2HBxIY8zTE,8783
+mirascope-1.23.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mirascope-1.23.3.dist-info/licenses/LICENSE,sha256=LAs5Q8mdawTsVdONpDGukwsoc4KEUBmmonDEL39b23Y,1072
+mirascope-1.23.3.dist-info/RECORD,,