You could fine-tune the embedding model to reduce cosine distance on a more specific function.