2
2
from importlib import import_module
3
3
from typing import Any
4
4
5
+ # Import advanced tools functionality
6
+ from .advanced_tools import (
7
+ tool , cache , external , user_input ,
8
+ Field , InputGroup , Choice , Range , Pattern ,
9
+ ToolContext , Hook , CacheConfig , ExternalConfig , Priority ,
10
+ set_global_hooks , clear_global_hooks , register_external_handler ,
11
+ invalidate_cache , clear_all_caches , get_cache_stats
12
+ )
13
+
5
14
# Map of function names to their module and class (if any)
6
15
TOOL_MAPPINGS = {
7
16
# Direct functions
@@ -199,4 +208,11 @@ def __getattr__(name: str) -> Any:
199
208
method = getattr (_instances [class_name ], name )
200
209
return method
201
210
202
- __all__ = list (TOOL_MAPPINGS .keys ())
211
+ __all__ = list (TOOL_MAPPINGS .keys ()) + [
212
+ # Advanced tools functionality
213
+ 'tool' , 'cache' , 'external' , 'user_input' ,
214
+ 'Field' , 'InputGroup' , 'Choice' , 'Range' , 'Pattern' ,
215
+ 'ToolContext' , 'Hook' , 'CacheConfig' , 'ExternalConfig' , 'Priority' ,
216
+ 'set_global_hooks' , 'clear_global_hooks' , 'register_external_handler' ,
217
+ 'invalidate_cache' , 'clear_all_caches' , 'get_cache_stats'
218
+ ]
0 commit comments