File tree Expand file tree Collapse file tree 1 file changed +9
-12
lines changed Expand file tree Collapse file tree 1 file changed +9
-12
lines changed Original file line number Diff line number Diff line change @@ -1326,8 +1326,7 @@ def update_memory(
13261326        message : BaseMessage ,
13271327        role : OpenAIBackendRole ,
13281328        timestamp : Optional [float ] =  None ,
1329-         return_records : bool  =  False ,
1330-     ) ->  Optional [List [MemoryRecord ]]:
1329+     ) ->  None :
13311330        r"""Updates the agent memory with a new message. 
13321331
13331332        Args: 
@@ -1339,17 +1338,15 @@ def update_memory(
13391338                (default: :obj:`None`) 
13401339                    (default: obj:`None`) 
13411340        """ 
1342- 
1343-         self .memory .write_record (
1344-             MemoryRecord (
1345-                 message = message ,
1346-                 role_at_backend = role ,
1347-                 timestamp = timestamp 
1348-                 if  timestamp  is  not None 
1349-                 else  time .time_ns () /  1_000_000_000 ,  # Nanosecond precision 
1350-                 agent_id = self .agent_id ,
1351-             )
1341+         record  =  MemoryRecord (
1342+             message = message ,
1343+             role_at_backend = role ,
1344+             timestamp = timestamp 
1345+             if  timestamp  is  not None 
1346+             else  time .time_ns () /  1_000_000_000 ,  # Nanosecond precision 
1347+             agent_id = self .agent_id ,
13521348        )
1349+         self .memory .write_record (record )
13531350        self ._register_record_addition (role )
13541351
13551352    def  load_memory (self , memory : AgentMemory ) ->  None :
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments