llm/llm.py

class ModelRoles:
    def __init__(self, system_role: str, user_role: str, ai_role: str):
        self.system_role: str = system_role
        self.user_role: str = user_role
        self.ai_role: str = ai_role


class Model:
    def __init__(self, name: str, roles: ModelRoles):
        self.name: str = name
        self.roles: ModelRoles = roles