♻️ 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,
|
system_prompt=system_prompt,
|
||||||
tools=mcp_tools,
|
tools=mcp_tools,
|
||||||
auto_approve=True,
|
auto_approve=True,
|
||||||
|
enable_memory=False,
|
||||||
workspace_root=workspace_root
|
workspace_root=workspace_root
|
||||||
)
|
)
|
||||||
else:
|
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")
|
logger.info(f"create {config.robot_type} elapsed: {time.time() - create_start:.3f}s")
|
||||||
return agent, checkpointer
|
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):
|
class CustomSkillsMiddleware(SkillsMiddleware):
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
@ -319,7 +335,7 @@ def create_custom_cli_agent(
|
|||||||
# Add memory middleware
|
# Add memory middleware
|
||||||
if enable_memory:
|
if enable_memory:
|
||||||
agent_middleware.append(
|
agent_middleware.append(
|
||||||
AgentMemoryMiddleware(settings=settings, assistant_id=assistant_id)
|
CustomAgentMemoryMiddleware(settings=settings, assistant_id=assistant_id)
|
||||||
)
|
)
|
||||||
|
|
||||||
# Add skills middleware
|
# Add skills middleware
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user