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
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
# coding: utf-8
|
|
# Copyright (c) 2025 inclusionAI.
|
|
|
|
import asyncio
|
|
from functools import wraps
|
|
from typing import Callable, Optional, Union, Any, Dict
|
|
|
|
|
|
class Functionable:
|
|
def __init__(self, function: Callable[..., Any], *args: Any, **kwargs: Dict[str, Any]) -> None:
|
|
self.function = function
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
self.done: bool = False
|
|
self.error: bool = False
|
|
self.result: Optional[Any] = None
|
|
self.exception: Optional[Exception] = None
|
|
|
|
def __call__(self) -> None:
|
|
try:
|
|
self.result = self.function(*self.args, **self.kwargs)
|
|
except Exception as e:
|
|
self.error = True
|
|
self.exception = e
|
|
self.done = True
|
|
|
|
def call(self):
|
|
self.__call__()
|
|
|
|
|
|
def async_decorator(*func, delay: Optional[Union[int, float]] = 0.5) -> Callable:
|
|
def wrapper(function: Callable[..., Any]) -> Callable[..., Any]:
|
|
@wraps(function)
|
|
async def inner_wrapper(*args: Any, **kwargs: Any) -> Any:
|
|
sleep_time = 0 if delay is None else delay
|
|
task = Functionable(function, *args, **kwargs)
|
|
# TODO: Use thread pool to process task
|
|
task.call()
|
|
if task.error:
|
|
raise task.exception
|
|
await asyncio.sleep(sleep_time)
|
|
return task.result
|
|
|
|
return inner_wrapper
|
|
|
|
if not func:
|
|
return wrapper
|
|
else:
|
|
if asyncio.iscoroutinefunction(func[0]):
|
|
# coroutine function, return itself
|
|
return func[0]
|
|
return wrapper(func[0])
|
|
|
|
def async_func(function: Callable[..., Any]) -> Callable[..., Any]:
|
|
@wraps(function)
|
|
async def inner_wrapper(*args: Any, **kwargs: Any) -> Any:
|
|
task = Functionable(function, *args, **kwargs)
|
|
task.call()
|
|
if task.error:
|
|
raise task.exception
|
|
return task.result
|
|
|
|
if asyncio.iscoroutinefunction(function):
|
|
# coroutine function, return itself
|
|
return function
|
|
return inner_wrapper
|