Coverage for src/lite_agent/types/__init__.py: 100%

1# Export all types from submodules 

2from .events import ( 

3 AgentChunk, 

4 AgentChunkType, 

5 AssistantMessageEvent, 

6 CompletionRawEvent, 

7 ContentDeltaEvent, 

8 FunctionCallDeltaEvent, 

9 FunctionCallEvent, 

10 FunctionCallOutputEvent, 

11 ResponseRawEvent, 

12 Timing, 

13 TimingEvent, 

14 UsageEvent, 

15) 

16from .events import ( 

17 Usage as EventUsage, 

18) 

19from .messages import ( 

20 # Legacy compatibility aliases (temporary) 

21 AgentAssistantMessage, 

22 AgentSystemMessage, 

23 AgentUserMessage, 

24 AssistantMessage, 

25 # New types 

26 AssistantMessageContent, 

27 AssistantMessageDict, 

28 AssistantMessageMeta, 

29 AssistantTextContent, 

30 AssistantToolCall, 

31 AssistantToolCallResult, 

32 BasicMessageMeta, 

33 FlexibleInputMessage, 

34 FlexibleRunnerMessage, 

35 FunctionCallDict, 

36 FunctionCallOutputDict, 

37 LLMResponseMeta, 

38 MessageDict, 

39 # New metadata types 

40 MessageMeta, 

41 MessageUsage, 

42 NewAssistantMessage, 

43 NewMessage, 

44 NewMessages, 

45 NewSystemMessage, 

46 # New structured message types 

47 NewUserMessage, 

48 ResponseInputImage, 

49 ResponseInputText, 

50 RunnerMessage, 

51 RunnerMessages, 

52 SystemMessageDict, 

53 UserFileContent, 

54 UserImageContent, 

55 UserInput, 

56 # New content types 

57 UserMessageContent, 

58 UserMessageContentItemImageURL, 

59 UserMessageContentItemImageURLImageURL, 

60 UserMessageContentItemText, 

61 UserMessageDict, 

62 UserTextContent, 

63 # Utility functions 

64 assistant_message_to_llm_dict, 

65 message_to_llm_dict, 

66 messages_to_llm_format, 

67 system_message_to_llm_dict, 

68 user_message_to_llm_dict, 

69) 

70from .tool_calls import ToolCall, ToolCallFunction 

71 

72__all__ = [ 

73 "AgentAssistantMessage", 

74 "AgentChunk", 

75 "AgentChunkType", 

76 "AgentSystemMessage", 

77 "AgentUserMessage", 

78 "AssistantMessage", 

79 "AssistantMessageContent", 

80 "AssistantMessageDict", 

81 "AssistantMessageEvent", 

82 "AssistantMessageMeta", 

83 "AssistantTextContent", 

84 "AssistantToolCall", 

85 "AssistantToolCallResult", 

86 "BasicMessageMeta", 

87 "CompletionRawEvent", 

88 "ContentDeltaEvent", 

89 "EventUsage", 

90 "FlexibleInputMessage", 

91 "FlexibleRunnerMessage", 

92 "FunctionCallDeltaEvent", 

93 "FunctionCallDict", 

94 "FunctionCallEvent", 

95 "FunctionCallOutputDict", 

96 "FunctionCallOutputEvent", 

97 "LLMResponseMeta", 

98 "MessageDict", 

99 "MessageMeta", 

100 "MessageUsage", 

101 "NewAssistantMessage", 

102 "NewMessage", 

103 "NewMessages", 

104 "NewSystemMessage", 

105 "NewUserMessage", 

106 "ResponseInputImage", 

107 "ResponseInputText", 

108 "ResponseRawEvent", 

109 "RunnerMessage", 

110 "RunnerMessages", 

111 "SystemMessageDict", 

112 "Timing", 

113 "TimingEvent", 

114 "ToolCall", 

115 "ToolCallFunction", 

116 "UsageEvent", 

117 "UserFileContent", 

118 "UserImageContent", 

119 "UserInput", 

120 "UserMessageContent", 

121 "UserMessageContentItemImageURL", 

122 "UserMessageContentItemImageURLImageURL", 

123 "UserMessageContentItemText", 

124 "UserMessageDict", 

125 "UserTextContent", 

126 "assistant_message_to_llm_dict", 

127 "message_to_llm_dict", 

128 "messages_to_llm_format", 

129 "system_message_to_llm_dict", 

130 "user_message_to_llm_dict", 

131]