Fine-tuning/backend/app/engines/text_engine.py

from app.engines.base import BaseEngine


class TextEngine(BaseEngine):
    """Training engine for LLaMA, Qwen, and other text-only LLMs."""

    async def load_model(self, model_id: str, **kwargs):
        raise NotImplementedError

    def get_peft_config(self, method: str, params: dict):
        raise NotImplementedError

    async def preprocess_dataset(self, dataset_path: str, output_path: str, **kwargs):
        raise NotImplementedError

    async def train(self, job_id: str, dataset_path: str, peft_config, training_args: dict):
        raise NotImplementedError

    def get_model_info(self, model_id: str):
        raise NotImplementedError