diff --git a/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/__init__.py b/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/__init__.py
index f68a3197d9fc..f56657f6f6e4 100644
--- a/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/__init__.py
+++ b/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/__init__.py
@@ -1,3 +1,4 @@
-from .memory_controller import MemoryController
+from .memory_controller import MemoryController, MemoryControllerConfig
+from ._memory_bank import MemoryBankConfig
 
-__all__ = ["MemoryController"]
+__all__ = ["MemoryController", "MemoryControllerConfig", "MemoryBankConfig"]
diff --git a/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/utils/__init__.py b/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/utils/__init__.py
index 992e0e8ef1c0..82bf516d28d0 100644
--- a/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/utils/__init__.py
+++ b/python/packages/autogen-ext/src/autogen_ext/experimental/task_centric_memory/utils/__init__.py
@@ -1,7 +1,15 @@
-from .apprentice import Apprentice
+from .apprentice import Apprentice, ApprenticeConfig
 from .chat_completion_client_recorder import ChatCompletionClientRecorder
 from .grader import Grader
-from .page_logger import PageLogger
+from .page_logger import PageLogger, PageLoggerConfig
 from .teachability import Teachability
 
-__all__ = ["Apprentice", "ChatCompletionClientRecorder", "Grader", "PageLogger", "Teachability"]
+__all__ = [
+    "Apprentice",
+    "ChatCompletionClientRecorder",
+    "Grader",
+    "PageLogger",
+    "Teachability",
+    "ApprenticeConfig",
+    "PageLoggerConfig",
+]