From c6512c919b69046fb707ff2e4f13b5417d030aa1 Mon Sep 17 00:00:00 2001
From: jyh20030112 <944878197@qq.com>
Date: Tue, 28 Jul 2026 23:57:51 +0800
Subject: [PATCH] refactor: scope tool configuration to tool middleware

---
 src/ejagent/middleware/base.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/ejagent/middleware/base.py b/src/ejagent/middleware/base.py
index af87a3e..4462d7b 100644
--- a/src/ejagent/middleware/base.py
+++ b/src/ejagent/middleware/base.py
@@ -37,12 +37,6 @@ def __init__(self, *, name: str | None = None, enabled: bool = True) -> None:
         self.name = name or type(self).__name__
         self.enabled = enabled
 
-    def configure_tools(
-        self,
-        tool_definitions: Sequence[ToolDefinition],
-    ) -> None:
-        """Validate or cache the static tool set before resource startup."""
-
     async def startup(self) -> None:
         """Initialize optional middleware resources."""
 
@@ -56,6 +50,12 @@ async def on_task_start(self) -> None:
 class ToolMiddleware(Middleware):
     """Decorator around one tool execution."""
 
+    def configure_tools(
+        self,
+        tool_definitions: Sequence[ToolDefinition],
+    ) -> None:
+        """Validate or cache the resolved tool set before middleware startup."""
+
     async def __call__(
         self,
         context: ToolCallContext,