class TemplateQwen():
    # --- Prompt Template Configuration (ChatML) ---
    IM_START = "<|im_start|>"
    IM_END = "<|im_end|>"
    ROLE_SYSTEM = "system"
    ROLE_USER = "user"
    ROLE_ASSISTANT = "assistant"

class TemplateGemma():
    # --- Prompt Template Configuration (Gemma3) ---
    IM_START = "<start_of_turn>"
    IM_END = "<end_of_turn>"
    ROLE_SYSTEM = "user" # Gemma has no system role
    ROLE_USER = "user"
    ROLE_ASSISTANT = "model"

agent_template = TemplateQwen()
repl_template = TemplateGemma()