Email Smart Reply with AI-enriched Context