from fastapi import Request

from common_logging import get_logger

logger = get_logger(__name__)


class Translator:

    def t(self, key: str) -> str:
        translations = {'sft_task.not_found': 'SFT task not found', 'sft_task.deleted': 'SFT task deleted', 'dpo_task.not_found': 'DPO task not found', 'dpo_task.deleted': 'DPO task deleted'}
        return translations.get(key, key)

def get_translator(request: Request) -> Translator:
    return Translator()