repository/repository_abc.py

import abc


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, model_name: str, model_roles: ModelRoles):
 self.name = model_name
 self.roles = model_roles


class Repository(abc.ABC):

 def get_model_info(self) -> Model:
 pass

 def get_model_roles(self) -> ModelRoles:
 pass

 def get_message_history(self) -> list[dict[str, str]]:
 pass

 def send_prompt(self, prompt: str, add_to_history: bool) -> dict[str, str]:
 pass

 def set_message_for_role(self, role: str, message: str):
 self.get_message_history().append({"role": role, "content": message})

 def init(self):
 pass