Coverage for integrations / service_tools / __init__.py: 100.0%

1""" 

2Service Tools - Dynamic HTTP tool registry for external microservices. 

3 

4Extends the MCP integration pattern to support any HTTP-based tool service 

5(HevolveAI, AceStep, Wan2GP, TTS-Audio-Suite, etc.) with health checking, 

6auto-discovery, and autogen/langchain compatible function generation. 

7 

8Runtime tools are managed by RuntimeToolManager which handles: 

9- Model download (git clone / HuggingFace) 

10- VRAM-aware start/stop of sidecar servers (dynamic ports) 

11- State persistence (skip re-download on restart) 

12- Auto-registration with service_tool_registry → AutoGen + LangChain 

13""" 

14 

15from .registry import ServiceToolRegistry, ServiceToolInfo, service_tool_registry 

16from .crawl4ai_tool import Crawl4AITool 

17from .acestep_tool import AceStepTool 

18from .wan2gp_tool import Wan2GPTool 

19from .tts_audio_suite_tool import TTSAudioSuiteTool 

20from .whisper_tool import WhisperTool 

21from .omniparser_tool import OmniParserTool 

22from .model_storage import ModelStorageManager, model_storage 

23from .vram_manager import VRAMManager, vram_manager 

24from .runtime_manager import RuntimeToolManager, runtime_tool_manager 

25from .model_lifecycle import ModelLifecycleManager, get_model_lifecycle_manager 

26from .media_agent import (generate_media, check_media_status, 

27 synthesize_multilingual_audio, register_media_tools) 

28 

29__all__ = [ 

30 "ServiceToolRegistry", 

31 "ServiceToolInfo", 

32 "service_tool_registry", 

33 "Crawl4AITool", 

34 "AceStepTool", 

35 "Wan2GPTool", 

36 "TTSAudioSuiteTool", 

37 "WhisperTool", 

38 "OmniParserTool", 

39 "ModelStorageManager", 

40 "model_storage", 

41 "VRAMManager", 

42 "vram_manager", 

43 "RuntimeToolManager", 

44 "runtime_tool_manager", 

45 "ModelLifecycleManager", 

46 "get_model_lifecycle_manager", 

47 "generate_media", 

48 "check_media_status", 

49 "synthesize_multilingual_audio", 

50 "register_media_tools", 

51]