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:
@@ -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", "")
|
||||
|
||||
Reference in New Issue
Block a user