models/base_model.py

class Simulator:

    def __init__(self, model_name_or_path):
        raise NotImplementedError


    def generate_query(self, messages):
        """ user simulator
        :param messages:
        :return:
        """
        raise NotImplementedError

    def generate_response(self, messages):
        """ assistant simulator
        :param messages:
        :return:
        """
        raise NotImplementedError