Initial commit

templates.py

@@ -0,0 +1,18 @@
+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()