22from importlib import import_module
33from typing import Any
44
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+
514# Map of function names to their module and class (if any)
615TOOL_MAPPINGS = {
716 # Direct functions
@@ -199,4 +208,11 @@ def __getattr__(name: str) -> Any:
199208 method = getattr (_instances [class_name ], name )
200209 return method
201210
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