liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from importlib import import_module
|
|
from wrapt import ObjectProxy
|
|
|
|
|
|
def unwrap(obj: object, attr: str):
|
|
"""Given a function that was wrapped by wrapt.wrap_function_wrapper, unwrap it
|
|
|
|
The object containing the function to unwrap may be passed as dotted module path string.
|
|
|
|
Args:
|
|
obj: Object that holds a reference to the wrapped function or dotted import path as string
|
|
attr (str): Name of the wrapped function
|
|
"""
|
|
if isinstance(obj, str):
|
|
try:
|
|
module_path, class_name = obj.rsplit(".", 1)
|
|
except ValueError as exc:
|
|
raise ImportError(
|
|
f"Cannot parse '{obj}' as dotted import path"
|
|
) from exc
|
|
module = import_module(module_path)
|
|
try:
|
|
obj = getattr(module, class_name)
|
|
except AttributeError as exc:
|
|
raise ImportError(
|
|
f"Cannot import '{class_name}' from '{module}'"
|
|
) from exc
|
|
|
|
func = getattr(obj, attr, None)
|
|
if func and isinstance(func, ObjectProxy) and hasattr(func, "__wrapped__"):
|
|
setattr(obj, attr, func.__wrapped__)
|