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,