Skip to content

Commit bc69be7

Browse files
committed
Update chat_agent.py
1 parent d184944 commit bc69be7

File tree

1 file changed

+9
-12
lines changed

1 file changed

+9
-12
lines changed

camel/agents/chat_agent.py

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)