55
66
77class AssistantMemoryType (str , Enum ):
8- zero = "zero"
9- naive = "naive"
10- message_window = "message_window"
8+ ZERO = "zero"
9+ NAIVE = "naive"
10+ MESSAGE_WINDOW = "message_window"
1111
1212
1313class AssistantMemory (BaseModel , ABC ):
1414 type : AssistantMemoryType = Field (..., description = "The type of the memory." )
1515
1616
1717class AssistantMessageWindowMemory (AssistantMemory ):
18- type : AssistantMemoryType = Field (AssistantMemoryType .message_window , Literal = AssistantMemoryType .message_window )
18+ type : AssistantMemoryType = Field (AssistantMemoryType .MESSAGE_WINDOW , Literal = AssistantMemoryType .MESSAGE_WINDOW )
1919 max_messages : int = Field (...)
2020 max_tokens : int = Field (...)
2121
2222
2323class AssistantNaiveMemory (AssistantMemory ):
24- type : AssistantMemoryType = Field (AssistantMemoryType .naive , Literal = AssistantMemoryType .naive )
24+ type : AssistantMemoryType = Field (AssistantMemoryType .NAIVE , Literal = AssistantMemoryType .NAIVE )
2525
2626
2727class AssistantZeroMemory (AssistantMemory ):
28- type : AssistantMemoryType = Field (AssistantMemoryType .zero , Literal = AssistantMemoryType .zero )
28+ type : AssistantMemoryType = Field (AssistantMemoryType .ZERO , Literal = AssistantMemoryType .ZERO )
2929
3030
3131def build_assistant_memory (memory_dict : Dict ) -> Optional [AssistantMemory ]:
@@ -35,15 +35,15 @@ def build_assistant_memory(memory_dict: Dict) -> Optional[AssistantMemory]:
3535
3636 memory_type = memory_dict ['type' ]
3737
38- if memory_type == AssistantMemoryType .zero .value :
38+ if memory_type == AssistantMemoryType .ZERO .value :
3939 # For zero memory, no additional information is needed
4040 return AssistantZeroMemory ()
4141
42- elif memory_type == AssistantMemoryType .naive .value :
42+ elif memory_type == AssistantMemoryType .NAIVE .value :
4343 # For naive memory, no additional configuration is needed
4444 return AssistantNaiveMemory ()
4545
46- elif memory_type == AssistantMemoryType .message_window .value :
46+ elif memory_type == AssistantMemoryType .MESSAGE_WINDOW .value :
4747 # For message window memory, additional configuration is needed
4848 max_messages = memory_dict .get ('max_messages' )
4949 max_tokens = memory_dict .get ('max_tokens' )
0 commit comments