diff --git a/__init__.py b/__init__.py
index 03193a2..104f676 100644
--- a/__init__.py
+++ b/__init__.py
@@ -350,11 +350,11 @@ class Mem0LocalMemoryProvider(MemoryProvider):
         def _sync():
             try:
                 client = self._get_client()
-                # Combine user and assistant content for context
                 combined = f"User: {user_content}\nAssistant: {assistant_content}"
                 client.add(
                     message=combined,
                     user_id=self._user_id,
+                    agent_id=self._agent_id,
                 )
                 self._record_success()
             except Exception as e: