- Replaced `tool` argument with `tool_cls`. - Removed `space_type` argument that's not present in the method. Ref: !118900