Add context support for deploy and forecast