Augmented Embeddings for Custom Retrievals