diff --git a/reme/extension/procedural_memory/retriever/__init__.py b/reme/workflow/__init__.py
similarity index 100%
rename from reme/extension/procedural_memory/retriever/__init__.py
rename to reme/workflow/__init__.py
diff --git a/reme/extension/procedural_memory/__init__.py b/reme/workflow/procedural_memory/__init__.py
similarity index 100%
rename from reme/extension/procedural_memory/__init__.py
rename to reme/workflow/procedural_memory/__init__.py
diff --git a/reme/extension/procedural_memory/summary/__init__.py b/reme/workflow/procedural_memory/retriever/__init__.py
similarity index 100%
rename from reme/extension/procedural_memory/summary/__init__.py
rename to reme/workflow/procedural_memory/retriever/__init__.py
diff --git a/reme/extension/procedural_memory/summarizer/__init__.py b/reme/workflow/procedural_memory/summarizer/__init__.py
similarity index 100%
rename from reme/extension/procedural_memory/summarizer/__init__.py
rename to reme/workflow/procedural_memory/summarizer/__init__.py
diff --git a/reme/extension/procedural_memory/summarizer/success_extraction.py b/reme/workflow/procedural_memory/summarizer/success_extraction.py
similarity index 100%
rename from reme/extension/procedural_memory/summarizer/success_extraction.py
rename to reme/workflow/procedural_memory/summarizer/success_extraction.py
diff --git a/reme/extension/procedural_memory/summarizer/success_extraction.yaml b/reme/workflow/procedural_memory/summarizer/success_extraction.yaml
similarity index 100%
rename from reme/extension/procedural_memory/summarizer/success_extraction.yaml
rename to reme/workflow/procedural_memory/summarizer/success_extraction.yaml
diff --git a/reme/extension/procedural_memory/summarizer/trajectory_preprocess.py b/reme/workflow/procedural_memory/summarizer/trajectory_preprocess.py
similarity index 100%
rename from reme/extension/procedural_memory/summarizer/trajectory_preprocess.py
rename to reme/workflow/procedural_memory/summarizer/trajectory_preprocess.py
diff --git a/reme/workflow/procedural_memory/summary/__init__.py b/reme/workflow/procedural_memory/summary/__init__.py
new file mode 100644
index 00000000..e69de29b