Coverage for src / osiris_cli / tool_migration.py: 0%

1#!/usr/bin/env python3 

2""" 

3Tool Migration - Register existing tools with new registry 

4""" 

5 

6from .tools_registry import tools_registry 

7from .tools import ToolRegistry as OldToolRegistry 

8from .logger import get_logger 

9 

10# Import existing tool implementations 

11old_registry = OldToolRegistry() 

12 

13 

14def migrate_tools(): 

15 """Migrate all tools from old registry to new registry""" 

16 logger = get_logger() 

17 

18 # Get all tools from old registry 

19 old_tools = old_registry.tools 

20 

21 for name, tool_data in old_tools.items(): 

22 # Extract function and metadata 

23 func = tool_data["func"] 

24 function_def = tool_data["function"] 

25 

26 # Register with new registry 

27 tools_registry.register( 

28 name=name, 

29 description=function_def["description"], 

30 parameters=function_def["parameters"], 

31 function=func, 

32 requires_approval=False, # All tools auto-approved in YOLO mode 

33 category="core" 

34 ) 

35 

36 logger.info(f"Migrated {len(old_tools)} tools to new registry") 

37 

38 

39# Auto-migrate on import 

40migrate_tools()