All other methods that have `tool_context` as an argument, refer to it as "tool_context" in the docstring under Args. Co-authored-by: Hangfei Lin <hangfei@google.com>
