diff --git a/agent/deep_assistant.py b/agent/deep_assistant.py index fa8bf11..2c467e7 100644 --- a/agent/deep_assistant.py +++ b/agent/deep_assistant.py @@ -289,19 +289,6 @@ class CustomAgentMemoryMiddleware(MemoryMiddleware): class CustomSkillsMiddleware(SkillsMiddleware): """自定义的 SkillsMiddleware,使用新的签名格式""" - def __init__( - self, - *, - backend: FilesystemBackend, - sources: list[str], - project_skills_display: str = "./skills", - ) -> None: - super().__init__( - backend=backend, - sources=sources, - ) - self.project_skills_display = project_skills_display - def before_agent(self, state, runtime, config): """Load skills metadata before agent execution. @@ -314,14 +301,26 @@ class CustomSkillsMiddleware(SkillsMiddleware): Returns: Updated state with skills_metadata populated. """ - state = super().before_agent(state, runtime) - # 新版本的 skills_metadata 结构可能不同,需要检查 - if "skills_metadata" in state and state["skills_metadata"]: - for item in state["skills_metadata"]: - # 修改路径显示为相对路径 - if "path" in item and item["path"]: - # 简化为只显示相对路径 - item["path"] = f"./skills/{item.get('name', 'unknown')}/SKILL.md" + if "skills_metadata" in state and len(state["skills_metadata"]) ==0 : + del state["skills_metadata"] + state = super().before_agent(state, runtime, config) + return state + + async def abefore_agent(self, state, runtime, config): + """Load skills metadata before agent execution. + + 修改路径显示为相对路径。 + + Args: + state: Current agent state. + runtime: Runtime context. + + Returns: + Updated state with skills_metadata populated. + """ + if "skills_metadata" in state and len(state["skills_metadata"]) ==0 : + del state["skills_metadata"] + state = await super().abefore_agent(state, runtime, config) return state def create_custom_cli_agent( @@ -442,8 +441,7 @@ def create_custom_cli_agent( # Add skills middleware (using new signature) if enable_skills: - skills_sources = [workspace_root] - skills_sources.append(workspace_root + "/skills") + skills_sources = ["./skills"] agent_middleware.append( CustomSkillsMiddleware( @@ -480,8 +478,7 @@ def create_custom_cli_agent( # Add skills middleware if enable_skills: - skills_sources = [workspace_root] - skills_sources.append(workspace_root + "/skills") + skills_sources = ["/skills"] agent_middleware.append( CustomSkillsMiddleware(