GPTCache

1
import time
2

3
from gptcache import cache
4

5

6
def time_cal(func, func_name=None, report_func=None):
7
    def inner(*args, **kwargs):
8
        time_start = time.time()
9
        res = func(*args, **kwargs)
10
        delta_time = time.time() - time_start
11
        if cache.config.log_time_func:
12
            cache.config.log_time_func(
13
                func.__name__ if func_name is None else func_name, delta_time
14
            )
15
        if report_func is not None:
16
            report_func(delta_time)
17
        return res
18

19
    return inner
20