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

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

2 

3from agentos.core.async_loop import ( 

4 AsyncAgentLoop, 

5 AsyncContextManager, 

6 AsyncInvocationResult, 

7 AsyncLoopConfig, 

8) 

9from agentos.core.code_agent import ( 

10 CodeAgent, 

11 CodeResult, 

12 CodeStep, 

13) 

14from agentos.core.context import ( 

15 AgentContext, 

16 ContextManager, 

17) 

18from agentos.core.context import ( 

19 Message as CoreMessage, 

20) 

21from agentos.core.context import ( 

22 ToolCall as CoreToolCall, 

23) 

24from agentos.core.context import ( 

25 ToolResult as CoreToolResult, 

26) 

27from agentos.core.di import ( 

28 Agent, 

29 Depends, 

30 RunContext, 

31 inject_tool, 

32 requires_context, 

33) 

34from agentos.core.handoff import ( 

35 Handoff, 

36 HandoffResult, 

37 can_handle, 

38 transfer_to, 

39) 

40from agentos.core.session import ( 

41 Session, 

42 SessionStore, 

43) 

44from agentos.core.state_machine import ( 

45 AgentState, 

46 AgentStateMachine, 

47 StateTimeoutError, 

48 StateTransition, 

49 TransitionError, 

50) 

51from agentos.core.streaming import ( 

52 ResponseCollector, 

53 StreamChunk, 

54 StreamEmitter, 

55 StreamEvent, 

56) 

57 

58__all__ = [ 

59 "Agent", 

60 "RunContext", 

61 "Depends", 

62 "inject_tool", 

63 "requires_context", 

64 "Handoff", 

65 "HandoffResult", 

66 "transfer_to", 

67 "can_handle", 

68 "CodeAgent", 

69 "CodeResult", 

70 "CodeStep", 

71 "AgentContext", 

72 "ContextManager", 

73 "CoreMessage", 

74 "CoreToolCall", 

75 "CoreToolResult", 

76 "AgentStateMachine", 

77 "AgentState", 

78 "StateTransition", 

79 "TransitionError", 

80 "StateTimeoutError", 

81 "StreamChunk", 

82 "StreamEmitter", 

83 "StreamEvent", 

84 "ResponseCollector", 

85 "Session", 

86 "SessionStore", 

87 "AsyncAgentLoop", 

88 "AsyncLoopConfig", 

89 "AsyncInvocationResult", 

90 "AsyncContextManager", 

91]