diff --git a/reme_ai/retrieve/__init__.py b/reme_ai/retrieve/__init__.py
index 9c95d946..142d2630 100644
--- a/reme_ai/retrieve/__init__.py
+++ b/reme_ai/retrieve/__init__.py
@@ -4,14 +4,17 @@ This module provides submodules for different types of memory retrieval:
 - personal: Personal memory retrieval operations
 - task: Task memory retrieval operations
 - tool: Tool memory retrieval operations
+- working: Working-memory retrieval operations
 """
 
 from . import personal
 from . import task
 from . import tool
+from . import working
 
 __all__ = [
     "personal",
     "task",
     "tool",
+    "working",
 ]
diff --git a/reme_ai/summary/__init__.py b/reme_ai/summary/__init__.py
index 0dd27ce5..47688a15 100644
--- a/reme_ai/summary/__init__.py
+++ b/reme_ai/summary/__init__.py
@@ -1,17 +1,20 @@
-"""Summary operations module.
+"""Summary operations' module.
 
 This module provides summary operations for different types of memories:
 - Personal memory summary operations
 - Task memory summary operations
 - Tool memory summary operations
+- Working memory summary operations
 """
 
 from . import personal
 from . import task
 from . import tool
+from . import working
 
 __all__ = [
     "personal",
     "task",
     "tool",
+    "working",
 ]