diff --git a/agent/deep_assistant.py b/agent/deep_assistant.py
index 75ca1bd..fd79b6f 100644
--- a/agent/deep_assistant.py
+++ b/agent/deep_assistant.py
@@ -169,6 +169,7 @@ async def init_agent(config: AgentConfig):
             system_prompt=system_prompt,
             tools=mcp_tools,
             auto_approve=True,
+            enable_memory=False,
             workspace_root=workspace_root
         )
     else:
@@ -215,6 +216,21 @@ async def init_agent(config: AgentConfig):
     logger.info(f"create {config.robot_type} elapsed: {time.time() - create_start:.3f}s")
     return agent, checkpointer
 
+class CustomAgentMemoryMiddleware(AgentMemoryMiddleware):
+    def __init__(
+            self,
+            *,
+            settings,
+            assistant_id: str,
+            system_prompt_template: str | None = None,
+        ) -> None:
+            super().__init__(
+                settings=settings,
+                assistant_id=assistant_id,
+                system_prompt_template=system_prompt_template
+            )
+            self.agent_dir_display = f"."
+
 class CustomSkillsMiddleware(SkillsMiddleware):
     def __init__(
         self,
@@ -319,7 +335,7 @@ def create_custom_cli_agent(
         # Add memory middleware
         if enable_memory:
             agent_middleware.append(
-                AgentMemoryMiddleware(settings=settings, assistant_id=assistant_id)
+                CustomAgentMemoryMiddleware(settings=settings, assistant_id=assistant_id)
             )
 
         # Add skills middleware