Ty_Tech/heretic
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add more CoT templates

Browse Source 
Suggested by u/Chromix_ on Reddit
This commit is contained in:
Philipp Emanuel Weidmann
2025-12-20 17:12:46 +05:30
parent 92d0c0d551
commit 5ddef6fd2f
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/heretic/main.py
+6
View File
@@ -291,6 +291,12 @@ def run():
elif model.response_prefix.startswith("<|channel|>analysis<|message|>"): elif model.response_prefix.startswith("<|channel|>analysis<|message|>"):
# gpt-oss. # gpt-oss.
model.response_prefix = "<|channel|>analysis<|message|><|end|><|start|>assistant<|channel|>final<|message|>" 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: if model.response_prefix:
print(f"* Prefix found: [bold]{model.response_prefix!r}[/]") print(f"* Prefix found: [bold]{model.response_prefix!r}[/]")