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

1"""Google A2A (Agent2Agent) Protocol Integration - Dynamic Agent System""" 

2from .google_a2a_integration import ( 

3 TaskState, AgentCard, A2ATask, A2AMessageHandler, A2AProtocolServer, 

4 initialize_a2a_server, get_a2a_server, A2A_PROTOCOL_VERSION 

5) 

6from .dynamic_agent_registry import ( 

7 DynamicAgentDiscovery, DynamicAgentExecutor, TrainedAgent, 

8 get_dynamic_discovery, get_dynamic_executor 

9) 

10from .register_dynamic_agents import ( 

11 register_all_dynamic_agents, get_registered_agent_info, list_available_agents 

12) 

13 

14# Keep old imports for backward compatibility but mark as deprecated 

15from .a2a_agent_registry import ( 

16 register_all_agents as register_all_agents_legacy, 

17 assistant_executor, helper_executor, executor_executor, verify_executor, 

18 ASSISTANT_SKILLS, HELPER_SKILLS, EXECUTOR_SKILLS, VERIFY_SKILLS 

19) 

20 

21# Primary API uses dynamic agents 

22register_all_agents = register_all_dynamic_agents 

23 

24__all__ = [ 

25 # Core A2A Protocol 

26 'TaskState', 'AgentCard', 'A2ATask', 'A2AMessageHandler', 'A2AProtocolServer', 

27 'initialize_a2a_server', 'get_a2a_server', 'A2A_PROTOCOL_VERSION', 

28 

29 # Dynamic Agent System (NEW - Primary API) 

30 'register_all_agents', 'register_all_dynamic_agents', 

31 'DynamicAgentDiscovery', 'DynamicAgentExecutor', 'TrainedAgent', 

32 'get_dynamic_discovery', 'get_dynamic_executor', 

33 'get_registered_agent_info', 'list_available_agents', 

34 

35 # Legacy Hardcoded Agents (DEPRECATED - for backward compatibility) 

36 'register_all_agents_legacy', 

37 'assistant_executor', 'helper_executor', 'executor_executor', 'verify_executor', 

38 'ASSISTANT_SKILLS', 'HELPER_SKILLS', 'EXECUTOR_SKILLS', 'VERIFY_SKILLS' 

39]