Coverage for agentos/core/__init__.py: 100%

1"""Core module - DI system, Handoff protocol, CodeAgent, Agent Loop, State Machine""" 

2 

3from agentos.core.di import ( 

4 Agent, 

5 RunContext, 

6 Depends, 

7 inject_tool, 

8 requires_context, 

9) 

10from agentos.core.handoff import ( 

11 Handoff, 

12 HandoffResult, 

13 transfer_to, 

14 can_handle, 

15) 

16from agentos.core.code_agent import ( 

17 CodeAgent, 

18 CodeResult, 

19 CodeStep, 

20) 

21from agentos.core.context import ( 

22 AgentContext, 

23 ContextManager, 

24 Message as CoreMessage, 

25 ToolCall as CoreToolCall, 

26 ToolResult as CoreToolResult, 

27) 

28from agentos.core.state_machine import ( 

29 AgentStateMachine, 

30 AgentState, 

31 StateTransition, 

32 TransitionError, 

33 StateTimeoutError, 

34) 

35from agentos.core.streaming import ( 

36 StreamChunk, 

37 StreamEmitter, 

38 StreamEvent, 

39 ResponseCollector, 

40) 

41from agentos.core.session import ( 

42 Session, 

43 SessionStore, 

44) 

45from agentos.core.async_loop import ( 

46 AsyncAgentLoop, 

47 AsyncLoopConfig, 

48 AsyncInvocationResult, 

49 AsyncContextManager, 

50) 

51 

52__all__ = [ 

53 "Agent", 

54 "RunContext", 

55 "Depends", 

56 "inject_tool", 

57 "requires_context", 

58 "Handoff", 

59 "HandoffResult", 

60 "transfer_to", 

61 "can_handle", 

62 "CodeAgent", 

63 "CodeResult", 

64 "CodeStep", 

65 "AgentContext", 

66 "ContextManager", 

67 "CoreMessage", 

68 "CoreToolCall", 

69 "CoreToolResult", 

70 "AgentStateMachine", 

71 "AgentState", 

72 "StateTransition", 

73 "TransitionError", 

74 "StateTimeoutError", 

75 "StreamChunk", 

76 "StreamEmitter", 

77 "StreamEvent", 

78 "ResponseCollector", 

79 "Session", 

80 "SessionStore", 

81 "AsyncAgentLoop", 

82 "AsyncLoopConfig", 

83 "AsyncInvocationResult", 

84 "AsyncContextManager", 

85]