Improve get_visual_retriever to enforce images or videos.