feat: add more CoT templates

Suggested by u/Chromix_ on Reddit
2025-12-20 17:12:46 +05:30
parent 92d0c0d551
commit 5ddef6fd2f
1 changed files with 6 additions and 0 deletions
@@ -291,6 +291,12 @@ def run():
    elif model.response_prefix.startswith("<|channel|>analysis<|message|>"):
        # gpt-oss.
        model.response_prefix = "<|channel|>analysis<|message|><|end|><|start|>assistant<|channel|>final<|message|>"
+    elif model.response_prefix.startswith("<thought>"):
+        # Unknown, suggested by user.
+        model.response_prefix = "<thought></thought>"
+    elif model.response_prefix.startswith("[THINK]"):
+        # Unknown, suggested by user.
+        model.response_prefix = "[THINK][/THINK]"

    if model.response_prefix:
        print(f"* Prefix found: [bold]{model.response_prefix!r}[/]")