Yep, that will work. Now, if that's okay with you @saurabh_aher that can be solution. If you want to stick to using databricks sql function then what I wrote before still applies.

"Unfortunately, if we're talking about databricks sql functions I don't know workaround. "