mirascope 1.25.7__py3-none-any.whl → 2.0.0a0__py3-none-any.whl

mirascope/integrations/_middleware_factory.py

@@ -1,403 +0,0 @@
-"""The `middleware_factory` method for handling the call response."""
-
-from collections.abc import AsyncGenerator, Awaitable, Callable, Generator
-from contextlib import AbstractContextManager, contextmanager
-from functools import wraps
-from typing import (
-    Any,
-    ParamSpec,
-    TypeVar,
-    cast,
-    overload,
-)
-
-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
-
-_BaseCallResponseT = TypeVar("_BaseCallResponseT", bound=BaseCallResponse)
-_BaseStructuredStreamT = TypeVar("_BaseStructuredStreamT", bound=BaseStructuredStream)
-_BaseStreamT = TypeVar("_BaseStreamT", bound=BaseStream)
-_ResponseModelT = TypeVar("_ResponseModelT", bound=BaseModel | BaseType)
-ResponseModel = BaseModel | BaseType
-_P = ParamSpec("_P")
-SyncFunc = Callable[
-    _P, _BaseCallResponseT | _BaseStreamT | _ResponseModelT | _BaseStructuredStreamT
-]
-AsyncFunc = Callable[
-    _P,
-    Awaitable[
-        _BaseCallResponseT | _BaseStreamT | _ResponseModelT | _BaseStructuredStreamT
-    ],
-]
-_T = TypeVar("_T")
-_R = TypeVar("_R")
-
-
-@contextmanager
-def default_context_manager(
-    fn: SyncFunc | AsyncFunc,
-) -> Generator[None, None, None]:
-    yield None
-
-
-def middleware_factory(
-    *,
-    custom_context_manager: Callable[
-        [SyncFunc | AsyncFunc], AbstractContextManager[_T]
-    ] = default_context_manager,
-    custom_decorator: Callable | None = None,
-    handle_call_response: Callable[
-        [BaseCallResponse, SyncFunc | AsyncFunc, _T | None], None
-    ]
-    | None = None,
-    handle_call_response_async: Callable[
-        [BaseCallResponse, SyncFunc | AsyncFunc, _T | None], Awaitable[None]
-    ]
-    | None = None,
-    handle_stream: Callable[[BaseStream, SyncFunc | AsyncFunc, _T | None], None]
-    | None = None,
-    handle_stream_async: Callable[
-        [BaseStream, SyncFunc | AsyncFunc, _T | None], Awaitable[None]
-    ]
-    | None = None,
-    handle_response_model: Callable[
-        [ResponseModel, SyncFunc | AsyncFunc, _T | None], None
-    ]
-    | None = None,
-    handle_response_model_async: Callable[
-        [ResponseModel, SyncFunc | AsyncFunc, _T | None], Awaitable[None]
-    ]
-    | None = None,
-    handle_structured_stream: Callable[
-        [BaseStructuredStream, SyncFunc | AsyncFunc, _T | None], None
-    ]
-    | None = None,
-    handle_structured_stream_async: Callable[
-        [BaseStructuredStream, SyncFunc | AsyncFunc, _T | None], Awaitable[None]
-    ]
-    | None = None,
-    handle_error: Callable[[Exception, SyncFunc | AsyncFunc, _T | None], Any]
-    | None = None,
-    handle_error_async: Callable[
-        [Exception, SyncFunc | AsyncFunc, _T | None], Awaitable[Any]
-    ]
-    | None = None,
-) -> Callable[[Callable[_P, _R]], Callable[_P, _R]]:
-    '''A factory method for creating middleware decorators.
-
-    Example:
-
-    ```python
-    from mirascope.core import openai, prompt_template
-    from mirascope.integrations import middleware_factory
-
-    def with_saving():
-        """Saves some data after a Mirascope call."""
-
-        return middleware_factory(
-            custom_context_manager=custom_context_manager,
-            custom_decorator=custom_decorator,
-            handle_call_response=handle_call_response,
-            handle_call_response_async=handle_call_response_async,
-            handle_stream=handle_stream,
-            handle_stream_async=handle_stream_async,
-            handle_response_model=handle_response_model,
-            handle_response_model_async=handle_response_model_async,
-            handle_structured_stream=handle_structured_stream,
-            handle_structured_stream_async=handle_structured_stream_async,
-            handle_error_async=handle_error_async,
-            handle_error=handle_error,
-        )
-
-
-    @with_saving()
-    @openai.call("gpt-4o-mini")
-    def recommend_book(genre: str) -> str:
-        return f"Recommend a {genre} book"
-
-
-    response = recommend_book("fantasy")  # `with_saving` automatically run
-    print(response.content)
-    ```
-    '''
-
-    @overload
-    def decorator(fn: Callable[_P, _R]) -> Callable[_P, _R]: ...
-
-    @overload
-    def decorator(fn: Callable[_P, Awaitable[_R]]) -> Callable[_P, Awaitable[_R]]: ...
-
-    def decorator(
-        fn: Callable[_P, _R | Awaitable[_R]],
-    ) -> Callable[_P, _R | Awaitable[_R]]:
-        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)  # pyright: ignore [reportGeneralTypeIssues]
-                except Exception as e:
-                    if handle_error_async:
-                        try:
-                            result = await handle_error_async(e, fn, context)
-                        except Exception as new_e:
-                            # If handle_error_async raises an exception, exit context manager and re-raise
-                            context_manager.__exit__(
-                                type(new_e), new_e, new_e.__traceback__
-                            )
-                            raise
-                        else:
-                            # Exception was handled, exit context manager and continue
-                            context_manager.__exit__(None, None, None)
-                    else:
-                        # No handle_error_async provided, exit context manager with exception and re-raise
-                        context_manager.__exit__(type(e), e, e.__traceback__)
-                        raise
-
-                if (
-                    isinstance(result, BaseCallResponse)
-                    and handle_call_response_async is not None
-                ):
-                    await handle_call_response_async(result, fn, context)
-                    context_manager.__exit__(None, None, None)
-                elif isinstance(result, BaseStream):
-                    original_class = type(result)
-                    original_aiter = result.__aiter__
-
-                    def new_stream_aiter(
-                        self: Any,  # noqa: ANN401
-                    ) -> AsyncGenerator[tuple[Any, Any | None], Any]:  # noqa: ANN401
-                        async def generator() -> AsyncGenerator[
-                            tuple[Any, Any | None], Any
-                        ]:
-                            try:
-                                async for chunk, tool in original_aiter():
-                                    yield chunk, tool
-                            except Exception as e:
-                                if handle_error_async:
-                                    try:
-                                        await handle_error_async(e, fn, context)
-                                    except Exception as new_e:
-                                        context_manager.__exit__(
-                                            type(new_e), new_e, new_e.__traceback__
-                                        )
-                                        raise
-                                    else:
-                                        context_manager.__exit__(None, None, None)
-                                        return
-                                else:
-                                    context_manager.__exit__(
-                                        type(e), e, e.__traceback__
-                                    )
-                                    raise
-                            finally:
-                                if handle_stream_async is not None:
-                                    await handle_stream_async(result, fn, context)
-                                context_manager.__exit__(None, None, None)
-
-                        return generator()
-
-                    class MiddlewareAsyncStream(original_class):
-                        __aiter__ = (
-                            custom_decorator(fn)(new_stream_aiter)
-                            if custom_decorator
-                            else new_stream_aiter
-                        )
-
-                    result.__class__ = MiddlewareAsyncStream
-                elif (
-                    isinstance(result, ResponseModel)
-                    and handle_response_model_async is not None
-                ):
-                    await handle_response_model_async(result, fn, context)
-                    context_manager.__exit__(None, None, None)
-                elif isinstance(result, BaseStructuredStream):
-                    original_class = type(result)
-                    original_aiter = result.__aiter__
-
-                    def new_aiter(
-                        self: Any,  # noqa: ANN401
-                    ) -> AsyncGenerator[tuple[Any, Any | None], Any]:  # noqa: ANN401
-                        async def generator() -> AsyncGenerator[
-                            tuple[Any, Any | None], Any
-                        ]:
-                            try:
-                                async for chunk in original_aiter():
-                                    yield chunk
-                            except Exception as e:
-                                if handle_error_async:
-                                    try:
-                                        await handle_error_async(e, fn, context)
-                                    except Exception as new_e:
-                                        context_manager.__exit__(
-                                            type(new_e), new_e, new_e.__traceback__
-                                        )
-                                        raise
-                                    else:
-                                        context_manager.__exit__(None, None, None)
-                                        return
-                                else:
-                                    context_manager.__exit__(
-                                        type(e), e, e.__traceback__
-                                    )
-                                    raise
-                            finally:
-                                if handle_structured_stream_async is not None:
-                                    await handle_structured_stream_async(
-                                        result, fn, context
-                                    )
-                                context_manager.__exit__(None, None, None)
-
-                        return generator()
-
-                    class MiddlewareAsyncStructuredStream(original_class):
-                        __aiter__ = (
-                            custom_decorator(fn)(new_aiter)
-                            if custom_decorator
-                            else new_aiter
-                        )
-
-                    result.__class__ = MiddlewareAsyncStructuredStream
-                else:
-                    context_manager.__exit__(None, None, None)
-                return cast(_R, result)
-
-            return (
-                custom_decorator(fn)(wrapper_async)
-                if custom_decorator
-                else wrapper_async
-            )
-        else:
-
-            @wraps(fn)
-            def wrapper(*args: _P.args, **kwargs: _P.kwargs) -> _R:
-                context_manager = custom_context_manager(fn)
-                context = context_manager.__enter__()
-                try:
-                    result = fn(*args, **kwargs)
-                except Exception as e:
-                    if handle_error:
-                        try:
-                            result = handle_error(e, fn, context)
-                        except Exception as new_e:
-                            # If handle_error raises an exception, exit context manager and re-raise
-                            context_manager.__exit__(
-                                type(new_e), new_e, new_e.__traceback__
-                            )
-                            raise
-                        else:
-                            # Exception was handled, exit context manager and continue
-                            context_manager.__exit__(None, None, None)
-                    else:
-                        # No handle_error provided, exit context manager with exception and re-raise
-                        context_manager.__exit__(type(e), e, e.__traceback__)
-                        raise
-
-                if (
-                    isinstance(result, BaseCallResponse)
-                    and handle_call_response is not None
-                ):
-                    handle_call_response(result, fn, context)
-                    context_manager.__exit__(None, None, None)
-                elif isinstance(result, BaseStream):
-                    original_class = type(result)
-                    original_iter = result.__iter__
-
-                    def new_stream_iter(
-                        self: Any,  # noqa: ANN401
-                    ) -> Generator[tuple[Any, Any | None], None, None]:  # noqa: ANN401
-                        # Create the context manager when user iterates over the stream
-                        try:
-                            yield from original_iter()
-                        except Exception as e:
-                            if handle_error:
-                                try:
-                                    handle_error(e, fn, context)
-                                except Exception as new_e:
-                                    context_manager.__exit__(
-                                        type(new_e), new_e, new_e.__traceback__
-                                    )
-                                    raise
-                                else:
-                                    context_manager.__exit__(None, None, None)
-                                    return
-                            else:
-                                context_manager.__exit__(type(e), e, e.__traceback__)
-                                raise
-                        finally:
-                            if handle_stream is not None:
-                                handle_stream(result, fn, context)
-                            context_manager.__exit__(None, None, None)
-
-                    class MiddlewareStream(original_class):  # pyright: ignore [reportGeneralTypeIssues]
-                        __iter__ = (
-                            custom_decorator(fn)(new_stream_iter)
-                            if custom_decorator
-                            else new_stream_iter
-                        )
-
-                    result.__class__ = MiddlewareStream
-                elif (
-                    isinstance(result, ResponseModel)
-                    and handle_response_model is not None
-                ):
-                    handle_response_model(result, fn, context)
-                    context_manager.__exit__(None, None, None)
-                elif isinstance(result, BaseStructuredStream):
-                    original_class = type(result)
-                    original_iter = result.__iter__
-
-                    def new_iter(
-                        self: Any,  # noqa: ANN401
-                    ) -> Generator[
-                        Generator[tuple[Any, Any | None], None, None]
-                        | Generator[Any, None, None],
-                        Any,
-                        None,
-                    ]:  # noqa: ANN401
-                        # Create the context manager when user iterates over the stream
-                        try:
-                            yield from original_iter()
-                        except Exception as e:
-                            if handle_error:
-                                try:
-                                    handle_error(e, fn, context)
-                                except Exception as new_e:
-                                    context_manager.__exit__(
-                                        type(new_e), new_e, new_e.__traceback__
-                                    )
-                                    raise
-                                else:
-                                    context_manager.__exit__(None, None, None)
-                                    return
-                            else:
-                                context_manager.__exit__(type(e), e, e.__traceback__)
-                                raise
-                        finally:
-                            if handle_structured_stream is not None:
-                                handle_structured_stream(result, fn, context)
-                            context_manager.__exit__(None, None, None)
-
-                    class MiddlewareStructuredStream(original_class):  # pyright: ignore [reportGeneralTypeIssues]
-                        __iter__ = (
-                            custom_decorator(fn)(new_iter)
-                            if custom_decorator
-                            else new_iter
-                        )
-
-                    result.__class__ = MiddlewareStructuredStream
-                else:
-                    context_manager.__exit__(None, None, None)
-                return cast(_R, result)
-
-            return custom_decorator(fn)(wrapper) if custom_decorator else wrapper
-
-    return decorator

mirascope/integrations/langfuse/__init__.py

@@ -1,3 +0,0 @@
-from ._with_langfuse import with_langfuse
-
-__all__ = ["with_langfuse"]

mirascope/integrations/langfuse/_utils.py

@@ -1,114 +0,0 @@
-from collections.abc import Callable
-from typing import Any
-
-from langfuse.decorators import langfuse_context
-from pydantic import BaseModel
-
-from mirascope.core.base._utils._base_type import BaseType
-
-from ...core.base import BaseCallResponse
-from ...core.base._utils import get_metadata
-from ...core.base.stream import BaseStream
-from ...core.base.structured_stream import BaseStructuredStream
-
-
-class ModelUsage(BaseModel):
-    input: int | float | None
-    output: int | float | None
-    unit: str
-
-
-def _get_call_response_observation(
-    result: BaseCallResponse, fn: Callable
-) -> dict[str, Any]:
-    metadata = get_metadata(fn, {})
-    tags = metadata.get("tags", [])
-    return {
-        "name": f"{fn.__name__} with {result.model}",
-        "input": result.messages,
-        "metadata": result.response,
-        "tags": tags,
-        "model": result.model,
-        "output": result.content,
-    }
-
-
-def handle_call_response(result: BaseCallResponse, fn: Callable, context: None) -> None:
-    langfuse_context.update_current_observation(
-        **_get_call_response_observation(result, fn),
-        usage=ModelUsage(
-            input=result.input_tokens, output=result.output_tokens, unit="TOKENS"
-        ),
-    )
-
-
-def handle_stream(stream: BaseStream, fn: Callable, context: None) -> None:
-    usage = ModelUsage(
-        input=stream.input_tokens,
-        output=stream.output_tokens,
-        unit="TOKENS",
-    )
-    langfuse_context.update_current_observation(
-        **_get_call_response_observation(stream.construct_call_response(), fn),
-        usage=usage,
-    )
-
-
-def handle_response_model(
-    result: BaseModel | BaseType, fn: Callable, context: None
-) -> None:
-    if isinstance(result, BaseModel):
-        response: BaseCallResponse = result._response  # pyright: ignore [reportAttributeAccessIssue]
-        call_response_observation = _get_call_response_observation(response, fn)
-        call_response_observation.pop("output")
-        langfuse_context.update_current_observation(
-            **call_response_observation,
-            usage=ModelUsage(
-                input=response.input_tokens,
-                output=response.output_tokens,
-                unit="TOKENS",
-            ),
-            output=result,
-        )
-    else:
-        langfuse_context.update_current_observation(
-            output=result,
-        )
-
-
-def handle_structured_stream(
-    result: BaseStructuredStream, fn: Callable, context: None
-) -> None:
-    stream: BaseStream = result.stream
-    usage = ModelUsage(
-        input=stream.input_tokens,
-        output=stream.output_tokens,
-        unit="TOKENS",
-    )
-    langfuse_context.update_current_observation(
-        **_get_call_response_observation(stream.construct_call_response(), fn),
-        usage=usage,
-        output=result.constructed_response_model,
-    )
-
-
-async def handle_call_response_async(
-    result: BaseCallResponse, fn: Callable, context: None
-) -> None:
-    handle_call_response(result, fn, None)
-
-
-async def handle_stream_async(stream: BaseStream, fn: Callable, context: None) -> None:
-    handle_stream(stream, fn, None)
-
-
-async def handle_response_model_async(
-    result: BaseModel | BaseType, fn: Callable, context: None
-) -> None:
-    handle_response_model(result, fn, None)
-
-
-async def handle_structured_stream_async(
-    result: BaseStructuredStream, fn: Callable, context: None
-) -> None:
-    handle_structured_stream(result, fn, None)

mirascope/integrations/langfuse/_with_langfuse.py

@@ -1,70 +0,0 @@
-"""Mirascope x Langfuse Integration."""
-
-from collections.abc import Callable
-from typing import Any, ParamSpec, TypeVar
-
-from langfuse.decorators import observe
-
-from .._middleware_factory import middleware_factory
-from ._utils import (
-    handle_call_response,
-    handle_call_response_async,
-    handle_response_model,
-    handle_response_model_async,
-    handle_stream,
-    handle_stream_async,
-    handle_structured_stream,
-    handle_structured_stream_async,
-)
-
-F = TypeVar("F", bound=Callable[..., Any])
-
-
-def custom_decorator(fn: Callable) -> Callable[[F], F]:
-    return observe(
-        name=fn.__name__,
-        as_type="generation",
-        capture_input=False,
-        capture_output=False,
-    )
-
-
-_P = ParamSpec("_P")
-_R = TypeVar("_R")
-
-
-def with_langfuse() -> Callable[[Callable[_P, _R]], Callable[_P, _R]]:
-    """Wraps a Mirascope function with Langfuse.
-
-    Example:
-
-    ```python
-    from mirascope.core import anthropic, prompt_template
-    from mirascope.integrations.langfuse import with_langfuse
-
-
-    def format_book(title: str, author: str):
-        return f"{title} by {author}"
-
-
-    @with_langfuse()
-    @anthropic.call(model="claude-3-5-sonnet-20240620", tools=[format_book])
-    def recommend_book(genre: str) -> str:
-        return f"Recommend a {genre} book"
-
-
-    print(recommend_book("fantasy"))
-    ```
-    """
-
-    return middleware_factory(
-        custom_decorator=custom_decorator,
-        handle_call_response=handle_call_response,
-        handle_call_response_async=handle_call_response_async,
-        handle_stream=handle_stream,
-        handle_stream_async=handle_stream_async,
-        handle_response_model=handle_response_model,
-        handle_response_model_async=handle_response_model_async,
-        handle_structured_stream=handle_structured_stream,
-        handle_structured_stream_async=handle_structured_stream_async,
-    )

mirascope/integrations/logfire/__init__.py

@@ -1,3 +0,0 @@
-from ._with_logfire import with_logfire
-
-__all__ = ["with_logfire"]