decorators

P = ParamSpec('P') module-attribute

T = TypeVar('T') module-attribute

observable(wrapped=None, **dkwargs)

observable(
    wrapped: None = None, **dkwargs: Any
) -> Callable[[Callable[P, T]], Callable[P, T]]

observable(
    wrapped: Callable[P, T], **dkwargs: Any
) -> Callable[P, T]

Source code in griptape/common/decorators.py

def observable(wrapped: Optional[Callable[P, T] | Any] = None, **dkwargs: Any) -> Any:
    if wrapped is None:
        return functools.partial(observable, **dkwargs)

    if not callable(wrapped):
        raise ValueError("Non-callable positional argument passed. Use kwargs to pass arguments to the decorator.")

    @wrapt.decorator
    def wrapper(wrapped: Callable[P, T], instance: Any, args: Any, kwargs: Any) -> T:
        from griptape.common.observable import Observable
        from griptape.observability.observability import Observability

        return cast(
            "T",
            Observability.observe(
                Observable.Call(
                    func=wrapped,
                    instance=instance,
                    args=args,
                    kwargs=kwargs,
                    decorator_args=(),
                    decorator_kwargs=dkwargs,
                )
            ),
        )

    return wrapper(wrapped)  # pyright: ignore[reportCallIssue]