refactor: improve code readability and update model schema access in tool_usage.py

refactor: improve code readability and update model schema access in tool_usage.py #1854