♻️ 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:
parent
a6f166d51e
commit
8c67630bbd
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user