From da0facfd4d8779d8a099216e92260bfc66df005b Mon Sep 17 00:00:00 2001 From: gleb Date: Mon, 25 May 2026 23:27:25 +0300 Subject: [PATCH] fix: add __init__ to SanitizeToolCallsMiddleware AgentMiddleware is a plain Generic class, not a Pydantic BaseModel, so class-level field declarations don't work. Explicit __init__ required. Co-Authored-By: Claude Sonnet 4.6 --- src/agent/middlewares/sanitize_tool_calls.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/agent/middlewares/sanitize_tool_calls.py b/src/agent/middlewares/sanitize_tool_calls.py index a11a754..033b660 100644 --- a/src/agent/middlewares/sanitize_tool_calls.py +++ b/src/agent/middlewares/sanitize_tool_calls.py @@ -11,7 +11,8 @@ class SanitizeToolCallsMiddleware(AgentMiddleware[AgentState[Any], Any]): """Перехватывает вызовы инструментов с неизвестными именами и возвращает понятное сообщение об ошибке вместо падения рантайма.""" - known_tools: set[str] + def __init__(self, known_tools: set[str]): + self.known_tools = known_tools def _reject(self, request) -> ToolMessage: name = request.tool_call.get("name", "")