hold lock on memory read to prevent reading during a write.

PiperOrigin-RevId: 672851018 Change-Id: I797a7a0379531a98ed9fa55f911edff627e497dc
google-deepmind · Sep 10, 2024 · 4066851 · 4066851
1 parent 7b9e9b3
commit 4066851
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/concordia/components/agent/memory_component.py b/concordia/components/agent/memory_component.py
@@ -74,7 +74,8 @@ def retrieve(
       A list of memory results.
     """
     self._check_phase()
-    return self._memory.retrieve(query, scoring_fn, limit)
+    with self._lock:
+      return self._memory.retrieve(query, scoring_fn, limit)
 
   def add(
       self,