Only add to memory if the state has changed

PiperOrigin-RevId: 602669867 Change-Id: I3f90dbce4f91933fa27037d98701e53753814664
google-deepmind · Jan 30, 2024 · b50d4eb · b50d4eb
1 parent 06b5388
commit b50d4eb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/concordia/components/agent/person_by_situation.py b/concordia/components/agent/person_by_situation.py
@@ -97,6 +97,7 @@ def update(self) -> None:
     if self._clock_now is not None:
       question = f'Current time: {self._clock_now()}.\n{question}'
 
+    old_state = self._state
     self._state = prompt.open_question(
         question,
         answer_prefix=f'{self._agent_name} would ',
@@ -106,7 +107,8 @@ def update(self) -> None:
 
     self._state = f'{self._agent_name} would {self._state}'
 
-    self._memory.add(f'[intent reflection] {self._state}')
+    if old_state != self._state:
+      self._memory.add(f'[intent reflection] {self._state}')
 
     self._last_chain = prompt
     if self._verbose:

diff --git a/concordia/components/agent/self_perception.py b/concordia/components/agent/self_perception.py
@@ -84,7 +84,7 @@ def update(self) -> None:
     question = (
         f'Given the memories above, what kind of person is {self._agent_name}?'
     )
-
+    old_state = self._state
     self._state = prompt.open_question(
         question,
         answer_prefix=f'{self._agent_name} is ',
@@ -94,7 +94,8 @@ def update(self) -> None:
 
     self._state = f'{self._agent_name} is {self._state}'
 
-    self._memory.add(f'[self reflection] {self._state}')
+    if old_state != self._state:
+      self._memory.add(f'[self reflection] {self._state}')
 
     self._last_chain = prompt
     if self._verbose: