Coverage for mindsdb / utilities / hooks / __init__.py: 92%

1def empty_fn(*args, **kwargs): 

2 pass 

3 

4 

5try: 

6 from mindsdb.utilities.hooks.after_predict import after_predict 

7except ImportError: 

8 after_predict = empty_fn 

9 

10 

11try: 

12 from mindsdb.utilities.hooks.after_api_query import after_api_query 

13except ImportError: 

14 after_api_query = empty_fn 

15 after_palm_query = empty_fn 

16 

17 

18try: 

19 from mindsdb.utilities.hooks.profiling import send_profiling_results 

20except ImportError: 

21 send_profiling_results = empty_fn 

22 

23 

24try: 

25 from mindsdb.utilities.hooks.openai_query import before_openai_query 

26except ImportError: 

27 before_openai_query = empty_fn 

28 before_palm_query = empty_fn 

29 

30try: 

31 from mindsdb.utilities.hooks.openai_query import after_openai_query 

32except ImportError: 

33 after_openai_query = empty_fn 

34 after_palm_query = empty_fn