Coverage for src/lite_agent/constants.py: 100%

1from typing import Literal 

2 

3 

4class CompletionMode: 

5 """Agent completion modes.""" 

6 

7 STOP: Literal["stop"] = "stop" # Traditional completion until model decides to stop 

8 CALL: Literal["call"] = "call" # Completion until specific tool is called 

9 

10 

11class ToolName: 

12 """System tool names.""" 

13 

14 TRANSFER_TO_AGENT = "transfer_to_agent" 

15 TRANSFER_TO_PARENT = "transfer_to_parent" 

16 WAIT_FOR_USER = "wait_for_user" 

17 

18 

19class StreamIncludes: 

20 """Default stream includes configuration.""" 

21 

22 DEFAULT_INCLUDES = ( 

23 "completion_raw", 

24 "usage", 

25 "function_call", 

26 "function_call_output", 

27 "content_delta", 

28 "function_call_delta", 

29 "assistant_message", 

30 )