Update ai_caching module