APEX AI Agent Logging with Request & Response Handlers

Thanks for this article. It helped me a lot.

But, I have a query, when I use APEX_AI.GENERATE on button click, this works well but can I pass a custom param in that case for logging my internal tool call because in case of APEX_AI.GENERATE, it doesn't add any tool or doesn't pass any extra custom param.