From 9e9951d314e679523518a695c64fb7a22dd16612 Mon Sep 17 00:00:00 2001 From: Charles Packer Date: Sun, 10 Mar 2024 15:40:45 -0700 Subject: [PATCH] fix: small typing fixes (#1123) --- memgpt/agent.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/memgpt/agent.py b/memgpt/agent.py index 4db481801a..65d15d223d 100644 --- a/memgpt/agent.py +++ b/memgpt/agent.py @@ -108,8 +108,8 @@ def construct_system_with_memory( system: str, memory: InContextMemory, memory_edit_timestamp: str, - archival_memory: ArchivalMemory = None, - recall_memory: RecallMemory = None, + archival_memory: Optional[ArchivalMemory] = None, + recall_memory: Optional[RecallMemory] = None, include_char_count: bool = True, ): full_system_message = "\n".join( @@ -135,9 +135,9 @@ def initialize_message_sequence( model: str, system: str, memory: InContextMemory, - archival_memory: ArchivalMemory = None, - recall_memory: RecallMemory = None, - memory_edit_timestamp: str = None, + archival_memory: Optional[ArchivalMemory] = None, + recall_memory: Optional[RecallMemory] = None, + memory_edit_timestamp: Optional[str] = None, include_initial_boot_message: bool = True, ): if memory_edit_timestamp is None: