diff --git a/backend/src/core/agentscope/tools/__init__.py b/backend/src/core/agentscope/tools/__init__.py
index e8718b1..f8afc58 100644
--- a/backend/src/core/agentscope/tools/__init__.py
+++ b/backend/src/core/agentscope/tools/__init__.py
@@ -1,3 +1 @@
-from core.agentscope.tools.toolkit import build_stage_toolkit, build_toolkit
-
-__all__ = ["build_toolkit", "build_stage_toolkit"]
+"""AgentScope tools package."""
diff --git a/backend/src/core/agentscope/tools/toolkit.py b/backend/src/core/agentscope/tools/toolkit.py
index 432bb68..eb1c2da 100644
--- a/backend/src/core/agentscope/tools/toolkit.py
+++ b/backend/src/core/agentscope/tools/toolkit.py
@@ -29,8 +29,8 @@ TOOL_FUNCTIONS: dict[str, Any] = {
 
 
 AGENT_TYPE_TO_GROUPS: dict[AgentType, set[ToolGroup]] = {
-    AgentType.ROUTER: {ToolGroup.READ},
     AgentType.WORKER: {ToolGroup.READ, ToolGroup.WRITE},
+    AgentType.MEMORY: {ToolGroup.READ, ToolGroup.WRITE},
 }