Apply AIR302-context check only in @task function (#15711)

This PR updates `AIR302` to only apply the context keys check in `@task`
decorated function.

Ref: #15144

Author: dhruvmanila