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 <noreply@anthropic.com>
This commit is contained in:
2026-05-25 23:27:25 +03:00
parent c95d7375cc
commit da0facfd4d
+2 -1
View File
@@ -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", "")