♻️ refactor(deep-agent): customize agent memory middleware path display

Override AgentMemoryMiddleware to simplify agent directory display.
The CustomAgentMemoryMiddleware shows "." instead of full path.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
朱潮 2025-12-31 21:00:49 +08:00
parent a6f166d51e
commit 8c67630bbd

View File

@ -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