core/refiner/base_refiner.py

from models import BaseModel
class BaseRefiner:
    def __init__(self,
                 sys_prompt: str, 
                 model: BaseModel,
        ) -> None:
        self.sys_prompt = sys_prompt
        self.model = model
    
    def refine(self, message: str) -> str:
        raise NotImplementedError