from datasets import Dataset, disable_progress_bars
from sentence_transformers import SentenceTransformer
from sentence_transformers.util import mine_hard_negatives


disable_progress_bars()

pairs = Dataset.from_list(
    [
        {"query": "reset a user password", "answer": "Open the user profile and send a password reset email."},
        {"query": "unlock a user account", "answer": "Open the admin user record and clear the account lock."},
        {"query": "enable two-factor authentication", "answer": "Open account security and enroll an authenticator app."},
        {"query": "rotate an API token", "answer": "Revoke the old API token and create a replacement token."},
        {"query": "restore a deleted project", "answer": "Open deleted projects and restore the selected project."},
        {"query": "export audit logs", "answer": "Open compliance reports and export the audit log CSV."},
        {"query": "invite a new team member", "answer": "Open team settings and send an invitation email."},
        {"query": "change the billing contact", "answer": "Open billing settings and update the primary contact."},
    ]
)