Class _FakeTool

tests/lib/tools/test_session_runner.py:238–252

class _FakeTool:
    def __init__(
        self,
        name: str,
        fn: Callable[[dict[str, Any]], Awaitable[Any]],
        *,
        close: Callable[[], Any] | None = None,
    ) -> None:
        self.name = name
        self._fn = fn
        if close is not None:
            self.close = close

    def call(self, input: dict[str, Any]) -> Any:
        return self._fn(input)


class _SyncTool(BetaBuiltinFunctionTool):