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:
|
def _reject(self, request) -> ToolMessage:
|
||||||
name = request.tool_call.get("name", "")
|
name = request.tool_call.get("name", "")
|
||||||
|
|||||||
Reference in New Issue
Block a user