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

1""" 

2Agent Lightning Integration Module 

3 

4Integrates Microsoft Agent Lightning for continuous agent training and optimization. 

5Provides minimal-change wrappers and auto-tracing for existing AutoGen agents. 

6""" 

7 

8__version__ = "1.0.0" 

9 

10# Configuration 

11from .config import ( 

12 AGENT_LIGHTNING_CONFIG, 

13 is_enabled, 

14 get_agent_config, 

15 get_reward_value 

16) 

17 

18# Core components 

19from .wrapper import AgentLightningWrapper, instrument_autogen_agent 

20from .tracer import enable_auto_tracing, disable_auto_tracing, LightningTracer, Span 

21from .rewards import RewardCalculator, RewardType 

22from .store import LightningStore 

23 

24__all__ = [ 

25 # Config 

26 'AGENT_LIGHTNING_CONFIG', 

27 'is_enabled', 

28 'get_agent_config', 

29 'get_reward_value', 

30 

31 # Core 

32 'AgentLightningWrapper', 

33 'instrument_autogen_agent', 

34 'enable_auto_tracing', 

35 'disable_auto_tracing', 

36 'LightningTracer', 

37 'Span', 

38 'RewardCalculator', 

39 'RewardType', 

40 'LightningStore', 

41]